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