112 lines
2.6 KiB
Python
112 lines
2.6 KiB
Python
|
|
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(self.path, False)
|
|
|
|
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].path == '/'
|
|
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].path == '/path/'
|
|
assert rv[0].text == 'Baz'
|
|
|
|
def test_add_return(self):
|
|
|
|
self.db.add('/', comment(text='1'))
|
|
self.db.add('/', comment(text='2'))
|
|
|
|
assert self.db.add('/path/', comment(text='1')).id == 1
|
|
|
|
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 test_recent(self):
|
|
|
|
self.db.add('/path/', comment(text='2'))
|
|
|
|
for x in range(5):
|
|
self.db.add('/', comment(text='%i' % (x+1)))
|
|
|
|
assert len(list(self.db.recent(mode=7))) == 6
|
|
assert len(list(self.db.recent(mode=7, limit=5))) == 5
|
|
|
|
def tearDown(self):
|
|
os.unlink(self.path)
|
|
|
|
|
|
class TestSQLitePending(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
fd, self.path = tempfile.mkstemp()
|
|
self.db = SQLite(self.path, 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)
|