root / userdata / tests.py @ eee0487e
History | View | Annotate | Download (1.6 kB)
1 | eee0487e | Kostas Papadimitriou | """
|
---|---|---|---|
2 | eee0487e | Kostas Papadimitriou | This file demonstrates two different styles of tests (one doctest and one
|
3 | eee0487e | Kostas Papadimitriou | unittest). These will both pass when you run "manage.py test".
|
4 | eee0487e | Kostas Papadimitriou |
|
5 | eee0487e | Kostas Papadimitriou | Replace these with more appropriate tests for your application.
|
6 | eee0487e | Kostas Papadimitriou | """
|
7 | eee0487e | Kostas Papadimitriou | |
8 | eee0487e | Kostas Papadimitriou | from django.test import TestCase |
9 | eee0487e | Kostas Papadimitriou | from django.conf import settings |
10 | eee0487e | Kostas Papadimitriou | from django.test.client import Client |
11 | eee0487e | Kostas Papadimitriou | from django.core.urlresolvers import clear_url_caches |
12 | eee0487e | Kostas Papadimitriou | |
13 | eee0487e | Kostas Papadimitriou | from synnefo.userdata.models import User |
14 | eee0487e | Kostas Papadimitriou | from synnefo.userdata.models import * |
15 | eee0487e | Kostas Papadimitriou | |
16 | eee0487e | Kostas Papadimitriou | class AaiClient(Client): |
17 | eee0487e | Kostas Papadimitriou | |
18 | eee0487e | Kostas Papadimitriou | def request(self, **request): |
19 | eee0487e | Kostas Papadimitriou | request['HTTP_X_AUTH_TOKEN'] = '46e427d657b20defe352804f0eb6f8a2' |
20 | eee0487e | Kostas Papadimitriou | return super(AaiClient, self).request(**request) |
21 | eee0487e | Kostas Papadimitriou | |
22 | eee0487e | Kostas Papadimitriou | class TestRestViews(TestCase): |
23 | eee0487e | Kostas Papadimitriou | |
24 | eee0487e | Kostas Papadimitriou | fixtures = ['users']
|
25 | eee0487e | Kostas Papadimitriou | |
26 | eee0487e | Kostas Papadimitriou | def setUp(self): |
27 | eee0487e | Kostas Papadimitriou | settings.ROOT_URLCONF = 'synnefo.userdata.urls'
|
28 | eee0487e | Kostas Papadimitriou | clear_url_caches() |
29 | eee0487e | Kostas Papadimitriou | self.client = AaiClient()
|
30 | eee0487e | Kostas Papadimitriou | self.user = User.objects.get(pk=1) |
31 | eee0487e | Kostas Papadimitriou | |
32 | eee0487e | Kostas Papadimitriou | def test_keys_collection_get(self): |
33 | eee0487e | Kostas Papadimitriou | resp = self.client.get("/keys/") |
34 | eee0487e | Kostas Papadimitriou | self.assertEqual(resp.content, "[]") |
35 | eee0487e | Kostas Papadimitriou | |
36 | eee0487e | Kostas Papadimitriou | PublicKeyPair.objects.create(user=self.user, name="key pair 1", |
37 | eee0487e | Kostas Papadimitriou | content="content1")
|
38 | eee0487e | Kostas Papadimitriou | |
39 | eee0487e | Kostas Papadimitriou | resp = self.client.get("/keys/") |
40 | eee0487e | Kostas Papadimitriou | self.assertEqual(resp.content, """[{"content": "content1", "id": 1, "name": "key pair 1"}]""") |
41 | eee0487e | Kostas Papadimitriou | |
42 | eee0487e | Kostas Papadimitriou | PublicKeyPair.objects.create(user=self.user, name="key pair 2", |
43 | eee0487e | Kostas Papadimitriou | content="content2")
|
44 | eee0487e | Kostas Papadimitriou | |
45 | eee0487e | Kostas Papadimitriou | resp = self.client.get("/keys/") |
46 | eee0487e | Kostas Papadimitriou | self.assertEqual(resp.content, """[{"content": "content1", "id": 1, "name": "key pair 1"}, {"content": "content2", "id": 2, "name": "key pair 2"}]""") |
47 | eee0487e | Kostas Papadimitriou | |
48 | eee0487e | Kostas Papadimitriou | def test_keys_resourse_get(self): |
49 | eee0487e | Kostas Papadimitriou | pass |