| 322 | t.database |
| 323 | |
| 324 | def test_delete(self) -> None: |
| 325 | t1 = Table('table1') |
| 326 | c1 = Column('col1', 'int') |
| 327 | t1.add_column(c1) |
| 328 | t2 = Table('table2') |
| 329 | c2 = Column('col2', 'int') |
| 330 | t2.add_column(c2) |
| 331 | ref = Reference('>', [c1], [c2]) |
| 332 | tg = TableGroup('mytablegroup', [t1, t2]) |
| 333 | e = Enum('myenum', [EnumItem('a'), EnumItem('b')]) |
| 334 | p = Project('myproject') |
| 335 | database = Database() |
| 336 | database.add(t1) |
| 337 | database.add(t2) |
| 338 | database.add(e) |
| 339 | database.add(tg) |
| 340 | database.add(ref) |
| 341 | database.add(p) |
| 342 | |
| 343 | database.delete(t1) |
| 344 | database.delete(t2) |
| 345 | database.delete(e) |
| 346 | database.delete(tg) |
| 347 | database.delete(ref) |
| 348 | database.delete(p) |
| 349 | self.assertIsNone(t1.database) |
| 350 | self.assertIsNone(t2.database) |
| 351 | self.assertIsNone(e.database) |
| 352 | self.assertIsNone(tg.database) |
| 353 | self.assertIsNone(ref.database) |
| 354 | self.assertIsNone(p.database) |
| 355 | self.assertIsNone(database.project) |
| 356 | self.assertNotIn(t1, database.tables) |
| 357 | self.assertNotIn(t2, database.tables) |
| 358 | self.assertNotIn(tg, database.table_groups) |
| 359 | self.assertNotIn(e, database.enums) |
| 360 | self.assertNotIn(ref, database.refs) |
| 361 | |
| 362 | def test_delete_bad(self) -> None: |
| 363 | class Test: |