84 lines
3.0 KiB
Python
84 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from govoplan_core.core.identity import IdentitySearchProvider
|
|
from govoplan_core.db.base import Base
|
|
from govoplan_core.db.session import configure_database, reset_database
|
|
from govoplan_identity.backend.db.models import Identity, IdentityAccountLink
|
|
from govoplan_identity.backend.directory import SqlIdentityDirectory
|
|
|
|
|
|
class IdentityDirectoryTests(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
self.database = configure_database("sqlite:///:memory:")
|
|
Base.metadata.create_all(
|
|
self.database.engine,
|
|
tables=[Identity.__table__, IdentityAccountLink.__table__],
|
|
)
|
|
with self.database.session() as session:
|
|
active = Identity(
|
|
id="identity-active",
|
|
display_name="Ada Example",
|
|
external_subject="subject-active",
|
|
source="local",
|
|
is_active=True,
|
|
settings={},
|
|
)
|
|
inactive = Identity(
|
|
id="identity-inactive",
|
|
display_name="Inactive Person",
|
|
external_subject="subject-inactive",
|
|
source="local",
|
|
is_active=False,
|
|
settings={},
|
|
)
|
|
session.add_all([active, inactive])
|
|
session.flush()
|
|
session.add_all(
|
|
[
|
|
IdentityAccountLink(
|
|
id="link-active",
|
|
identity_id=active.id,
|
|
account_id="account-ada",
|
|
is_primary=True,
|
|
source="local",
|
|
),
|
|
IdentityAccountLink(
|
|
id="link-inactive",
|
|
identity_id=inactive.id,
|
|
account_id="account-inactive",
|
|
is_primary=True,
|
|
source="local",
|
|
),
|
|
]
|
|
)
|
|
session.commit()
|
|
|
|
def tearDown(self) -> None:
|
|
reset_database(dispose=True)
|
|
|
|
def test_search_capability_filters_inactive_identities_and_searches_account_ids(self) -> None:
|
|
directory = SqlIdentityDirectory()
|
|
|
|
self.assertIsInstance(directory, IdentitySearchProvider)
|
|
self.assertEqual(
|
|
("identity-active",),
|
|
tuple(identity.id for identity in directory.search_identities()),
|
|
)
|
|
matches = directory.search_identities("account-ada")
|
|
self.assertEqual(("identity-active",), tuple(identity.id for identity in matches))
|
|
self.assertEqual(("account-ada",), matches[0].account_ids)
|
|
self.assertEqual("account-ada", matches[0].primary_account_id)
|
|
|
|
def test_search_capability_can_include_inactive_identities(self) -> None:
|
|
directory = SqlIdentityDirectory()
|
|
|
|
matches = directory.search_identities("inactive", include_inactive=True)
|
|
|
|
self.assertEqual(("identity-inactive",), tuple(identity.id for identity in matches))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|