test on_message instead of on_query

fetchai · Jul 6, 2023 · 4c6db8e · 4c6db8e
1 parent 1f96584
commit 4c6db8e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_field_descr.py b/tests/test_field_descr.py
@@ -49,7 +49,7 @@ def test_model_digest(self):
         self.assertEqual(model_digest_no_descr, model_digest_with_descr)
 
     def test_protocol(self):
-        @self.protocol_no_descr.on_query(create_message_no_descr())
+        @self.protocol_no_descr.on_message(create_message_no_descr())
         def _(_ctx, _sender, _msg):
             pass
 
@@ -64,7 +64,7 @@ def _(_ctx, _sender, _msg):
             self.protocol_no_descr
         )
 
-        @self.protocol_with_descr.on_query(create_message_with_descr())
+        @self.protocol_with_descr.on_message(create_message_with_descr())
         def _(_ctx, _sender, _msg):
             pass