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())
