class TestBasePool(TestCase):
def setUp(self):
self.table = BaseTable('Test table')
self.pool = FakePool()
def test_should_set_and_get_like_a_dict(self):
self.pool[self.table.name] = self.table
self.assertEqual(self.pool.get(self.table.name), self.table)
def test_should_add_multiple_tables(self):
self.pool[self.table.name] = self.table
self.pool['Test table2'] = self.table
self.pool['Test table3'] = self.table
self.assertEqual(self.pool.get(self.table.name), self.table)
self.assertEqual(self.pool.get('Test table2'), self.table)
self.assertEqual(self.pool.get('Test table3'), self.table)
self.assertEqual(len(self.pool), 3)
def setUp(self):
self.table = BaseTable('Test table')
self.pool = FakePool()
