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)
