import os import json import time import tempfile import unittest import isso from isso.models import Comment from isso.db import SQLite def comment(**kw): return Comment.fromjson(json.dumps(kw)) class TestSQLite(unittest.TestCase): def setUp(self): fd, self.path = tempfile.mkstemp() self.db = SQLite(isso.Isso({'SQLITE': self.path})) def test_get(self): rv = self.db.add('/', comment(text='Spam')) c = self.db.get('/', rv.id) assert c.id == 1 assert c.text == 'Spam' def test_add(self): self.db.add('/', comment(text='Foo')) self.db.add('/', comment(text='Bar')) self.db.add('/path/', comment(text='Baz')) rv = list(self.db.retrieve('/')) assert rv[0].id == 1 assert rv[0].text == 'Foo' assert rv[1].id == 2 assert rv[1].text == 'Bar' rv = list(self.db.retrieve('/path/')) assert rv[0].id == 1 assert rv[0].text == 'Baz' def test_update(self): rv = self.db.add('/', comment(text='Foo')) time.sleep(0.1) rv = self.db.update('/', rv.id, comment(text='Bla')) c = self.db.get('/', rv.id) assert c.id == 1 assert c.text == 'Bla' assert c.created < c.modified def test_delete(self): rv = self.db.add('/', comment( text='F**CK', author='P*NIS', website='http://somebadhost.org/')) assert self.db.delete('/', rv.id) == None def tearDown(self): os.unlink(self.path) class TestSQLitePending(unittest.TestCase): def setUp(self): fd, self.path = tempfile.mkstemp() self.db = SQLite(isso.Isso({'SQLITE': self.path, 'MODERATION': True})) def test_retrieve(self): self.db.add('/', comment(text='Foo')) assert len(list(self.db.retrieve('/'))) == 0 def test_activate(self): self.db.add('/', comment(text='Foo')) self.db.add('/', comment(text='Bar')) self.db.activate('/', 2) assert len(list(self.db.retrieve('/'))) == 1 assert len(list(self.db.retrieve('/', mode=3))) == 2 def tearDown(self): os.unlink(self.path)