def test_getHouseStructure(self): db = MockDatabase() h = House(db) room = MockRoom(1, "lounge") h.rooms = {1: room} self.assertEqual(h.getStructure(), {'rooms': [{ 'id': 1, 'name': "lounge", 'items': [] }]})
def test_getHouseStructure_emptylist(self): db = MockDatabase() h = House(db) h.rooms = {} self.assertEqual(h.getStructure(), {'rooms': []})
def test_getHouseStructure(self): db = MockDatabase() h = House(db) room = MockRoom(1, "lounge") h.rooms = {1: room} self.assertEqual(h.getStructure(), {'rooms': [{'id': 1, 'name': "lounge", 'items': []}]})