🚨 test(conftest.py): add basic_graph_data fixture to load basic example data

🚨 test(test_websocket.py): add tests for build init, stream, and websocket endpoint
The `basic_graph_data` fixture is added to load the basic example data. The `test_init_build`, `test_stream_build`, and `test_websocket_endpoint` tests are added to test the build init, stream, and websocket endpoint. The `test_websocket_endpoint_after_build` test is added to test the websocket endpoint after the build.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-12 13:06:39 -03:00
commit a7c9b04611
2 changed files with 51 additions and 39 deletions

View file

@ -63,6 +63,12 @@ def get_graph(_type="basic"):
return Graph(nodes, edges)
@pytest.fixture
def basic_graph_data():
with open(pytest.BASIC_EXAMPLE_PATH, "r") as f:
return json.load(f)
@pytest.fixture
def basic_graph():
return get_graph()