WS-3151: unit tests

Author: Adam Soos

examples/record_similarity.py

@@ -34,8 +34,8 @@ def run(key, alt_url='https://api.rosette.com/rest/v1/'):
         }
     }
     properties = {
-        "threshold": 0.1,
-        "includeExplainInfo": False
+        "threshold": 0.7,
+        "includeExplainInfo": True
     }
     records = {
         "left": [

rosette/api.py

@@ -344,6 +344,7 @@ def validate(self):
                 "Required Name De-Duplication parameter is missing: names",
                 repr("names"))
 
+
 class RecordSimilarityParameters(_RequestParametersBase):
     """Parameter object for C{record-similarity} endpoint.
     Required:
@@ -362,7 +363,7 @@ def validate(self):
             if self[option] is None:
                 raise RosetteException(
                     "missingParameter",
-                    "Required Name Similarity parameter is missing: " + option,
+                    "Required Record Similarity parameter is missing: " + option,
                     repr(option))
 
 

tests/test_rosette_api.py

@@ -29,6 +29,7 @@
                          NameTranslationParameters,
                          NameSimilarityParameters,
                          NameDeduplicationParameters,
+                         RecordSimilarityParameters,
                          RosetteException)
 
 _ISPY3 = sys.version_info[0] == 3
@@ -484,6 +485,10 @@ def test_the_name_requests_with_text(api, json_response):
         result = api.address_similarity("should fail")
     assert e_rosette.value.status == 'incompatible'
 
+    with pytest.raises(RosetteException) as e_rosette:
+        result = api.record_similarity("should fail")
+    assert e_rosette.value.status == 'incompatible'
+
     httpretty.disable()
     httpretty.reset()
 
@@ -958,4 +963,61 @@ def test_the_events_endpoint(api, json_response, doc_params):
     result = api.events(doc_params)
     assert result["name"] == "Rosette"
     httpretty.disable()
+    httpretty.reset()
+
+# Test the record similarity endpoint
+
+
+def test_the_record_similarity_endpoint(api, json_response):
+    """Test the record similarity endpoint"""
+    httpretty.enable()
+    httpretty.register_uri(httpretty.POST, "https://api.rosette.com/rest/v1/record-similarity",
+                           body=json_response, status=200, content_type="application/json")
+
+    params = RecordSimilarityParameters()
+    params["fields"] = {}
+    params["properties"] = {}
+    params["records"] = []
+    result = api.record_similarity(params)
+    assert result["name"] == "Rosette"
+    httpretty.disable()
+    httpretty.reset()
+
+
+# Tests for required record-similarities parameters
+def test_for_record_similarity_required_parameters(api, json_response):
+    """Test record similarity parameters"""
+    httpretty.enable()
+    httpretty.register_uri(httpretty.POST, "https://api.rosette.com/rest/v1/record-similarity",
+                           body=json_response, status=200, content_type="application/json")
+
+    params = RecordSimilarityParameters()
+
+    with pytest.raises(RosetteException) as e_rosette:
+        api.record_similarity(params)
+
+    assert e_rosette.value.status == 'missingParameter'
+    assert e_rosette.value.message == 'Required Record Similarity parameter is missing: fields'
+
+    params["fields"] = {}
+
+    with pytest.raises(RosetteException) as e_rosette:
+        api.record_similarity(params)
+
+    assert e_rosette.value.status == 'missingParameter'
+    assert e_rosette.value.message == 'Required Record Similarity parameter is missing: properties'
+
+    params["properties"] = {}
+
+    with pytest.raises(RosetteException) as e_rosette:
+        api.record_similarity(params)
+
+    assert e_rosette.value.status == 'missingParameter'
+    assert e_rosette.value.message == 'Required Record Similarity parameter is missing: records'
+
+    params["records"] = []
+
+    result = api.record_similarity(params)
+    assert result["name"] == "Rosette"
+    httpretty.disable()
     httpretty.reset()