#23 refactor to use shared code between services, part 2

tests/adapter/test_cleverthis_service_adapter.py

@@ -18,7 +18,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
 
         # Create adapter instance
         self.adapter = CleverThisServiceAdapter(
-            amq_configuration=self.mock_amq_config, session=self.mock_session
+            amq_configuration=self.mock_amq_config, routes=[], session=self.mock_session
         )
 
     def test_init(self):
@@ -56,7 +56,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
 
         # Test when authentication is required but not provided
         response = await self.adapter.on_message(mock_message)
-        self.assertEqual(response.response_code, 401)
+        self.assertEqual(response.response_code(), "401")
         mock_get_user.assert_not_called()
 
     @patch.object(CleverThisServiceAdapter, "get_current_user")
@@ -91,9 +91,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
         mock_message.trace_info = {}
 
         mock_response = MagicMock(spec=DataResponse)
-        with patch.object(
-            self.adapter, "post", return_value=mock_response
-        ) as mock_post:
+        with patch.object(self.adapter, "post", return_value=mock_response) as mock_post:
             response = await self.adapter.on_message(mock_message)
             self.assertEqual(response, mock_response)
             mock_get_user.assert_not_called()
@@ -139,7 +137,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
 
         response = await self.adapter.get(mock_message, None)
 
-        self.assertEqual(response.response_code, 200)
+        self.assertEqual(response.response_code(), "200")
         self.mock_session.get.assert_called_once_with(
             "http://testhost:8080/api/test",
             headers={"X-Test": "value", "trace-id": "123"},
@@ -155,7 +153,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
 
         response = await self.adapter.get(mock_message, None)
 
-        self.assertEqual(response.response_code, 404)
+        self.assertEqual(response.response_code(), "404")
 
     async def test_put_with_form_handling(self):
         mock_message = MagicMock(spec=DataMessage)
@@ -179,7 +177,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
         ) as mock_handler:
             response = await self.adapter.put(mock_message, None)
 
-            self.assertEqual(response.response_code, 200)
+            self.assertEqual(response.response_code(), "200")
             mock_handler.assert_called_once()
 
     async def test_post_with_form_handling(self):
@@ -204,7 +202,7 @@ class TestCleverThisServiceAdapter(unittest.TestCase):
         ) as mock_handler:
             response = await self.adapter.post(mock_message, None)
 
-            self.assertEqual(response.response_code, 201)
+            self.assertEqual(response.response_code(), "201")
             mock_handler.assert_called_once()
 
     async def test_head_not_implemented(self):

tests/adapter/test_data_response_factory.py

@@ -12,7 +12,7 @@ class TestDataResponseFactory(TestCase):
         _f = DataResponseFactory()
         _g = DataMessageFactory(111)
         _resp = _f.create_async_response_message(_g.generate_snowflake_id())
-        self.assertEqual(200, _resp.response_code)
+        self.assertEqual("200", _resp.response_code())
         self.assertEqual(b"OK", _resp.base64body())
 
     def test_serialize(self):
@@ -37,5 +37,5 @@ class TestDataResponseFactory(TestCase):
         _ser = _f.serialize(_resp)
         _deser = _f.from_bytes(_ser)
         self.assertEqual(_snowflakeId.as_string(), _deser.id)
-        self.assertEqual(200, _deser.response_code)
+        self.assertEqual("200", _deser.response_code())
         self.assertEqual(b"OK", _deser.base64body())