fix: implemented well-notes tests

Author: jirhiker

tests/features/steps/well-notes.py

@@ -16,41 +16,90 @@
 from behave import when, then
 
 
-@when("the user retrieves the well 1")
+@when("the user retrieves the well 9999")
 def step_impl(context):
-    context.response = context.client.get("thing/water-well/1")
+    context.response = context.client.get("thing/water-well/9999")
+    context.notes = {}
 
 
-@when("the user retrieves the well 9999")
+@then("the response should include an error message indicating the well was not found")
 def step_impl(context):
-    context.response = context.client.get("thing/water-well/9999")
+    assert {"detail": "Thing with ID 9999 not found."} == context.response.json()
 
 
-@then("the response should contain a current_location field")
+@then("the notes should be a non-empty string")
 def step_impl(context):
-    assert "current_location" in context.response.json()
+    for k, note in context.notes.items():
+        assert note, f"{k} Note is empty"
 
 
-@then("the response should include notes")
+@when("the user retrieves the well by ID via path parameter")
 def step_impl(context):
-    assert "notes" in context.response.json()
-    context.notes = context.response.json()["notes"]
+    context.response = context.client.get(
+        f"thing/water-well/{context.objects['wells'][0].id}"
+    )
+    context.notes = {}
 
 
-@then("the response should include an error message indicating the well was not found")
+@then(
+    "null values in the response should be represented as JSON null (not placeholder strings)"
+)
 def step_impl(context):
-    assert {"detail": "Thing with ID 9999 not found."} == context.response.json()
+    data = context.response.json()
+    for k, v in data.items():
+        if v == "":
+            assert v is None, f"Value for key {k} is an empty string but should be null"
 
 
-@then("the response should include well_construction_notes")
+@then(
+    "the response should include location notes (i.e. driving directions and geographic well location notes)"
+)
 def step_impl(context):
-    assert "well_construction_notes" in context.response.json()
-    context.notes = context.response.json()["well_construction_notes"]
+    data = context.response.json()
+    location = data["current_location"]
+    assert "notes" in location, "Response does not include location notes"
+    assert location["notes"] is not None, "Location notes is null"
+    context.notes["location"] = location["notes"]
 
 
-@then("the notes should be a non-empty string")
+@then(
+    "the response should include construction notes (i.e. pump notes and other construction notes)"
+)
+def step_impl(context):
+    data = context.response.json()
+    assert (
+        "well_construction_notes" in data
+    ), "Response does not include construction notes"
+    assert data["well_construction_notes"] is not None, "Construction notes is null"
+    context.notes["construction"] = data["well_construction_notes"]
+
+
+@then("the response should include general well notes (catch all notes field)")
+def step_impl(context):
+    data = context.response.json()
+    assert "notes" in data, "Response does not include notes"
+    assert data["notes"] is not None, "Notes is null"
+    context.notes["general"] = data["notes"]
+
+
+@then(
+    "the response should include measuring notes (notes about measuring/visiting the well, on Access form)"
+)
+def step_impl(context):
+    data = context.response.json()
+    assert "measuring_notes" in data, "Response does not include measuring notes"
+    assert data["measuring_notes"] is not None, "Measuring notes is null"
+    context.notes["measuring"] = data["measuring_notes"]
+
+
+@then(
+    "the response should include water notes (i.e. water bearing zone information and other info from ose reports)"
+)
 def step_impl(context):
-    assert bool(context.notes) == True
+    data = context.response.json()
+    assert "water_notes" in data, "Response does not include water notes"
+    assert data["water_notes"] is not None, "Water notes is null"
+    context.notes["water"] = data["water_notes"]
 
 
 # ============= EOF =============================================