batch copy folders

tests/api/endpoints/test_repos_batch.py

@@ -1,8 +1,11 @@
+import os
 import json
+import posixpath
 from seaserv import seafile_api, ccnet_api
 from django.core.urlresolvers import reverse
 from tests.common.utils import randstring
 from seahub.test_utils import BaseTestCase
+from seahub.utils import normalize_dir_path
 
 class ReposBatchViewTest(BaseTestCase):
 
@@ -193,3 +196,130 @@ class ReposBatchViewTest(BaseTestCase):
         }
         resp = self.client.post(self.url, data)
         self.assertEqual(403, resp.status_code)
+
+
+class ReposBatchCopyDirView(BaseTestCase):
+
+    def setUp(self):
+        self.user_name = self.user.username
+        self.admin_name = self.admin.username
+        self.repo_id = self.repo.id
+        self.url = reverse('api-v2.1-repos-batch-copy-dir')
+
+    def tearDown(self):
+        self.remove_repo()
+        self.remove_group()
+
+    def get_random_path(self):
+        return '/%s/%s/%s/' % (randstring(2), \
+                randstring(2), randstring(2))
+
+    def create_new_repo(self, username):
+        new_repo_id = seafile_api.create_repo(name=randstring(10),
+                desc='', username=username, passwd=None)
+
+        return new_repo_id
+
+    def test_copy_dir(self):
+
+        self.login_as(self.user)
+
+        # create two folders in src repo
+        src_folder_1 = self.get_random_path()
+        src_folder_2 = self.get_random_path()
+        for path in [src_folder_1, src_folder_2]:
+            seafile_api.mkdir_with_parents(self.repo_id,
+                    '/', path.strip('/'), self.user_name)
+
+        # share admin's tmp repo to user
+        tmp_repo_id = self.create_new_repo(self.admin_name)
+        seafile_api.share_repo(tmp_repo_id, self.admin_name,
+                self.user_name, 'rw')
+
+        # create two folders as parent dirs in dst repo for admin user
+        dst_folder_1 = self.get_random_path()
+        seafile_api.mkdir_with_parents(tmp_repo_id,
+                '/', dst_folder_1.strip('/'), self.admin_name)
+
+        dst_folder_2 = '/'
+
+        # copy folders
+        data = {
+            "src_repo_id": self.repo_id,
+            "dst_repo_id": tmp_repo_id,
+            "path": [
+                {"src_path": src_folder_1, "dst_path": dst_folder_1},
+                {"src_path": src_folder_2, "dst_path": dst_folder_2},
+            ]
+        }
+
+        resp = self.client.post(self.url, json.dumps(data),
+                'application/json')
+        self.assertEqual(200, resp.status_code)
+
+        json_resp = json.loads(resp.content)
+        assert len(json_resp['success']) == 2
+        assert len(json_resp['failed']) == 0
+
+        def folder_exist(src_folder, dst_repo_id, dst_folder):
+            src_obj_name = os.path.basename(src_folder.rstrip('/'))
+            full_dst_folder_path = posixpath.join(dst_folder.strip('/'),
+                    src_obj_name.strip('/'))
+            full_dst_folder_path = normalize_dir_path(full_dst_folder_path)
+            return seafile_api.get_dir_id_by_path(dst_repo_id,
+                    full_dst_folder_path) is not None
+
+        assert folder_exist(src_folder_1, tmp_repo_id, dst_folder_1)
+        assert folder_exist(src_folder_2, tmp_repo_id, dst_folder_2)
+
+        self.remove_repo(tmp_repo_id)
+
+    def test_copy_dir_with_invalid_repo_permisson(self):
+
+        self.login_as(self.user)
+
+        # create two folders as parent dirs in dst repo for admin user
+        tmp_repo_id = self.create_new_repo(self.admin_name)
+
+        # copy folders
+        data = {
+            "src_repo_id": self.repo_id,
+            "dst_repo_id": tmp_repo_id,
+            "path": [
+                {"src_path": '/', "dst_path": '/'},
+                {"src_path": '/', "dst_path": '/'},
+            ]
+        }
+
+        resp = self.client.post(self.url, json.dumps(data),
+                'application/json')
+        self.assertEqual(403, resp.status_code)
+
+    def test_copy_dir_with_src_path_is_root_folder(self):
+
+        self.login_as(self.user)
+
+        # create two folders as parent dirs in dst repo for admin user
+        tmp_repo_id = self.create_new_repo(self.admin_name)
+        seafile_api.share_repo(tmp_repo_id, self.admin_name,
+                self.user_name, 'rw')
+
+        # copy folders
+        data = {
+            "src_repo_id": self.repo_id,
+            "dst_repo_id": tmp_repo_id,
+            "path": [
+                {"src_path": '/', "dst_path": '/'},
+            ]
+        }
+
+        resp = self.client.post(self.url, json.dumps(data),
+                'application/json')
+        self.assertEqual(200, resp.status_code)
+
+        json_resp = json.loads(resp.content)
+        assert len(json_resp['success']) == 0
+        assert len(json_resp['failed']) == 1
+
+        assert json_resp['failed'][0]['error_msg'] == \
+                "The source path can not be '/'."