add tests on new performance params and get_latest_result (#58)

* add tests

* naming fix

Author: grkhr

tests/e2e/test_async_client.py

@@ -33,6 +33,21 @@ async def test_refresh_context_manager(self):
             results = (await cl.refresh(self.query)).get_rows()
         self.assertGreater(len(results), 0)
 
+    async def test_refresh_context_manager_performance_large(self):
+        async with AsyncDuneClient(self.valid_api_key) as cl:
+            results = (await cl.refresh(self.query, performance="large")).get_rows()
+        self.assertGreater(len(results), 0)
+
+    async def test_get_latest_result_with_query_object(self):
+        async with AsyncDuneClient(self.valid_api_key) as cl:
+            results = (await cl.get_latest_result(self.query)).get_rows()
+        self.assertGreater(len(results), 0)
+
+    async def test_get_latest_result_with_query_id(self):
+        async with AsyncDuneClient(self.valid_api_key) as cl:
+            results = (await cl.get_latest_result(self.query.query_id)).get_rows()
+        self.assertGreater(len(results), 0)
+
 
 if __name__ == "__main__":
     unittest.main()

tests/e2e/test_client.py

@@ -45,6 +45,11 @@ def test_refresh(self):
         results = dune.refresh(self.query).get_rows()
         self.assertGreater(len(results), 0)
 
+    def test_refresh_performance_large(self):
+        dune = DuneClient(self.valid_api_key)
+        results = dune.refresh(self.query, performance="large").get_rows()
+        self.assertGreater(len(results), 0)
+
     def test_refresh_into_dataframe(self):
         dune = DuneClient(self.valid_api_key)
         pd = dune.refresh_into_dataframe(self.query)
@@ -160,6 +165,16 @@ def test_invalid_job_id_error(self):
             "{'error': 'The requested execution ID (ID: Wonky Job ID) is invalid.'}",
         )
 
+    def test_get_latest_result_with_query_object(self):
+        dune = DuneClient(self.valid_api_key)
+        results = dune.get_latest_result(self.query).get_rows()
+        self.assertGreater(len(results), 0)
+
+    def test_get_latest_result_with_query_id(self):
+        dune = DuneClient(self.valid_api_key)
+        results = dune.get_latest_result(self.query.query_id).get_rows()
+        self.assertGreater(len(results), 0)
+
 
 if __name__ == "__main__":
     unittest.main()