class MembershipTestCase(TestCase):
def setUp(self):
self.organization = Organization()
self.organization.save()
self.actor = Actor()
self.actor.save()
self.membership = Membership()
self.membership.organization = self.organization
self.membership.actor = self.actor
self.membership.save()
def tearDown(self):
self.actor.delete()
self.organization.delete()
def test__actor__returns_actor(self):
self.assertEquals(self.membership.actor_id, self.actor.id)
self.assertEquals(self.membership.actor.id, self.actor.id)
self.assertEquals(self.organization.actors.get().id, self.actor.id)
def test__organization__returns_organization(self):
self.assertEquals(self.membership.organization_id,
self.organization.id)
self.assertEquals(self.membership.organization.id,
self.organization.id)
self.assertEquals(self.actor.organizations.get().id,
self.organization.id)
class OrganizationsAPITestCase(APITestCase):
def setUp(self):
self.user = self.create_user()
self.organization1 = Organization(name="tk1")
self.organization1.save()
self.organization2 = Organization(name="tk2")
self.organization2.save()
self.organization3 = Organization(name="tk3")
self.organization3.save()
self.actor = Actor(user_id=self.user.id)
self.actor.save()
self.membership1 = Membership(organization=self.organization1,
actor=self.actor)
self.membership1.save()
self.membership2 = Membership(organization=self.organization2,
actor=self.actor)
self.membership2.save()
def create_user(self):
user = User.objects.create_user(username="******",
email="*****@*****.**",
password="******")
self.client.login(username='******', password='******')
return user
def tearDown(self):
self.actor.delete()
self.organization1.delete()
self.organization2.delete()
self.user.delete()
def test__list_organizations__lists_all_organizations_user_belongs(self):
response = self.client.get(reverse('organizations-list'))
self.assertEquals(response.status_code, status.HTTP_200_OK)
self.assertEquals(response.data, [{
'id': self.organization1.id,
'name': self.organization1.name
}, {
'id': self.organization2.id,
'name': self.organization2.name
}])
def test__list_organizations__filters_on_name(self):
url = reverse('organizations-list') + '?name=tk2'
response = self.client.get(url)
self.assertEquals(response.status_code, status.HTTP_200_OK)
self.assertEquals(response.data, [{
'id': self.organization2.id,
'name': self.organization2.name
}])
def test__get_organization__returns_org_if_user_belongs(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization1.id})
response = self.client.get(url)
self.assertEquals(response.status_code, status.HTTP_200_OK)
self.assertEquals(response.data, {
'id': self.organization1.id,
'name': self.organization1.name
})
def test__get_organization__returns_404_if_user_doesnt_belong(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization3.id})
response = self.client.get(url)
self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)
def test__create_organization__creates_organization(self):
url = reverse('organizations-list')
response = self.client.post(url, {'name': 'new org'})
self.assertEquals(response.status_code, status.HTTP_201_CREATED)
self.assertEquals(response.data['name'], 'new org')
def test__patch_organization__updates_name(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization1.id})
response = self.client.patch(url, {'name': 'name1'})
self.assertEquals(response.status_code, status.HTTP_200_OK)
self.assertEquals(response.data, {
'name': 'name1',
'id': self.organization1.id
})
def test__patch_organization__returns_404_if_user_doesnt_belong(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization3.id})
response = self.client.patch(url, {'name': 'name1'})
self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)
def test__put_organization__updates_name(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization1.id})
response = self.client.put(url, {'name': 'name1'})
self.assertEquals(response.status_code, status.HTTP_200_OK)
self.assertEquals(response.data, {
'name': 'name1',
'id': self.organization1.id
})
def test__put_organization__returns_404_if_user_doesnt_belong(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization3.id})
response = self.client.put(url, {'name': 'name1'})
self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)
def test__delete_organization__deletes_org(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization1.id})
response = self.client.delete(url)
self.assertEquals(response.status_code, status.HTTP_204_NO_CONTENT)
self.assertEquals(
Organization.objects.filter(pk=self.organization1.id).count(), 0)
def test__delete_organization__returns_404_if_user_doesnt_belong(self):
url = reverse('organizations-detail',
kwargs={'pk': self.organization3.id})
response = self.client.delete(url, {'name': 'name1'})
self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)