def populate_pool(self, pool_name, operators, denominations, suffixes):
pool = VoucherPool(pool_name, self.conn)
return pool.import_vouchers([
{
'operator': operator,
'denomination': denomination,
'voucher': '%s-%s-%s' % (operator, denomination, suffix),
}
for operator in operators
for denomination in denominations
for suffix in suffixes
])
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),
]
