def test_add():
q = Quotes(':memory:', connection)
q.add("test_id", "author", "text")
c = connection.cursor()
c.execute(
'SELECT * from quotes WHERE chat_id=? and author = ? and quote = ?',
('test_id', 'author', 'text'))
assert c.fetchone() is not None
return "Quotes adding works"
def test_search():
q = Quotes(':memory:', connection)
q.add('test', 'antani', 'cacca')
q.add('test', 'antani', 'culo')
q.add('no', 'antani', 'cacca')
res = q.search('test', 'cacca')
assert res[0][2] == 'antani'
assert len(res) == 1
def test_random():
q = Quotes(':memory:', connection)
q.add('a', 'b', 'c')
assert q.random('a') is not None
