add missing test

projecthorus · May 19, 2024 · 4fdf302 · 4fdf302
1 parent 84cc085
commit 4fdf302
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lambda/sonde_api_to_iot_core/__main__.py b/lambda/sonde_api_to_iot_core/__main__.py
@@ -222,5 +222,16 @@ def test_dfm_misid_payload(self):
         self.assertNotIn("temp", sns_call[0])
         self.assertIn("invalid_temp", sns_call[0])
 
+    def test_dfm_09id_payload(self):
+        payload = copy.deepcopy(example_body)
+        payload[0]["datetime"] = datetime.datetime.now().isoformat()
+        payload[0]["type"] = "DFM"
+        payload[0]["subtype"] = "DFM09"
+        payload[0]["serial"] = "21068595"
+        output = lambda_handler(compress_payload(payload), fakeContext())
+        sns.publish.assert_called()
+        self.assertEqual(output["body"], "^v^ telm logged")
+        self.assertEqual(output["statusCode"], 200)
+
 if __name__ == '__main__':
     unittest.main()