Skip to content

Commit

Permalink
add basic integration testing for custom endpoints
Browse files Browse the repository at this point in the history
  • Loading branch information
philippWassibauer committed Jun 19, 2024
1 parent 9aa2f86 commit 92254e1
Showing 1 changed file with 25 additions and 0 deletions.
25 changes: 25 additions & 0 deletions tests/e2e/test_custom_endpoints.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
import copy
import os
import time
import unittest

import dotenv

from dune_client.client import DuneClient

dotenv.load_dotenv()


class TestCustomEndpoints(unittest.TestCase):
def setUp(self) -> None:
self.valid_api_key = os.environ["DUNE_API_KEY"]

def test_get_execution_status(self):
dune = DuneClient(self.valid_api_key)
results = dune.get_custom_endpoint_result("dune", "new-test")
self.assertEqual(len(results.get_rows()), 10)
self.assertEqual(len(results.get_columns()), 5)


if __name__ == "__main__":
unittest.main()

0 comments on commit 92254e1

Please sign in to comment.