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