120 lines
4.7 KiB
Python
120 lines
4.7 KiB
Python
"""outpost tests"""
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from django.test import TestCase
|
|
from kubernetes.client import AppsV1Api
|
|
from kubernetes.client.exceptions import OpenApiException
|
|
|
|
from authentik.core.tests.utils import create_test_flow
|
|
from authentik.lib.config import CONFIG
|
|
from authentik.outposts.controllers.k8s.deployment import DeploymentReconciler
|
|
from authentik.outposts.controllers.k8s.triggers import NeedsUpdate
|
|
from authentik.outposts.models import KubernetesServiceConnection, Outpost, OutpostType
|
|
from authentik.outposts.tasks import outpost_connection_discovery
|
|
from authentik.providers.proxy.controllers.kubernetes import ProxyKubernetesController
|
|
from authentik.providers.proxy.models import ProxyProvider
|
|
|
|
|
|
class OutpostKubernetesTests(TestCase):
|
|
"""Test Kubernetes Controllers"""
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
# Ensure that local connection have been created
|
|
outpost_connection_discovery() # pylint: disable=no-value-for-parameter
|
|
self.provider: ProxyProvider = ProxyProvider.objects.create(
|
|
name="test",
|
|
internal_host="http://localhost",
|
|
external_host="http://localhost",
|
|
authorization_flow=create_test_flow(),
|
|
)
|
|
self.service_connection = KubernetesServiceConnection.objects.first()
|
|
self.outpost: Outpost = Outpost.objects.create(
|
|
name="test",
|
|
type=OutpostType.PROXY,
|
|
service_connection=self.service_connection,
|
|
)
|
|
self.outpost.providers.add(self.provider)
|
|
self.outpost.config.kubernetes_json_patches = {
|
|
"deployment": [
|
|
{
|
|
"op": "add",
|
|
"path": "/spec/template/spec/containers/0/resources",
|
|
"value": {
|
|
"requests": {"cpu": "2000m", "memory": "2000Mi"},
|
|
"limits": {"cpu": "4000m", "memory": "8000Mi"},
|
|
},
|
|
}
|
|
]
|
|
}
|
|
self.outpost.providers.add(self.provider)
|
|
self.outpost.save()
|
|
|
|
@pytest.mark.timeout(120)
|
|
def test_deployment_reconciler(self):
|
|
"""test that deployment requires update"""
|
|
controller = ProxyKubernetesController(self.outpost, self.service_connection)
|
|
deployment_reconciler = DeploymentReconciler(controller)
|
|
|
|
self.assertIsNotNone(deployment_reconciler.retrieve())
|
|
|
|
config = self.outpost.config
|
|
config.kubernetes_replicas = 3
|
|
config.kubernetes_json_patches = {
|
|
"deployment": [
|
|
{
|
|
"op": "add",
|
|
"path": "/spec/template/spec/containers/0/resources",
|
|
"value": {
|
|
"requests": {"cpu": "1000m", "memory": "2000Mi"},
|
|
"limits": {"cpu": "2000m", "memory": "4000Mi"},
|
|
},
|
|
}
|
|
]
|
|
}
|
|
self.outpost.config = config
|
|
|
|
with self.assertRaises(NeedsUpdate):
|
|
deployment_reconciler.reconcile(
|
|
deployment_reconciler.retrieve(),
|
|
deployment_reconciler.get_reference_object(),
|
|
)
|
|
|
|
with CONFIG.patch("outposts.container_image_base", "test"):
|
|
with self.assertRaises(NeedsUpdate):
|
|
deployment_reconciler.reconcile(
|
|
deployment_reconciler.retrieve(),
|
|
deployment_reconciler.get_reference_object(),
|
|
)
|
|
|
|
deployment_reconciler.delete(deployment_reconciler.get_reference_object())
|
|
|
|
@pytest.mark.timeout(120)
|
|
def test_controller_rename(self):
|
|
"""test that objects get deleted and re-created with new names"""
|
|
controller = ProxyKubernetesController(self.outpost, self.service_connection)
|
|
|
|
self.assertIsNone(controller.up())
|
|
self.outpost.name = "foo"
|
|
self.assertIsNone(controller.up())
|
|
apps = AppsV1Api(controller.client)
|
|
with self.assertRaises(OpenApiException):
|
|
apps.read_namespaced_deployment("test", self.outpost.config.kubernetes_namespace)
|
|
controller.down()
|
|
|
|
@pytest.mark.timeout(120)
|
|
def test_controller_full_update(self):
|
|
"""Test an update that triggers all objects"""
|
|
controller = ProxyKubernetesController(self.outpost, self.service_connection)
|
|
|
|
self.assertIsNone(controller.up())
|
|
with patch(
|
|
"authentik.outposts.controllers.k8s.base.get_version", MagicMock(return_value="1234")
|
|
):
|
|
self.assertIsNone(controller.up())
|
|
deployment_reconciler = DeploymentReconciler(controller)
|
|
deployment = deployment_reconciler.retrieve()
|
|
self.assertEqual(deployment.metadata.labels["app.kubernetes.io/version"], "1234")
|
|
controller.down()
|