def test_query(self): invoice_id = db.insert_invoice('Cool Project', 20.0, 1.0) commit_id = db.insert_commit('Did a thing', 1405184155, 2.0, invoice_id) self.assertEqual(db.query_user(1), (0.0, 0.25, 0, 0, 0, 0)) self.assertEqual(db.query_invoice(1), ('Cool Project', 20.0, 1.0)) self.assertEqual(db.query_invoice('Cool Project'), ('Cool Project', 20.0, 1.0, 1)) self.assertEqual(db.query_commit(1), ('Did a thing', 1405184155, 2.0, invoice_id)) self.assertEqual(db.query_invoice_commit_meta(1), [('Did a thing', 1405184155, 2.0, invoice_id)])
def __init__(self, rowid=1): meta = db.query_user(rowid) if not meta: print("User with rowid %s doesn't exist." %rowid, file=sys.stderr) sys.exit() self.rowid = rowid self.rate = meta[0] self.rounding = meta[1] self.timer_running = bool(meta[2]) self.timer_start = meta[3] self.timer_total = meta[4] self.active_invoice_rowid = meta[5] if len(meta) == 6 else 0