added test for loading json file

This commit is contained in:
Andrei Stoica 2022-10-15 13:17:23 -04:00
parent 6189de8039
commit 9b9e629548
1 changed files with 21 additions and 0 deletions

View File

@ -1,7 +1,28 @@
import inspect import inspect
import json
import os
from recipe_graph import insert_sites from recipe_graph import insert_sites
from sqlalchemy import select from sqlalchemy import select
from sqlalchemy.exc import SQLAlchemyError from sqlalchemy.exc import SQLAlchemyError
import sqlalchemy import sqlalchemy
import pytest import pytest
@pytest.fixture
def json_data() -> list[dict]:
return [{"key": "value"}, {"test": "value1", "test2": "value2"}]
@pytest.fixture
def json_file(json_data: list[dict]) -> str:
f_path = "test.json"
with open(f_path, 'w') as f:
json.dump(json_data, f)
yield f_path
if os.path.exists(f_path):
os.remove(f_path)
def test_load_file(json_file: str, json_data):
test_data = insert_sites.load_file(json_file)
assert test_data == json_data