sjSYwgSq

1from django.test import TestCase
2from secretpost.models import Secret
3from secretpost.views import secretPost,secretGet
4from django.http import HttpRequest
5import asyncio
6from django.test import Client
7c = Client()
8
9
10class secretTestCase(TestCase):
11    def test_object(self):
12        obj = Secret(text = b'privet', secret_key = 'medved')
13        obj.save()
14        obj = Secret.objects.filter(text = b'privet')[0]
15        self.assertEqual(obj.secret_key, 'medved')
16
17
18    def test_endpoints(self):
19        payload = {'text':'cool', 'key':'some key goes here'}
20        resp = c.get('/secretpost/', payload)
21        self.assertEqual(resp.content, b'request is not POST')
22        resp = c.post('/secretpost/', {})
23        self.assertEqual(resp.content, b'payload is invalid, please provide "text" and "key"')
24        resp = c.post('/secretpost/', payload)
25        self.assertEqual(len(resp.content),16)
26        link = '/secretpost/' + str(resp.content.decode('utf-8'))
27        resp = c.get(link, {'key':payload['key']})
28        self.assertEqual(resp.content.decode('utf-8'), payload['text'])
29
30
31
32