isso/specs/test_comment.py

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)

    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

    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