def test_import_vouchers(self): pool = VoucherPool('testpool', self.conn) self.successResultOf(pool.create_tables()) self.successResultOf(pool.import_vouchers([ {'operator': 'Tank', 'denomination': 'red', 'voucher': 'Tr0'}, {'operator': 'Tank', 'denomination': 'red', 'voucher': 'Tr1'}, {'operator': 'Tank', 'denomination': 'blue', 'voucher': 'Tb0'}, {'operator': 'Tank', 'denomination': 'blue', 'voucher': 'Tb1'}, {'operator': 'Link', 'denomination': 'red', 'voucher': 'Lr0'}, {'operator': 'Link', 'denomination': 'red', 'voucher': 'Lr1'}, {'operator': 'Link', 'denomination': 'blue', 'voucher': 'Lb0'}, {'operator': 'Link', 'denomination': 'blue', 'voucher': 'Lb1'}, ])) result = self.successResultOf(pool.count_vouchers()) assert sorted(tuple(r) for r in result) == [ ('Link', 'blue', False, 2), ('Link', 'red', False, 2), ('Tank', 'blue', False, 2), ('Tank', 'red', False, 2), ]
def test_exists(self): pool = VoucherPool('testpool', self.conn) assert not self.successResultOf(pool.exists()) self.successResultOf(pool.create_tables()) assert self.successResultOf(pool.exists())