test(add_pdf): added test for supplemental material

Author: valearna

tests/literature/test_corpus.py

@@ -75,6 +75,18 @@ def test_load_from_wb_database(self):
                                  exclude_temp_pdf=True)
         self.assertFalse(any([paper.is_temp() for paper in cm.get_all_papers()]))
 
+    def test_load_supplemental(self):
+        db_config = read_db_config()
+        tazendra_config = read_tazendra_config()
+        cm = CorpusManager()
+        cm.load_from_wb_database(db_name=db_config["wb_database"]["db_name"],
+                                 db_user=db_config["wb_database"]["db_user"],
+                                 db_password=db_config["wb_database"]["db_password"],
+                                 db_host=db_config["wb_database"]["db_host"],
+                                 ssh_user=tazendra_config["ssh"]["ssh_user"],
+                                 ssh_passwd=tazendra_config["ssh"]["ssh_password"], paper_ids=["00062512"])
+        self.assertTrue(len(cm.get_paper("00062512").supplemental_docs) > 0)
+
     @unittest.skipIf(not os.path.exists(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "data",
                                                      "local_config", "db.cfg")), "Test DB config file not present")
     def test_load_from_wb_database_afp(self):