fix testing problem; reformat

Author: MHHukiewitz

src/aleph/sdk/models.py

@@ -37,7 +37,9 @@ class Post(BaseModel):
     )
     address: str = Field(description="The address of the sender of the POST message")
     ref: Optional[str] = Field(description="Other message referenced by this one")
-    channel: Optional[str] = Field(description="The channel where the POST message was published")
+    channel: Optional[str] = Field(
+        description="The channel where the POST message was published"
+    )
     created: datetime = Field(description="The time when the POST message was created")
     last_updated: datetime = Field(
         description="The time when the POST message was last updated"

src/aleph/sdk/node/post.py

@@ -58,7 +58,7 @@ def message_to_post(message: PostMessage) -> Post:
             "ref": message.content.ref if hasattr(message.content, "ref") else None,
             "channel": message.channel,
             "created": datetime.fromtimestamp(message.time),
-            "last_updated": datetime.fromtimestamp(message.time)
+            "last_updated": datetime.fromtimestamp(message.time),
         }
     )
 

tests/unit/conftest.py

@@ -113,10 +113,10 @@ def messages() -> List[AlephMessage]:
 
 @pytest.fixture
 def raw_messages_response(messages):
-    return {
-        "messages": [message.dict() for message in messages],
+    return lambda page: {
+        "messages": [message.dict() for message in messages] if page == 1 else [],
         "pagination_item": "messages",
-        "pagination_page": 1,
-        "pagination_per_page": 20,
-        "pagination_total": 2,
+        "pagination_page": page,
+        "pagination_per_page": max(len(messages), 20),
+        "pagination_total": len(messages) if page == 1 else 0,
     }

tests/unit/test_node.py

@@ -54,8 +54,9 @@ async def text(self):
 
 
 class MockGetResponse:
-    def __init__(self, response):
-        self.response = response
+    def __init__(self, response_message, page=1):
+        self.response_message = response_message
+        self.page = page
 
     async def __aenter__(self):
         return self
@@ -72,7 +73,7 @@ def raise_for_status(self):
             raise Exception("Bad status code")
 
     async def json(self):
-        return self.response
+        return self.response_message(self.page)
 
 
 @pytest.fixture
@@ -92,7 +93,10 @@ def mock_session_with_two_messages(
         sync=kwargs.get("sync", False),
     )
     http_session.get = MagicMock()
-    http_session.get.return_value = MockGetResponse(raw_messages_response)
+    http_session.get.side_effect = lambda *args, **kwargs: MockGetResponse(
+        response_message=raw_messages_response,
+        page=kwargs.get("params", {}).get("page", 1),
+    )
 
     client = AuthenticatedAlephClient(
         account=ethereum_account, api_server="http://localhost"

tests/unit/test_node_get.py

@@ -13,7 +13,8 @@
 
 from aleph.sdk.chains.ethereum import get_fallback_account
 from aleph.sdk.exceptions import MessageNotFoundError
-from aleph.sdk.node import MessageCache, message_to_post
+from aleph.sdk.node import MessageCache
+from aleph.sdk.node.post import message_to_post
 
 
 @pytest.mark.asyncio