modified publications endpoint and tests

Author: TeresasaZ

api/resources/gene_information.py

@@ -71,29 +71,37 @@ def get(self, species="", gene_id=""):
             return BARUtils.error_exit("There are no data found for the given gene")
 
 
-@gene_information.route("/gene_publications/<string:gene_id>")
+@gene_information.route("/gene_publications/<string:species>/<string:gene_id>")
 class GenePublications(Resource):
-    # @gene_information.param("species", _in="path", default="arabidopsis")
+    @gene_information.param("species", _in="path", default="arabidopsis")
     @gene_information.param("gene_id", _in="path", default="AT1G01010")
-    def get(self, gene_id=""):
+    def get(self, species="", gene_id=""):
         """This end point provides publications given a gene ID."""
         publications = []
 
         # Escape input
+        species = escape(species)
         gene_id = escape(gene_id)
 
-        # truncate
+        # truncate gene ID
         for i in range(len(gene_id)):
             if gene_id[i] == ".":
                 gene_id = gene_id[0:i]
                 break
 
-        if BARUtils.is_arabidopsis_gene_valid(gene_id):
-            rows = db.session.execute(db.select(EPlant2Publications).where(EPlant2Publications.gene == gene_id)).scalars().all()
-            for row in rows:
-                publications.append({"gene_id": row.gene, "author": row.author, "year": row.year, "journal": row.journal, "title": row.title, "pubmed": row.pubmed})
+        # Set the database and check if genes are valid
+        if species == "arabidopsis":
+            database = EPlant2Publications
+
+            if not BARUtils.is_arabidopsis_gene_valid(gene_id):
+                return BARUtils.error_exit("Invalid gene id"), 400
         else:
-            return BARUtils.error_exit("Invalid gene id"), 400
+            return BARUtils.error_exit("No data for the given species")
+
+        # Get data
+        rows = db.session.execute(db.select(database).where(database.gene == gene_id)).scalars().all()
+        for row in rows:
+            publications.append({"gene_id": row.gene, "author": row.author, "year": row.year, "journal": row.journal, "title": row.title, "pubmed": row.pubmed})
 
         # Return results if there are data
         if len(publications) > 0:

tests/resources/test_gene_information.py

@@ -46,7 +46,7 @@ def test_get_arabidopsis_gene_publications(self):
         :return:
         """
         # Valid data
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01020")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01020")
         expected = {
             "wasSuccessful": True,
             "data": [
@@ -70,7 +70,7 @@ def test_get_arabidopsis_gene_publications(self):
         }
         self.assertEqual(response.json, expected)
 
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01020.")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01020.")
         expected = {
             "wasSuccessful": True,
             "data": [
@@ -94,7 +94,7 @@ def test_get_arabidopsis_gene_publications(self):
         }
         self.assertEqual(response.json, expected)
 
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01020.0")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01020.0")
         expected = {
             "wasSuccessful": True,
             "data": [
@@ -118,7 +118,7 @@ def test_get_arabidopsis_gene_publications(self):
         }
         self.assertEqual(response.json, expected)
 
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01020.1")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01020.1")
         expected = {
             "wasSuccessful": True,
             "data": [
@@ -142,7 +142,7 @@ def test_get_arabidopsis_gene_publications(self):
         }
         self.assertEqual(response.json, expected)
 
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01020.12345")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01020.12345")
         expected = {
             "wasSuccessful": True,
             "data": [
@@ -167,25 +167,30 @@ def test_get_arabidopsis_gene_publications(self):
         self.assertEqual(response.json, expected)
 
         # Data not found, but gene is valid
-        response = self.app_client.get("/gene_information/gene_publications/AT1G01035")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G01035")
         expected = {
             "wasSuccessful": False,
             "error": "There are no data found for the given gene",
         }
         self.assertEqual(response.json, expected)
 
-        response = self.app_client.get("/gene_information/gene_publications/AT1G010400")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/AT1G010400")
         expected = {
             "wasSuccessful": False,
             "error": "There are no data found for the given gene",
         }
         self.assertEqual(response.json, expected)
 
         # Invalid Gene
-        response = self.app_client.get("/gene_information/gene_publications/001G01030")
+        response = self.app_client.get("/gene_information/gene_publications/arabidopsis/001G01030")
         expected = {"wasSuccessful": False, "error": "Invalid gene id"}
         self.assertEqual(response.json, expected)
 
+        # Invalid Species
+        response = self.app_client.get("/gene_information/gene_publications/x/AT1G01020")
+        expected = {"wasSuccessful": False, "error": "No data for the given species"}
+        self.assertEqual(response.json, expected)
+
     def test_get_arabidopsis_gene_isoform(self):
         """This tests checks GET request for gene isoforms Arabidopsis
         :return: