seahub/tests/api/test_shares.py

#coding: UTF-8
import json

from django.core.urlresolvers import reverse
from django.test import TestCase

from seaserv import seafile_api

from seahub.test_utils import Fixtures
from tests.common.utils import urljoin
from tests.api.apitestbase import ApiTestBase
from tests.api.urls import SHARED_LINKS_URL, SHARED_LIBRARIES_URL, \
    BESHARED_LIBRARIES_URL, SHARED_FILES_URL, F_URL, S_F_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 DirSharedItemsTest(TestCase, Fixtures):
    def setUp(self):
        self.folder_path = self.folder
        sub_repo_id = seafile_api.create_virtual_repo(self.repo.id,
                                                      self.folder_path,
                                                      self.repo.name, '',
                                                      self.user.username)
        # A user shares a folder to admin with permission 'rw.
        seafile_api.share_repo(sub_repo_id, self.user.username,
                               self.admin.username, 'rw')

    def tearDown(self):
        self.remove_repo()

    def _login_as(self, user):
        self.client.post(
            reverse('auth_login'), {'username': self.user.username,
                                    'password': 'secret'}
        )

    def test_can_list_all(self):
        self._login_as(self.user)

        resp = self.client.get('/api2/repos/%s/dir/shared_items/?p=%s&share_type=user,group' % (
            self.repo.id,
            self.folder_path))

        self.assertEqual(200, resp.status_code)
        json_resp = json.loads(resp.content)
        assert len(json_resp) == 1
        assert self.admin.username == json_resp[0]['user_info']['name']

    def test_can_list_without_share_type_arg(self):
        self._login_as(self.user)

        resp = self.client.get('/api2/repos/%s/dir/shared_items/?p=%s' % (
            self.repo.id,
            self.folder_path))

        self.assertEqual(200, resp.status_code)
        json_resp = json.loads(resp.content)
        assert len(json_resp) == 1
        assert self.admin.username == json_resp[0]['user_info']['name']

    def test_can_add(self):
        self._login_as(self.user)

        resp = self.client.put(
            '/api2/repos/%s/dir/shared_items/?p=%s' % (self.repo.id,
                                                       self.folder_path),
            "share_type=user&username=a@a.com&username=b@b.com",
            'application/x-www-form-urlencoded',
        )
        self.assertEqual(200, resp.status_code)
        json_resp = json.loads(resp.content)
        assert 'a@a.com' in json_resp['shared_success']
        assert 'b@b.com' in json_resp['shared_success']

    def test_can_update(self):
        self._login_as(self.user)

        resp = self.client.post('/api2/repos/%s/dir/shared_items/?p=%s' % (
            self.repo.id,
            self.folder_path), {

            }
        )
        print resp

    def test_can_delete(self):
        self._login_as(self.user)

        resp = self.client.delete('/api2/repos/%s/dir/shared_items/?p=%s&share_type=user&username=%s' % (
            self.repo.id,
            self.folder_path,
            self.admin.username
        ))
        self.assertEqual(200, resp.status_code)
        json_resp = json.loads(resp.content)
        assert json_resp[0]['success'] is True

        resp = self.client.get('/api2/repos/%s/dir/shared_items/?p=%s&share_type=user,group' % (
            self.repo.id,
            self.folder_path))

        json_resp = json.loads(resp.content)
        assert len(json_resp) == 0