feat: allow to pass repository identifier to all database methods

This commit is contained in:
2023-11-04 16:36:14 +02:00
parent 7b667c8157
commit b116e6fa07
3 changed files with 40 additions and 25 deletions

View File

@ -19,12 +19,10 @@ def test_build_queue_insert_clear_multi(database: SQLite, package_ahriman: Packa
must clear all packages from queue for specific repository
"""
database.build_queue_insert(package_ahriman)
database._repository_id = RepositoryId("i686", database._repository_id.name)
database.build_queue_insert(package_ahriman)
database.build_queue_insert(package_ahriman, RepositoryId("i686", database._repository_id.name))
database.build_queue_clear(None)
database._repository_id = RepositoryId("x86_64", database._repository_id.name)
assert database.build_queue_get() == [package_ahriman]
assert database.build_queue_get(RepositoryId("i686", database._repository_id.name)) == [package_ahriman]
def test_build_queue_insert_clear_specific(database: SQLite, package_ahriman: Package,
@ -68,19 +66,15 @@ def test_build_queue_insert_multi(database: SQLite, package_ahriman: Package) ->
assert database.build_queue_get() == [package_ahriman]
package_ahriman.version = "2"
database._repository_id = RepositoryId("i686", database._repository_id.name)
database.build_queue_insert(package_ahriman)
assert database.build_queue_get() == [package_ahriman]
database.build_queue_insert(package_ahriman, RepositoryId("i686", database._repository_id.name))
assert database.build_queue_get(RepositoryId("i686", database._repository_id.name)) == [package_ahriman]
package_ahriman.version = "1"
database._repository_id = RepositoryId("x86_64", database._repository_id.name)
assert database.build_queue_get() == [package_ahriman]
assert database.build_queue_get(RepositoryId("x86_64", database._repository_id.name)) == [package_ahriman]
package_ahriman.version = "3"
database._repository_id = RepositoryId(database._repository_id.architecture, "repo")
database.build_queue_insert(package_ahriman)
assert database.build_queue_get() == [package_ahriman]
database.build_queue_insert(package_ahriman, RepositoryId(database._repository_id.architecture, "repo"))
assert database.build_queue_get(RepositoryId(database._repository_id.architecture, "repo")) == [package_ahriman]
package_ahriman.version = "1"
database._repository_id = RepositoryId(database._repository_id.architecture, "aur-clone")
assert database.build_queue_get() == [package_ahriman]