def test_mark_recent(self): for idx in range(200): row = db.execute('INSERT INTO tracks (title) VALUES (?)', (str(idx), )) db.execute("INSERT INTO labels (track_id, label, email) VALUES (?, 'music', 'test')", (row, )) db.Open().mark_recent_music() new = db.fetchone("SELECT COUNT(*) FROM tracks WHERE id IN (SELECT track_id FROM labels WHERE label = 'recent')")[0] old = db.fetchone("SELECT COUNT(*) FROM tracks WHERE id NOT IN (SELECT track_id FROM labels WHERE label = 'recent')")[0] self.assertEquals(200, new + old) self.assertEquals(100, new)
def test_update(self): db.execute('DELETE FROM queue') row = db.execute('INSERT INTO queue (track_id, owner) VALUES (?, ?)', (1, 'ardj', )) self.assertEquals(row, 1) db.Open().update('queue', {'owner': 'test', 'id': row}) tmp = db.fetchone('SELECT * FROM queue') self.assertEquals(tmp, (1, 1, 'test'))
def test_queue(self): db.execute('INSERT INTO queue (track_id, owner) VALUES (?, ?)', (0, 'test', )) self.assertEquals(1, db.fetchone('SELECT COUNT(*) FROM queue')[0], 'Failed to insert a record into queue.')