#!/bin/bash

set -e

source common.sh

function test_sync() {
	dvc run -D code/code.sh -d data/foo -o data/foo1 bash code/code.sh data/foo data/foo1

	dvc_info "Checking status"
	dvc status | grep "new file" || dvc_fail

	dvc_info "Pushing data"
	dvc push

	dvc_info "Checking status"
	dvc status

	dvc_info "Removing all cache"
	rm -rf .dvc/cache/*
	rm -rf data/foo data/bar

	dvc_info "Checking status"
	dvc status | grep "deleted" || dvc_fail

	dvc_info "Pulling data"
	dvc pull
	dvc_check_files data/foo data/bar

	dvc_info "Checking status"
	dvc status
} 

function test_aws() {
	dvc_create_repo

	dvc_info "Setting up AWS cloud"
	dvc_clean_cloud_aws
	dvc config aws.storagepath $TEST_REPO_S3
	dvc config aws.region $TEST_REPO_REGION
	dvc config global.cloud AWS

	test_sync

	dvc_clean_cloud_aws
}

function test_gcp() {
	dvc_create_repo

	dvc_info "Setting up GCP cloud"
	dvc_clean_cloud_gcp
	dvc config gcp.storagepath $TEST_REPO_GCP
	dvc config gcp.projectname $TEST_REPO_GCP_PROJECT
	dvc config global.cloud GCP

	test_sync

	dvc_clean_cloud_gcp
}

if [[ "$TRAVIS_PULL_REQUEST" == "false" && "$TRAVIS_SECURE_ENV_VARS" == "true" ]]; then
	test_aws
	# Temporarily disabled for travis
	if [[ "$TRAVIS" != "true" ]]; then
		test_gcp
	fi
fi

dvc_pass