class TestMapElement(unittest.TestCase):
def setUp(self):
self.mox = mox.Mox()
self.elem = MapElement(1, 2, 3, 4, 'a', 5, 6)
def tearDown(self):
pass
def testPut(self):
self.mox.StubOutWithMock(map_element.db.Model, 'put')
map_element.db.Model.put()
self.mox.ReplayAll()
self.assertEqual(self.elem._children_save, None, 'wrong initiate')
self.elem.put()
self.assertEqual(
self.elem._children_save,
'{{,,,,},{,,,,},{,,,,},{,,,,},{,,,,},{,,,,}}',
self.elem._children_save)
self.mox.VerifyAll()
def testGetByKeyName(self):
self.elem._children_save = '{1,2,3}'
self.elem._children = None
self.mox.StubOutWithMock(map_element.db.Model, 'get')
map_element.db.Model.get('1').AndReturn(self.elem)
self.mox.ReplayAll()
elem = MapElement.get('1')
self.assertEqual(elem._children, ['1', '2', '3'], elem._children)
self.mox.VerifyAll()
def testGetByKeyName(self):
self.elem._children_save = '{1,2,3}'
self.elem._children = None
self.mox.StubOutWithMock(map_element.db.Model, 'get')
map_element.db.Model.get('1').AndReturn(self.elem)
self.mox.ReplayAll()
elem = MapElement.get('1')
self.assertEqual(elem._children, ['1', '2', '3'], elem._children)
self.mox.VerifyAll()
def setUp(self): self.mox = mox.Mox() self.elem = MapElement(1, 2, 3, 4, 'a', 5, 6)
