#coding: UTF-8 from seahub.share.models import FileShare, UploadLinkShare from seahub.test_utils import BaseTestCase from tests.common.utils import urljoin, randstring from tests.api.apitestbase import ApiTestBase from tests.api.urls import SHARED_LINKS_URL #class SharesApiTest(ApiTestBase): # def test_create_file_shared_link(self): # with self.get_tmp_repo() as repo: # fname, _ = self.create_file(repo) # fsurl = urljoin(repo.file_url, 'shared-link') # data = { # 'type': 'f', # 'p': '/' + fname, # } # res = self.put(fsurl, data=data, expected=201) # self.assertRegexpMatches(res.headers['Location'], \ # r'http(.*)/f/(\w{10,10})/') # # res = self.get(SHARED_LINKS_URL).json() # self.assertNotEmpty(res) # for fileshare in res['fileshares']: # self.assertIsNotNone(fileshare['username']) # self.assertIsNotNone(fileshare['repo_id']) # #self.assertIsNotNone(fileshare['ctime']) # self.assertIsNotNone(fileshare['s_type']) # self.assertIsNotNone(fileshare['token']) # self.assertIsNotNone(fileshare['view_cnt']) # self.assertIsNotNone(fileshare['path']) # # class FileSharedLinkApiTest(BaseTestCase): def tearDown(self): self.remove_repo() def test_create_file_shared_link_with_invalid_path(self): self.login_as(self.user) resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "ps=%s&type=f" % (self.file), 'application/x-www-form-urlencoded', ) self.assertEqual(400, resp.status_code) resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "type=f", 'application/x-www-form-urlencoded', ) self.assertEqual(400, resp.status_code) resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=f" % randstring(6), 'application/x-www-form-urlencoded', ) self.assertEqual(400, resp.status_code) def test_can_create_file_download_link(self): self.login_as(self.user) # create file download share link resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=f" % (self.file), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.assertRegexpMatches(resp._headers['location'][1], \ r'http(.*)/f/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] self.assertIsNotNone(FileShare.objects.get(token=token)) def test_can_create_file_download_link_with_exipre(self): self.login_as(self.user) # create file download share link with expire resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=f&expire=5" % (self.file), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.assertRegexpMatches(resp._headers['location'][1], \ r'http(.*)/f/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.expire_date) def test_can_create_file_download_link_with_password(self): self.login_as(self.user) # create file download share link with password resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=f&password=123" % (self.file), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.assertRegexpMatches(resp._headers['location'][1], \ r'http(.*)/f/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.password) def test_can_create_file_download_link_with_password_exipre(self): self.login_as(self.user) # create file download share link with password and expire resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=f&password=123&expire=5" % (self.file), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.assertRegexpMatches(resp._headers['location'][1], \ r'http(.*)/f/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.expire_date) self.assertIsNotNone(fileshare.password) def test_can_create_dir_download_link(self): self.login_as(self.user) # create dir download share link resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] self.assertIsNotNone(FileShare.objects.get(token=token)) def test_can_create_dir_download_link_with_exipre(self): self.login_as(self.user) # create dir download share link with expire resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d&expire=5" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.expire_date) def test_can_create_dir_download_link_with_password(self): self.login_as(self.user) # create dir download share link with password resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d&password=123" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.password) def test_can_create_dir_download_link_with_password_exipre(self): self.login_as(self.user) # create dir download share link with password and expire resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d&password=123&expire=5" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] fileshare = FileShare.objects.get(token=token) self.assertIsNotNone(fileshare.expire_date) self.assertIsNotNone(fileshare.password) def test_can_create_dir_upload_link(self): self.login_as(self.user) # create dir download share link resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d&share_type=upload" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/u/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] self.assertIsNotNone(UploadLinkShare.objects.get(token=token)) def test_can_create_dir_upload_link_with_password(self): self.login_as(self.user) # create dir download share link with password resp = self.client.put( '/api2/repos/%s/file/shared-link/' % (self.repo.id), "p=%s&type=d&share_type=upload&password=123" % (self.folder), 'application/x-www-form-urlencoded', ) self.assertEqual(201, resp.status_code) self.dir_link_location = resp._headers['location'][1] self.assertRegexpMatches(self.dir_link_location, \ r'http(.*)/u/d/(\w{10,10})/') token = resp._headers['location'][1].split('/')[-2] uls = UploadLinkShare.objects.get(token=token) self.assertIsNotNone(uls.password) class SharedFileDetailApiTest(BaseTestCase): def _add_file_shared_link_with_password(self): password = randstring(6) uls = FileShare.objects.create_file_link(self.user.username, self.repo.id, self.file, password) return (uls.token, password) def tearDown(self): self.remove_repo() def test_get_file_share_detail_with_password(self): self.login_as(self.user) token, password = self._add_file_shared_link_with_password() url = '/api2/f/%s/detail/' % (token) resp = self.client.get(url + '?password=' + password) self.assertEqual(200, resp.status_code) self.assertIsNotNone(resp) def test_get_file_share_detail_with_invalid_password(self): self.login_as(self.user) token, password = self._add_file_shared_link_with_password() url = '/api2/f/%s/detail/' % (token) resp = self.client.get(url + '?password=' + randstring(5)) self.assertEqual(403, resp.status_code) resp = self.client.get('/api2/f/%s/detail/' % (token)) self.assertEqual(403, resp.status_code) class SharedDirApiTest(BaseTestCase): def tearDown(self): self.remove_repo() def _add_dir_download_link(self): ls = FileShare.objects.create_dir_link(self.user.username, self.repo.id, self.folder) self.create_folder(repo_id=self.repo.id, parent_dir=self.folder, dirname='sub-folder', username='test@test.com') self.create_file(repo_id=self.repo.id, parent_dir=self.folder, filename='sub-test.txt', username='test@test.com') return ls.token def test_can_get_direntry_in_dir_download_link(self): self.login_as(self.user) token = self._add_dir_download_link() # get direntry in dir download share link resp = self.client.get( '/api2/d/%s/dir/' % (token), ) self.assertEqual(200, resp.status_code) self.assertIsNotNone(resp) class SharedLinksApiTest(BaseTestCase): def tearDown(self): self.remove_repo() def _add_file_shared_link(self): ls = FileShare.objects.create_dir_link(self.user.username, self.repo.id, self.folder) return ls.token def _add_upload_shared_link(self): uls = UploadLinkShare.objects.create_upload_link_share( self.user.username, self.repo.id, self.folder) return uls.token def test_can_delete_file_shared_link(self): self.login_as(self.user) token = self._add_file_shared_link() resp = self.client.get( '/api2/shared-links/?t=%s' % (token), ) self.assertEqual(200, resp.status_code) def test_can_delete_upload_shared_link(self): self.login_as(self.user) token = self._add_upload_shared_link() resp = self.client.get( '/api2/shared-links/?t=%s' % (token), ) self.assertEqual(200, resp.status_code)