Python OrganizationProfileViewSet.as_view方法代码示例

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_orgs_non_creator_delete(self):
        self._org_create()

        view = OrganizationProfileViewSet.as_view({
            'delete': 'members',
            'post': 'members'
        })

        self.profile_data['username'] = "alice"
        self.profile_data['email'] = '[email protected]'
        self._create_user_profile()

        alice_data = {'username': 'alice', 'email': '[email protected]'}
        request = self.factory.post(
            '/', data=json.dumps(alice_data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        expected_results = [u'denoinc', u'alice']
        self.assertEqual(status.HTTP_201_CREATED, response.status_code)
        self.assertEqual(expected_results, response.data)

        self._login_user_and_profile(extra_post_data=alice_data)

        request = self.factory.delete('/', data=json.dumps(alice_data),
                                      content_type="application/json",
                                      **self.extra)
        response = view(request, user='denoinc')
        expected_results = [u'denoinc']
        self.assertEqual(expected_results, response.data)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_put_bad_role(self):
        self._org_create()
        newname = 'aboy'
        view = OrganizationProfileViewSet.as_view({
            'get': 'retrieve',
            'post': 'members',
            'put': 'members'
        })

        User.objects.create(username=newname)
        data = {'username': newname}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)
        self.assertEqual(set(response.data), set([u'denoinc', newname]))

        data = {'username': newname, 'role': 42}
        request = self.factory.put(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 400)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
 def setUp(self):
     super(self.__class__, self).setUp()
     self.view = OrganizationProfileViewSet.as_view({
         'get': 'list',
         'post': 'create',
         'patch': 'partial_update',
     })

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_role_for_org_non_owner(self):
        # creating org with member
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'get': 'retrieve',
            'post': 'members'
        })

        self.profile_data['username'] = "aboy"
        self._create_user_profile()
        data = {'username': 'aboy'}
        user_role = 'member'
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)
        response = view(request, user='denoinc')

        # getting profile
        request = self.factory.get('/', **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 200)
        self.assertIn('users', response.data.keys())

        for user in response.data['users']:
            username = user['user']
            role = user['role']
            expected_role = 'owner' if username == 'denoinc' else user_role
            self.assertEqual(role, expected_role)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_remove_members_from_org(self):
        self._org_create()
        newname = 'aboy'
        view = OrganizationProfileViewSet.as_view({
            'post': 'members',
            'delete': 'members'
        })

        User.objects.create(username=newname)
        data = {'username': newname}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)
        self.assertEqual(set(response.data), set([u'denoinc', newname]))

        request = self.factory.delete(
            '/', json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)
        self.assertEqual(response.data, [u'denoinc'])

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_member_sees_orgs_added_to(self):
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'get': 'list',
            'post': 'members'
        })

        member = 'aboy'
        expected_data = self.company_data
        expected_data['users'].append({'role': 'member', 'user': member})
        cur_username = self.profile_data['username']
        self.profile_data['username'] = member
        self._login_user_and_profile()
        self.profile_data['username'] = cur_username
        self._login_user_and_profile()

        data = {'username': member}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)
        self.assertEqual(response.data, [u'denoinc', u'aboy'])

        self.profile_data['username'] = member
        self._login_user_and_profile()

        request = self.factory.get('/', **self.extra)
        response = view(request)
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data, [expected_data])

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_add_members_to_org_with_role(self):
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'post': 'members',
            'get': 'retrieve'
        })

        self.profile_data['username'] = "aboy"
        self._create_user_profile()
        data = {'username': 'aboy',
                'role': 'editor'}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)

        self.assertEqual(set(response.data), set([u'denoinc', u'aboy']))

        # getting profile
        request = self.factory.get('/', **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data['users'][1]['user'], 'aboy')
        self.assertEqual(response.data['users'][1]['role'], 'editor')

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_org_members_added_to_projects(self):
        # create org
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'post': 'members',
            'get': 'retrieve',
            'put': 'members'
        })

        # create aboy
        self.profile_data['username'] = "aboy"
        aboy = self._create_user_profile().user

        data = {'username': 'aboy',
                'role': 'owner'}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)

        # create a proj
        project_data = {
            'owner': self.company_data['user']
        }
        self._project_create(project_data)
        self._publish_xls_form_to_project()

        # create alice
        self.profile_data['username'] = "alice"
        alice = self._create_user_profile().user
        alice_data = {'username': 'alice',
                      'role': 'owner'}
        request = self.factory.post(
            '/', data=json.dumps(alice_data),
            content_type="application/json", **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 201)

        # Assert that user added in org is added to teams in proj
        self.assertTrue(OwnerRole.user_has_role(aboy, self.project))
        self.assertTrue(OwnerRole.user_has_role(alice, self.project))
        self.assertTrue(OwnerRole.user_has_role(aboy, self.xform))
        self.assertTrue(OwnerRole.user_has_role(alice, self.xform))

        # Org admins are added to owners in project
        projectView = ProjectViewSet.as_view({
            'get': 'retrieve'
        })
        request = self.factory.get('/', **self.extra)
        response = projectView(request, pk=self.project.pk)
        project_users = response.data.get('users')
        users_in_users = [user['user'] for user in project_users]

        self.assertIn('bob', users_in_users)
        self.assertIn('denoinc', users_in_users)
        self.assertIn('aboy', users_in_users)
        self.assertIn('alice', users_in_users)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_orgs_members_list(self):
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'get': 'members'
        })

        request = self.factory.get('/', **self.extra)
        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data, [u'denoinc'])

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
 def test_add_members_to_org_username_required(self):
     self._org_create()
     view = OrganizationProfileViewSet.as_view({
         'post': 'members'
     })
     request = self.factory.post('/', data={}, **self.extra)
     response = view(request, user='denoinc')
     self.assertEqual(response.status_code, 400)
     self.assertEqual(response.data,
                      {u'username': [u"This field is required."]})

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
 def test_orgs_get_anon(self):
     self._org_create()
     view = OrganizationProfileViewSet.as_view({
         'get': 'retrieve'
     })
     request = self.factory.get('/')
     response = view(request, user='denoinc')
     self.assertNotEqual(response.get('Last-Modified'), None)
     self.assertEqual(response.status_code, 200)
     self.assertEqual(response.data, self.company_data)
     self.assertIn('users', response.data.keys())
     for user in response.data['users']:
         self.assertEqual(user['role'], 'owner')
         self.assertEqual(type(user['user']), unicode)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_add_members_to_org_user_does_not_exist(self):
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'post': 'members'
        })
        data = {'username': 'aboy'}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json", **self.extra)

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 400)
        self.assertEqual(response.data,
                         {u'username': [u"User `aboy` does not exist."]})

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
 def test_orgs_get(self):
     self._org_create()
     view = OrganizationProfileViewSet.as_view({
         'get': 'retrieve'
     })
     request = self.factory.get('/', **self.extra)
     response = view(request)
     self.assertEqual(response.status_code, 400)
     self.assertEqual(
         response.data, {'detail': 'Expected URL keyword argument `user`.'})
     request = self.factory.get('/', **self.extra)
     response = view(request, user='denoinc')
     self.assertEqual(response.status_code, 200)
     self.assertEqual(response.data, self.company_data)

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
 def test_orgs_get_anon(self):
     self._org_create()
     view = OrganizationProfileViewSet.as_view({
         'get': 'retrieve'
     })
     request = self.factory.get('/')
     response = view(request, user='denoinc')
     self.assertNotEqual(response.get('Cache-Control'), None)
     self.assertEqual(response.status_code, 200)
     del self.company_data['metadata']
     self.assertEqual(response.data, self.company_data)
     self.assertIn('users', list(response.data))
     for user in response.data['users']:
         self.assertEqual(user['role'], 'owner')
         self.assertTrue(isinstance(user['user'], text))

# 需要导入模块: from onadata.apps.api.viewsets.organization_profile_viewset import OrganizationProfileViewSet [as 别名]
# 或者: from onadata.apps.api.viewsets.organization_profile_viewset.OrganizationProfileViewSet import as_view [as 别名]
    def test_add_members_to_org_with_anonymous_user(self):
        self._org_create()
        view = OrganizationProfileViewSet.as_view({
            'post': 'members'
        })

        User.objects.create(username='aboy')
        data = {'username': 'aboy'}
        request = self.factory.post(
            '/', data=json.dumps(data),
            content_type="application/json")

        response = view(request, user='denoinc')
        self.assertEqual(response.status_code, 401)
        self.assertNotEquals(set(response.data), set([u'denoinc', u'aboy']))