update tests

MinaFoundation · Apr 26, 2024 · 2c8b9c8 · 2c8b9c8
1 parent 2b91e3a
commit 2c8b9c8
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/tests/test_helper.py b/tests/test_helper.py
@@ -64,9 +64,9 @@ def test_array_dataframe():
             "block_hash_1",
             "remote_addr_1",
             "peer_id_1",
-            "snark_work_1",
             "graphql_control_port_1",
             "built_with_commit_sha_1",
+            "snark_work_1",
             "state_hash_1",
             "parent_1",
             "height_1",
@@ -81,9 +81,9 @@ def test_array_dataframe():
             "block_hash_2",
             "remote_addr_2",
             "peer_id_2",
-            "snark_work_2",
             "graphql_control_port_2",
             "built_with_commit_sha_2",
+            "snark_work_2",
             "state_hash_2",
             "parent_2",
             "height_2",

diff --git a/uptime_service_validation/coordinator/aws_keyspaces_client.py b/uptime_service_validation/coordinator/aws_keyspaces_client.py
@@ -159,7 +159,23 @@ def get_submissions(
                 "You have to provide either both submitted_at_start and submitted_at_end or neither"
             )
 
-        base_query = f"SELECT submitted_at_date, submitted_at, submitter, created_at, block_hash, remote_addr, peer_id, snark_work, graphql_control_port, built_with_commit_sha, state_hash, parent, height, slot, validation_error, verified FROM {self.aws_keyspace}.submissions"
+        base_query = f"""SELECT 
+                        submitted_at_date, 
+                        submitted_at, 
+                        submitter, 
+                        created_at, 
+                        block_hash, 
+                        remote_addr, 
+                        peer_id, 
+                        graphql_control_port, 
+                        built_with_commit_sha, 
+                        state_hash, 
+                        parent, 
+                        height, 
+                        slot, 
+                        validation_error, 
+                        verified 
+                       FROM {self.aws_keyspace}.submissions"""
 
         # For storing conditions and corresponding parameters
         conditions = []
@@ -226,7 +242,6 @@ def get_submissions(
                 block_hash=row.block_hash,
                 remote_addr=row.remote_addr,
                 peer_id=row.peer_id,
-                snark_work=row.snark_work,
                 graphql_control_port=row.graphql_control_port,
                 built_with_commit_sha=row.built_with_commit_sha,
                 state_hash=row.state_hash,