fix query building list parameter tests

This commit is contained in:
Simon
2025-02-14 16:39:07 +07:00
parent 46c974c078
commit e3488ca4d8
2 changed files with 15 additions and 17 deletions

View File

@@ -7,9 +7,9 @@ from playlist.src.query_building import QueryBuilder
def test_build_data():
"""test for correct key building"""
qb = QueryBuilder(
channel=["test_channel"],
subscribed=["true"],
type=["regular"],
channel="test_channel",
subscribed=True,
type="regular",
)
result = qb.build_data()
must_list = result["query"]["bool"]["must"]
@@ -22,7 +22,7 @@ def test_build_data():
def test_parse_type():
"""validate type"""
qb = QueryBuilder(type=["regular"])
qb = QueryBuilder(type="regular")
with pytest.raises(ValueError):
qb.parse_type("invalid")