[#28] fixed getAllModels, added unit tests

Author: Castronova

odm2api/ODM2/services/readService.py

@@ -908,7 +908,7 @@ def getDeploymentActionByCode(self, deploymentCode):
     def getAllModels(self):
 
         try:
-            return self._session.query(Model).all()
+            return self._session.query(Models).all()
         except:
             return None
 
@@ -927,7 +927,7 @@ def getAllSimulations(self):
 
     def getSimulationByName(self, simulationName):
         try:
-            return self._session.query(Simulation).filter(Simulation.SimulationName.ilike(simulationName)).first()
+            return self._session.query(Simulations).filter(Simulations.SimulationName.ilike(simulationName)).first()
         except:
             return None
 

tests/test_odm2/test_readservice.py

@@ -113,7 +113,15 @@ def test_getModelByCode(self):
     """
 
     def test_getRelatedModelsByID(self):
-        pass
+
+        # get one model from the database
+        res = self.engine.execute('SELECT * FROM Models').fetchone()
+        model_id = res[0]
+
+        # get related models by id using the api
+        resapi = self.reader.getRelatedModelsByID(model_id)
+
+        assert False
 
     def test_getRelatedModelsByCode(self):
         pass