Complete official repository support (#59)

tests/ahriman/core/database/data/test_data_init.py

@@ -15,11 +15,24 @@ def test_migrate_data_initial(connection: Connection, configuration: Configurati
     packages = mocker.patch("ahriman.core.database.data.migrate_package_statuses")
     patches = mocker.patch("ahriman.core.database.data.migrate_patches")
     users = mocker.patch("ahriman.core.database.data.migrate_users_data")
+    remotes = mocker.patch("ahriman.core.database.data.migrate_package_remotes")
 
     migrate_data(MigrationResult(old_version=0, new_version=900), connection, configuration)
     packages.assert_called_once_with(connection, repository_paths)
     patches.assert_called_once_with(connection, repository_paths)
     users.assert_called_once_with(connection, configuration)
+    remotes.assert_called_once_with(connection, repository_paths)
+
+
+def test_migrate_data_remotes(connection: Connection, configuration: Configuration,
+                              repository_paths: RepositoryPaths, mocker: MockerFixture) -> None:
+    """
+    must perform initial migration
+    """
+    remotes = mocker.patch("ahriman.core.database.data.migrate_package_remotes")
+
+    migrate_data(MigrationResult(old_version=1, new_version=900), connection, configuration)
+    remotes.assert_called_once_with(connection, repository_paths)
 
 
 def test_migrate_data_skip(connection: Connection, configuration: Configuration, mocker: MockerFixture) -> None:

tests/ahriman/core/database/data/test_package_remotes.py

@@ -0,0 +1,66 @@
+import pytest
+
+from pytest_mock import MockerFixture
+from sqlite3 import Connection
+
+from ahriman.core.database.data import migrate_package_remotes
+from ahriman.models.package import Package
+from ahriman.models.repository_paths import RepositoryPaths
+
+
+def test_migrate_package_remotes(package_ahriman: Package, connection: Connection, repository_paths: RepositoryPaths,
+                                 mocker: MockerFixture) -> None:
+    """
+    must put package remotes to database
+    """
+    mocker.patch(
+        "ahriman.core.database.operations.package_operations.PackageOperations._packages_get_select_package_bases",
+        return_value={package_ahriman.base: package_ahriman})
+    mocker.patch("pathlib.Path.exists", return_value=False)
+
+    migrate_package_remotes(connection, repository_paths)
+    connection.execute.assert_called_once_with(pytest.helpers.anyvar(str, strict=True), pytest.helpers.anyvar(int))
+
+
+def test_migrate_package_remotes_has_local(package_ahriman: Package, connection: Connection,
+                                           repository_paths: RepositoryPaths, mocker: MockerFixture) -> None:
+    """
+    must skip processing for packages which have local cache
+    """
+    mocker.patch(
+        "ahriman.core.database.operations.package_operations.PackageOperations._packages_get_select_package_bases",
+        return_value={package_ahriman.base: package_ahriman})
+    mocker.patch("pathlib.Path.exists", return_value=True)
+
+    migrate_package_remotes(connection, repository_paths)
+    connection.execute.assert_not_called()
+
+
+def test_migrate_package_remotes_vcs(package_ahriman: Package, connection: Connection,
+                                     repository_paths: RepositoryPaths, mocker: MockerFixture) -> None:
+    """
+    must process VCS packages with local cache
+    """
+    mocker.patch(
+        "ahriman.core.database.operations.package_operations.PackageOperations._packages_get_select_package_bases",
+        return_value={package_ahriman.base: package_ahriman})
+    mocker.patch("pathlib.Path.exists", return_value=True)
+    mocker.patch.object(Package, "is_vcs", True)
+
+    migrate_package_remotes(connection, repository_paths)
+    connection.execute.assert_called_once_with(pytest.helpers.anyvar(str, strict=True), pytest.helpers.anyvar(int))
+
+
+def test_migrate_package_remotes_no_remotes(package_ahriman: Package, connection: Connection,
+                                            repository_paths: RepositoryPaths, mocker: MockerFixture) -> None:
+    """
+    must skip processing in case if no remotes generated (should never happen)
+    """
+    mocker.patch(
+        "ahriman.core.database.operations.package_operations.PackageOperations._packages_get_select_package_bases",
+        return_value={package_ahriman.base: package_ahriman})
+    mocker.patch("pathlib.Path.exists", return_value=False)
+    mocker.patch("ahriman.models.remote_source.RemoteSource.from_remote", return_value=None)
+
+    migrate_package_remotes(connection, repository_paths)
+    connection.execute.assert_not_called()

tests/ahriman/core/database/data/test_users.py

@@ -9,7 +9,7 @@ from ahriman.core.database.data import migrate_users_data
 
 def test_migrate_users_data(connection: Connection, configuration: Configuration) -> None:
     """
-    must users to database
+    must put users to database
     """
     configuration.set_option("auth:read", "user1", "password1")
     configuration.set_option("auth:write", "user2", "password2")

tests/ahriman/core/database/migrations/test_m001_package_source.py

@@ -0,0 +1,8 @@
+from ahriman.core.database.migrations.m001_package_source import steps
+
+
+def test_migration_package_source() -> None:
+    """
+    migration must not be empty
+    """
+    assert steps

tests/ahriman/core/database/operations/test_package_operations.py

@@ -7,6 +7,8 @@ from unittest import mock
 from ahriman.core.database.sqlite import SQLite
 from ahriman.models.build_status import BuildStatus, BuildStatusEnum
 from ahriman.models.package import Package
+from ahriman.models.package_source import PackageSource
+from ahriman.models.remote_source import RemoteSource
 
 
 def test_package_remove_package_base(database: SQLite, connection: Connection) -> None:
@@ -166,3 +168,23 @@ def test_package_update_update(database: SQLite, package_ahriman: Package) -> No
     assert next(db_status.status
                 for db_package, db_status in database.packages_get()
                 if db_package.base == package_ahriman.base) == BuildStatusEnum.Failed
+
+
+def test_remote_update_get(database: SQLite, package_ahriman: Package) -> None:
+    """
+    must insert and retrieve package remote
+    """
+    database.remote_update(package_ahriman)
+    assert database.remotes_get()[package_ahriman.base] == package_ahriman.remote
+
+
+def test_remote_update_update(database: SQLite, package_ahriman: Package) -> None:
+    """
+    must perform package remote update for existing package
+    """
+    database.remote_update(package_ahriman)
+    remote_source = RemoteSource.from_remote(PackageSource.Repository, package_ahriman.base, "community")
+    package_ahriman.remote = remote_source
+
+    database.remote_update(package_ahriman)
+    assert database.remotes_get()[package_ahriman.base] == remote_source