def setUp(self):
self.user_dict = {
'username': '******',
'email': '*****@*****.**',
'password': '******'
}
self.user = User.objects.create_user(**self.user_dict)
self.scene_dict = {
'title': 'main scene',
'content': '<xml></xml>',
'owner': self.user
}
self.scene = Scene(**self.scene_dict)
self.scene.save()
class SceneTest(APITestCase):
def setUp(self):
self.user_dict = {
'username': '******',
'email': '*****@*****.**',
'password': '******'
}
self.user = User.objects.create_user(**self.user_dict)
self.scene_dict = {
'title': 'main scene',
'content': '<xml></xml>',
'owner': self.user
}
self.scene = Scene(**self.scene_dict)
self.scene.save()
def tearDown(self):
pass
def test_get_scene(self):
self.client.force_authenticate(self.user)
url = reverse('scene-detail', kwargs={'pk': self.scene.id})
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(dict(response.data),
{'url': 'http://testserver/scenes/{}'.
format(self.scene.id),
'title': self.scene_dict['title'],
'content': self.scene_dict['content'],
'owner': 'http://testserver' +
reverse('user-detail', kwargs={'pk': self.user.id})})
def test_get_scenes(self):
self.client.force_authenticate(self.user)
url = reverse('scene-list')
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual([dict(d) for d in response.data],
[{'url': 'http://testserver/scenes/{}'.
format(self.scene.id),
'title': self.scene_dict['title'],
'content': self.scene_dict['content'],
'owner': 'http://testserver' +
reverse('user-detail',
kwargs={'pk': self.user.id})}])
def test_get_scenes_unauthenticated(self):
url = reverse('scene-list')
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)
def test_get_scene_unauthenticated(self):
url = reverse('scene-detail', kwargs={'pk': self.scene.id})
response = self.client.get(url)
self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)
def test_create_scene_unathorized(self):
scene_dict = {
'title': 'other scene',
'content': '<xml></xml>',
'owner': self.user.username
}
url = reverse('scene-list')
response = self.client.post(url, scene_dict, format='json')
self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)
def test_create_scene(self):
self.client.force_authenticate(self.user)
scene_dict = {
'title': 'other scene',
'content': '<xml></xml>',
'owner': reverse('user-detail', kwargs={'pk': self.user.id})
}
url = reverse('scene-list')
response = self.client.post(url, scene_dict, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(
Scene.objects.filter(pk=response.data['url'][-1]).
first().title, scene_dict['title'])
def test_delete_scene(self):
self.client.force_authenticate(self.user)
# test scene existence
self.assertTrue(Scene.objects.filter(pk=self.scene.id).exists())
# delete scene
url = reverse('scene-detail', kwargs={'pk': self.scene.id})
print(url)
response = self.client.delete(url)
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
# test scene non-existence
self.assertFalse(Scene.objects.filter(pk=self.scene.id).exists())
def test_modify_scene(self):
self.client.force_authenticate(self.user)
scene_dict = {
'title': 'different scene title',
'content': '<xml>different xml</xml>',
}
scene = Scene.objects.filter(pk=self.scene.id).first()
self.assertEqual(scene.title, self.scene_dict['title'])
self.assertEqual(scene.content, self.scene_dict['content'])
scene.title = scene_dict['title']
scene.content = scene_dict['content']
scene.save()
scene = Scene.objects.filter(pk=self.scene.id).first()
self.assertEqual(scene.title, scene_dict['title'])
self.assertEqual(scene.content, scene_dict['content'])
