isso/specs/test_comment.py

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('/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
        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('/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

    def testGetInvalid(self):

        assert self.get('/comment/path/123').status_code == 404
        assert self.get('/comment/path/spam/123').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
        assert json.loads(r.data) == None
        assert self.get('/comment/path/1').status_code == 404

    def testDeleteWithReference(self):

        client = Client(self.app, Response)
        resp = client.post('/comment/path/new', data=json.dumps(comment(text='First')))
        self.post('/comment/path/new', data=json.dumps(comment(text='Second', parent=1)))

        r = client.delete('/comment/path/1')
        assert r.status_code == 200
        assert Comment(**json.loads(r.data)).deleted

        assert self.get('/comment/path/1').status_code == 200
        assert self.get('/comment/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('/comment/' + path + '/new',
                             data=json.dumps(comment(text='...'))).status_code == 201

        for path in paths:
            assert self.get('/comment/' + path)
            assert self.get('/comment/' + path + '/1')

    def testDeleteAndCreateByDifferentUsersButSamePostId(self):

        mallory = Client(self.app, Response)
        mallory.post('/comment/path/new', data=json.dumps(comment(text='Foo')))
        mallory.delete('/comment/path/1')

        bob = Client(self.app, Response)
        bob.post('/comment/path/new', data=json.dumps(comment(text='Bar')))

        assert mallory.delete('/comment/path/1').status_code == 403
        assert bob.delete('/comment/path/1').status_code == 200