From a05eddd746317f3274763a8eb3f30c2fecca0754 Mon Sep 17 00:00:00 2001
From: Ruslan Kuprieiev <kupruser@gmail.com>
Date: Tue, 27 Feb 2018 02:25:57 +0200
Subject: [PATCH] tests: add test for cli data cloud

Signed-off-by: Ruslan Kuprieiev <kupruser@gmail.com>
---
 tests/test_data_cloud.py | 51 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)

diff --git a/tests/test_data_cloud.py b/tests/test_data_cloud.py
index 9bbd908df..e86efa961 100644
--- a/tests/test_data_cloud.py
+++ b/tests/test_data_cloud.py
@@ -3,6 +3,7 @@ import stat
 import uuid
 import shutil
 
+from dvc.main import main
 from dvc.config import ConfigError
 from dvc.cloud.data_cloud import DataCloud, DataCloudAWS, DataCloudLOCAL, DataCloudGCP
 from dvc.cloud.base import STATUS_UNKNOWN, STATUS_OK, STATUS_MODIFIED, STATUS_NEW, STATUS_DELETED
@@ -178,3 +179,53 @@ class TestDataCloudLOCAL(TestDataCloudBase):
         if self._should_test():
             self._test_cloud()
             self.assertTrue(os.path.isdir(self.dname))
+
+
+class TestDataCloudLocalCli(TestDvc):
+    def main(self, args):
+        ret = main(args)
+        self.assertEqual(ret, 0)
+
+    def test(self):
+        #FIXME enable this test for windows
+        if os.name == 'nt':
+            return
+        
+        dname = 'cloud'
+        os.mkdir(dname)
+
+        self.main(['config', 'global.cloud', 'LOCAL'])
+        self.main(['config', 'LOCAL.StoragePath', dname])
+
+        stage = self.dvc.add(self.FOO)
+        cache = stage.outs[0].cache
+
+        stage_dir = self.dvc.add(self.DATA_DIR)
+        cache_dir = stage_dir.outs[0].cache
+
+        #FIXME check status output
+        self.main(['cache', 'status'])
+
+        self.main(['cache', 'push'])
+        self.assertTrue(os.path.exists(cache))
+        self.assertTrue(os.path.isfile(cache))
+        self.assertTrue(os.path.exists(cache_dir))
+        self.assertTrue(os.path.isdir(cache_dir))
+
+        self.main(['cache', 'status'])
+
+        os.chmod(cache, 0o777)
+        os.unlink(cache)
+        self.dvc._remove_cache(cache_dir)
+
+        self.main(['cache', 'status'])
+
+        self.main(['cache', 'pull'])
+        self.assertTrue(os.path.exists(cache))
+        self.assertTrue(os.path.isfile(cache))
+        with open(cache, 'r') as fd:
+            self.assertEqual(fd.read(), self.FOO_CONTENTS)
+        self.assertTrue(os.path.exists(cache_dir))
+        self.assertTrue(os.path.isdir(cache_dir))
+
+        self.main(['cache', 'status'])