isso/specs/test_db.py

68 lines
1.5 KiB
Python
Raw Normal View History

2012-10-16 17:11:59 +00:00
2012-10-16 17:32:55 +00:00
import os
import json
2012-10-16 19:04:20 +00:00
import time
2012-10-16 17:11:59 +00:00
import tempfile
import unittest
2012-10-17 16:32:53 +00:00
import isso
2012-10-16 19:07:29 +00:00
from isso.models import Comment
2012-10-16 17:11:59 +00:00
from isso.db import SQLite
def comment(**kw):
return Comment.fromjson(json.dumps(kw))
2012-10-16 17:11:59 +00:00
class TestSQLite(unittest.TestCase):
def setUp(self):
2012-10-16 17:32:55 +00:00
fd, self.path = tempfile.mkstemp()
2012-10-17 16:32:53 +00:00
self.db = SQLite(isso.Isso({'SQLITE': self.path}))
2012-10-16 17:11:59 +00:00
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'
2012-10-16 17:11:59 +00:00
def test_add(self):
self.db.add('/', comment(text='Foo'))
self.db.add('/', comment(text='Bar'))
self.db.add('/path/', comment(text='Baz'))
2012-10-16 17:11:59 +00:00
rv = list(self.db.retrieve('/'))
assert rv[0].id == 2
assert rv[0].text == 'Bar'
assert rv[1].id == 1
assert rv[1].text == 'Foo'
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'))
2012-10-16 19:04:20 +00:00
time.sleep(0.1)
rv = self.db.update('/', rv.id, comment(text='Bla'))
c = self.db.get('/', rv.id)
assert c.id == 1
2012-10-17 08:56:43 +00:00
assert c.text == 'Bla'
2012-10-16 19:04:20 +00:00
assert c.created < c.modified
def test_delete(self):
rv = self.db.add('/', comment(
text='F**CK', author='P*NIS', website='http://somebadhost.org/'))
2012-10-17 10:22:52 +00:00
assert self.db.delete('/', rv.id) == None
2012-10-16 17:11:59 +00:00
def tearDown(self):
2012-10-16 17:32:55 +00:00
os.unlink(self.path)