add test for checking if DOI is in HAL

Author: luclaurent

src/push2HAL/libHAL.py

@@ -34,7 +34,7 @@
 def getDataFromHAL(
     txtsearch=None,
     typeI=None,
-    typeDB=None,
+    typeDB="article",
     typeR="json",
     returnFields="title_s,author_s,halId_s,label_s,docid",
     url=dflt.HAL_API_SEARCH_URL,
@@ -76,6 +76,9 @@ def getDataFromHAL(
     elif typeI == "docId":
         Logger.debug("Searching for document's ID: {}".format(txtsearch))
         query = "halId_s:{}".format(txtsearch)
+    elif typeI == "doi":
+        Logger.debug("Searching for document's doi: {}".format(txtsearch))
+        query = "doiId_id:{}".format(txtsearch)
     #
     Logger.debug("Return format: {}".format(typeR))
 
@@ -100,6 +103,19 @@ def getDataFromHAL(
         return data
     return []
 
+def checkDoiInHAL(doi):
+    """ Check if DOI is already in HAL """
+    # request
+    dataFromHAL = getDataFromHAL(txtsearch=doi,
+                                    typeI='doi',
+                                    typeDB="article",
+                                    typeR="json")
+    return_code = False
+    if dataFromHAL:
+        if len(dataFromHAL) > 0:
+                return_code = True
+    return return_code
+
 
 def choose_from_results(
     results, forceSelection=False, maxNumber=dflt.DEFAULT_MAX_NUMBER_RESULTS

tests/libHAL_manu.py

@@ -0,0 +1,9 @@
+from push2HAL import libHAL
+
+doi = '10.1007/s11831-017-9226-3'
+res = libHAL.checkDoiInHAL('10.1007/s11831-017-9226-3')
+print('{} in HAL: {}'.format(doi,res))
+
+doi = '10.1007/XXXX'
+res = libHAL.checkDoiInHAL('10.1007/XXXX')
+print('{} in HAL: {}'.format(doi,res))

tests/test_libHAL.py

@@ -0,0 +1,10 @@
+import os
+from push2HAL import libHAL
+
+def test_doiInHAL():
+    res = libHAL.checkDoiInHAL('10.1007/s11831-017-9226-3')
+    assert res == True
+    
+def test_doiNotInHAL():
+    res = libHAL.checkDoiInHAL('10.1007/XXXX')
+    assert res == False