Update neon_must_respond to support external API access with update…

…d unit test (#487) Co-authored-by: Daniel McKnight <[email protected]>
NeonGeckoCom · Nov 27, 2023 · e7a12e3 · e7a12e3
1 parent 19185a1
commit e7a12e3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/neon_utils/message_utils.py b/neon_utils/message_utils.py
@@ -294,4 +294,7 @@ def neon_must_respond(message: Message = None) -> bool:
             else:
                 # Solo Private
                 return True
+    if message.context.get("client") in ("mana", "mq_api"):
+        # API requests should always return a response
+        return True
     return False
diff --git a/tests/message_util_tests.py b/tests/message_util_tests.py
@@ -277,12 +277,17 @@ def test_neon_must_respond(self):
                                               "conversation with Neon"},
                                 {"klat_data": {
                                     "title": "!PRIVATE:user"}})
+        mq_api_message = Message("", {}, {"client": "mq_api"})
+        mana_api_message = Message("", {}, {"client": "mana"})
+
         self.assertFalse(neon_must_respond())
         self.assertTrue(neon_must_respond(private_message_solo))
         self.assertTrue(neon_must_respond(private_message_neon))
         self.assertFalse(neon_must_respond(private_message_neon_plus))
         self.assertFalse(neon_must_respond(public_message))
         self.assertFalse(neon_must_respond(first_message))
+        self.assertTrue(neon_must_respond(mq_api_message))
+        self.assertTrue(neon_must_respond(mana_api_message))
 
 
 if __name__ == '__main__':