| 293 | self.assertEqual(list(iter(database)), [t1, t2]) |
| 294 | |
| 295 | def test_add(self) -> None: |
| 296 | t1 = Table('table1') |
| 297 | t2 = Table('table2') |
| 298 | tg = TableGroup('mytablegroup', [t1, t2]) |
| 299 | e = Enum('myenum', [EnumItem('a'), EnumItem('b')]) |
| 300 | database = Database() |
| 301 | database.add(t1) |
| 302 | database.add(t2) |
| 303 | database.add(e) |
| 304 | database.add(tg) |
| 305 | self.assertIs(t1.database, database) |
| 306 | self.assertIs(t2.database, database) |
| 307 | self.assertIs(e.database, database) |
| 308 | self.assertIs(tg.database, database) |
| 309 | self.assertIn(t1, database.tables) |
| 310 | self.assertIn(t2, database.tables) |
| 311 | self.assertIn(tg, database.table_groups) |
| 312 | self.assertIn(e, database.enums) |
| 313 | |
| 314 | def test_add_bad(self) -> None: |
| 315 | class Test: |