def test_get_problem_set(problem_set_entity):
sqlite_test_util.clean_start(drop_tables_list)
add_problem_set_to_db(problem_set_entity)
result = sqlite_manager.static_problem_set_repo.get(EXAMPLE_SET_ID)
assert isinstance(result, StaticProblemSetEntity)
def test_check_problem_set_id_availability(problem_set_entity):
sqlite_test_util.clean_start(drop_tables_list)
add_problem_set_to_db(problem_set_entity)
result = sqlite_manager.static_problem_set_repo.check_id_available(
problem_set_entity.id)
assert result is False
def test_add_problem_set(problem_set_entity):
sqlite_test_util.clean_start(drop_tables_list)
add_problem_set_to_db(problem_set_entity)
sql_fetch = """SELECT * FROM problem_set
"""
from problem_sets.static.data.sqlite.test.sqlite_test_util import conn
result = sqlite_util.query_fetch(conn, sql_fetch)
assert result is not None and len(result) == 1
def test_add_problem(problem_set_entity, problem_entity):
sqlite_test_util.clean_start(drop_tables_list)
add_problem_set_and_problem_to_db(problem_set_entity, problem_entity)
command_problem_list = f"""SELECT * FROM problem WHERE set_id=?
"""
from problem_sets.static.data.sqlite.test.sqlite_test_util import conn
result = sqlite_util.query_fetch(conn, command_problem_list,
(EXAMPLE_SET_ID, ))
assert len(result) == 1
def test_get_problem(problem_set_entity, problem_entity):
sqlite_test_util.clean_start(drop_tables_list)
add_problem_set_and_problem_to_db(problem_set_entity, problem_entity)
result = sqlite_manager.static_problem_repo.get(problem_entity.id)
mismatch = problem_entity
mismatch.id = 2
bad_result = sqlite_manager.static_problem_repo.get(mismatch.id)
assert isinstance(result, StaticProblemEntity)
assert bad_result is None
def clean_start(tables: Union[str, List[str]]):
sqlite_test_util.clean_start(tables)
data_manager.initialize()