def testDelNode(self): my_mox = mox.Mox() mock_node = my_mox.CreateMock(MapDataNode) my_map_data = MapData('map_id_0') my_map_data.nodes = ['key'] my_mox.StubOutWithMock(map_data_node.db.Model, 'get') map_data_node.db.Model.get('key').AndReturn(mock_node) mock_node.del_all_connect() mock_node.delete() my_mox.ReplayAll() my_map_data.del_node(key='key') self.assertEqual(my_map_data.nodes, [], my_map_data.nodes)
def testAddNode(self): my_mox = mox.Mox() my_mox.StubOutWithMock(map_data_node.db.Model, 'key') my_mox.StubOutWithMock(map_data_node.db.Model, '__init__') map_data_node.db.Model.key().MultipleTimes('1').AndReturn('key') map_data_node.db.Model.__init__( parent=IsA(map_data.MapData)) my_map_data = MapData('map_id_0') my_map_data.nodes = [] my_mox.ReplayAll() my_map_data.add_node() self.assertEqual(my_map_data.nodes, ['key'], my_map_data.nodes)