mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-08 11:46:35 +00:00
kevinveenbirkenbachandClaude Opus 4.7
300ea0a054
Generalises the postgres image build to be driven by a declarative extension→library registry instead of a hardcoded POSTGRES_VECTOR_ENABLED flag. Same Dockerfile.j2 now serves both the central svc-db-postgres image (libraries aggregated across all deployed consumers) and per- consumer images for shared:false postgres consumers whose extensions need libraries not in the postgis/postgis base (e.g. pgvector). postgres.yml.j2 switches between image: and build: based on whether custom libraries are needed. New postgres_libraries lookup filters base-image extensions out and returns recipes for the rest. Dedicated task path extracted into dedicated.yml + dedicated_postgres.yml so the block-with-when size limit stays satisfied and the postgres-type check happens once. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
"""Unit tests for ``plugins/lookup/postgres_libraries.py``."""
|
|
|
|
import importlib.util
|
|
import unittest
|
|
|
|
from ansible.errors import AnsibleError
|
|
|
|
from . import PROJECT_ROOT
|
|
|
|
|
|
def _load_module(rel_path: str, name: str):
|
|
path = PROJECT_ROOT / rel_path
|
|
spec = importlib.util.spec_from_file_location(name, str(path))
|
|
mod = importlib.util.module_from_spec(spec)
|
|
assert spec and spec.loader
|
|
spec.loader.exec_module(mod)
|
|
return mod
|
|
|
|
|
|
class PostgresLibrariesLookupTests(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.mod = _load_module(
|
|
"plugins/lookup/postgres_libraries.py",
|
|
"lookup_postgres_libraries",
|
|
)
|
|
|
|
def _run(self, terms):
|
|
lookup = self.mod.LookupModule()
|
|
return lookup.run(terms)
|
|
|
|
def test_zero_terms_raises(self):
|
|
with self.assertRaises(AnsibleError):
|
|
self._run([])
|
|
|
|
def test_two_terms_raises(self):
|
|
with self.assertRaises(AnsibleError):
|
|
self._run([["vector"], "extra"])
|
|
|
|
def test_non_list_first_term_raises(self):
|
|
with self.assertRaises(AnsibleError):
|
|
self._run(["vector"])
|
|
|
|
def test_empty_extension_list_returns_empty(self):
|
|
result = self._run([[]])
|
|
self.assertEqual(result, [[]])
|
|
|
|
def test_extension_in_registry_returns_recipe(self):
|
|
result = self._run([["vector"]])[0]
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(result[0]["extension"], "vector")
|
|
self.assertEqual(result[0]["name"], "pgvector")
|
|
self.assertIn("git_source", result[0])
|
|
|
|
def test_base_extensions_are_filtered_out(self):
|
|
result = self._run([["bloom", "postgis", "pg_trgm", "unaccent"]])[0]
|
|
self.assertEqual(result, [])
|
|
|
|
def test_mixed_extensions_returns_only_non_base(self):
|
|
result = self._run([["bloom", "vector", "postgis"]])[0]
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(result[0]["extension"], "vector")
|
|
|
|
def test_duplicates_are_deduplicated(self):
|
|
result = self._run([["vector", "vector"]])[0]
|
|
self.assertEqual(len(result), 1)
|
|
|
|
def test_blanks_are_skipped(self):
|
|
result = self._run([["", " ", "vector"]])[0]
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(result[0]["extension"], "vector")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|