130 lines
4.0 KiB
Python
130 lines
4.0 KiB
Python
|
|
import urllib
|
|
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):
|
|
|
|
def setUp(self):
|
|
fd, self.path = tempfile.mkstemp()
|
|
self.app = Isso({'SQLITE': self.path, 'PRODUCTION': False,
|
|
'MARKUP': 'isso.markup.Markup'})
|
|
|
|
self.client = Client(self.app, Response)
|
|
self.get = lambda *x, **z: self.client.get(*x, **z)
|
|
self.put = lambda *x, **z: self.client.put(*x, **z)
|
|
self.post = lambda *x, **z: self.client.post(*x, **z)
|
|
self.delete = lambda *x, **z: self.client.delete(*x, **z)
|
|
|
|
def testGet(self):
|
|
|
|
self.post('/1.0/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
|
|
r = self.get('/1.0/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('/1.0/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
|
|
|
|
assert rv.status_code == 201
|
|
assert len(filter(lambda header: header[0] == 'Set-Cookie', rv.headers)) == 1
|
|
|
|
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(20):
|
|
self.post('/1.0/path/new', data=json.dumps(comment(text='Spam')))
|
|
|
|
r = self.get('/1.0/path/')
|
|
assert r.status_code == 200
|
|
|
|
rv = json.loads(r.data)
|
|
assert len(rv) == 20
|
|
|
|
def testGetInvalid(self):
|
|
|
|
assert self.get('/1.0/path/123').status_code == 404
|
|
assert self.get('/1.0/path/spam/123').status_code == 404
|
|
assert self.get('/1.0/foo/').status_code == 404
|
|
|
|
def testUpdate(self):
|
|
|
|
self.post('/1.0/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
|
|
self.put('/1.0/path/1', data=json.dumps(comment(
|
|
text='Hello World', author='me', website='http://example.com/')))
|
|
|
|
r = self.get('/1.0/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('/1.0/path/new', data=json.dumps(comment(text='Lorem ipsum ...')))
|
|
r = self.delete('/1.0/path/1')
|
|
assert r.status_code == 200
|
|
assert json.loads(r.data) == None
|
|
assert self.get('/1.0/path/1').status_code == 404
|
|
|
|
def testDeleteWithReference(self):
|
|
|
|
client = Client(self.app, Response)
|
|
client.post('/1.0/path/new', data=json.dumps(comment(text='First')))
|
|
client.post('/1.0/path/new', data=json.dumps(comment(text='First', parent=1)))
|
|
|
|
r = client.delete('/1.0/path/1')
|
|
assert r.status_code == 200
|
|
assert Comment(**json.loads(r.data)).deleted
|
|
|
|
assert self.get('/1.0/path/1').status_code == 200
|
|
assert self.get('/1.0/path/2').status_code == 200
|
|
|
|
def testPathVariations(self):
|
|
|
|
paths = ['/sub/path/', '/path.html', '/sub/path.html', '%2Fpath/%2F', '/']
|
|
|
|
for path in paths:
|
|
assert self.post('/1.0/' + path + '/new',
|
|
data=json.dumps(comment(text='...'))).status_code == 201
|
|
|
|
for path in paths:
|
|
assert self.get('/1.0/' + path)
|
|
assert self.get('/1.0/' + path + '/1')
|
|
|
|
def testDeleteAndCreateByDifferentUsersButSamePostId(self):
|
|
|
|
mallory = Client(self.app, Response)
|
|
mallory.post('/1.0/path/new', data=json.dumps(comment(text='Foo')))
|
|
mallory.delete('/1.0/path/1')
|
|
|
|
bob = Client(self.app, Response)
|
|
bob.post('/1.0/path/new', data=json.dumps(comment(text='Bar')))
|
|
|
|
assert mallory.delete('/1.0/path/1').status_code == 403
|
|
assert bob.delete('/1.0/path/1').status_code == 200
|