nearly there!

aleph-im · Sep 14, 2023 · 10c7489 · 10c7489
1 parent b99a760
commit 10c7489
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/src/aleph_vrf/models.py b/src/aleph_vrf/models.py
@@ -59,7 +59,7 @@ def generate_response_hash_from_message(message: PostMessage) -> VRFResponseHash
         execution_id=content["execution_id"],
         vrf_request=ItemHash(content["vrf_request"]),
         random_bytes_hash=content["random_bytes_hash"],
-        message_hash=content["message_hash"],
+        message_hash=message.item_hash,
     )
 
 

diff --git a/tests/executor/test_integration.py b/tests/executor/test_integration.py
@@ -6,7 +6,7 @@
 import pytest
 from aleph_message.models import ItemType, MessageType, PostContent, Chain, ItemHash
 
-from aleph_vrf.models import VRFRequest, VRFResponseHash, VRFResponse
+from aleph_vrf.models import VRFRequest, VRFResponseHash, VRFResponse, VRFRandomBytes
 
 
 @pytest.mark.asyncio
@@ -54,6 +54,7 @@ def mock_vrf_request() -> VRFRequest:
 
     vrf_request = VRFRequest(
         nb_bytes=32,
+        nb_executors=4,
         nonce=42,
         vrf_function="deca" * 16,
         request_id=request_id,
@@ -110,4 +111,12 @@ async def test_normal_request_flow(
 
         resp = await executor_client.post(f"/publish/{response_hash.message_hash}")
         assert resp.status == 200, await resp.text()
-        print(await resp.text())
+        response_json = await resp.json()
+
+        random_bytes = VRFRandomBytes.parse_obj(response_json["data"])
+
+    assert random_bytes.request_id == mock_vrf_request.request_id
+    assert random_bytes.execution_id == response_hash.execution_id
+    assert random_bytes.vrf_request == item_hash
+    assert random_bytes.random_bytes_hash == response_hash.random_bytes_hash
+    # TODO: compare random number with hash