isso/specs/test_comment.py
2012-10-17 11:06:26 +02:00

93 lines
2.6 KiB
Python

import tempfile
import unittest
from werkzeug.test import Client
from werkzeug.wrappers import Response
from isso import Isso, json
from isso.models import Comment
def comment(**kw):
return Comment.fromjson(json.dumps(kw))
class TestComments(unittest.TestCase):
get = lambda self, *x, **z: Client(self.app, Response).get(*x, **z)
put = lambda self, *x, **z: Client(self.app, Response).put(*x, **z)
post = lambda self, *x, **z: Client(self.app, Response).post(*x, **z)
delete = lambda self, *x, **z: Client(self.app, Response).delete(*x, **z)
def setUp(self):
fd, self.path = tempfile.mkstemp()
self.app = Isso({'SQLITE': self.path})
def testGet(self):
self.post('/comment/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
r = self.get('/comment/path/1')
assert r.status_code == 200
rv = json.loads(r.data)
assert rv['id'] == 1
assert rv['text'] == 'Lorem ipsum ...'
def testCreate(self):
rv = self.post('/comment/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
assert rv.status_code == 201
# XXX assert cookie
c = Comment.fromjson(rv.data)
assert not c.pending
assert not c.deleted
assert c.text == 'Lorem ipsum ...'
def testCreateAndGetMultiple(self):
for i in range(100):
self.post('/comment/path/new', data=json.dumps(comment(text='Spam')))
r = self.get('/comment/path/')
assert r.status_code == 200
rv = json.loads(r.data)
assert len(rv) == 20
# XXX limit=100
def testGetInvalid(self):
assert self.get('/comment/path/123').status_code == 404
assert self.get('/comment/path/spam').status_code == 404
assert self.get('/comment/foo/').status_code == 404
def testUpdate(self):
self.post('/comment/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
self.put('/comment/path/1', data=json.dumps(comment(
text='Hello World', author='me', website='http://example.com/')))
r = self.get('/comment/path/1')
assert r.status_code == 200
rv = json.loads(r.data)
assert rv['text'] == 'Hello World'
assert rv['author'] == 'me'
assert rv['website'] == 'http://example.com/'
assert 'modified' in rv
def testDelete(self):
self.post('/comment/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
r = self.delete('/comment/path/1')
assert r.status_code == 200
c = Comment(**json.loads(r.data))
assert c.deleted