def testCreateNewUser(self): c = Controller() c._getState().tree_view = State.UNILEVEL_TREE master = MasterUser.Get() user1Id = c.createNewUser(master) tree = UnilevelTree() tree.treeToJson(User.Get(user1Id).active_unilevel_position) user2Id = c.createNewUser(User.Get(user1Id)) tree.treeToJson(User.Get(user1Id).active_unilevel_position) tree.treeToJson(User.Get(user2Id).active_unilevel_position)
def testGetState(self): c = Controller() state = c._getState() self.assertEqual(state, State.objects.get(id=1))