Test created nginx configuration contents

tests/scenario/test_nginx.py

@@ -5,6 +5,7 @@
 import pytest
 
 from nginx import Nginx
+from tempo import Tempo
 from tempo_cluster import TempoClusterProvider
 
 logger = logging.getLogger(__name__)
@@ -96,3 +97,67 @@ def test_nginx_config_is_parsed_with_workers(
 
     prepared_config = nginx.config()
     assert isinstance(prepared_config, str)
+
+
+@pytest.mark.parametrize(
+    "addresses",
+    (
+        {"all": {"1.2.3.4"}},
+        {"all": {"1.2.3.4", "1.2.3.5"}},
+        {
+            "all": {"1.2.3.4"},
+            "distributor": {"1.2.3.5"},
+            "ingester": {"1.2.3.6"},
+            "querier": {"1.2.4.7"},
+            "query_frontend": {"1.2.5.1"},
+            "compactor": {"1.2.6.6"},
+            "metrics_generator": {"1.2.8.4"},
+        },
+        {
+            "distributor": {"1.2.3.5"},
+            "ingester": {"1.2.3.6"},
+            "querier": {"1.2.4.7"},
+            "query_frontend": {"1.2.5.1"},
+            "compactor": {"1.2.6.6"},
+            "metrics_generator": {"1.2.8.4"},
+        },
+    ),
+)
+def test_nginx_config_contains_upstreams_and_proxy_pass(
+    context, nginx_container, tempo_cluster_provider, addresses
+):
+    tempo_cluster_provider.gather_addresses_by_role = MagicMock(return_value=addresses)
+
+    unit = MagicMock()
+    unit.get_container = nginx_container
+    tempo_charm = MagicMock()
+    tempo_charm.unit = MagicMock(return_value=unit)
+
+    nginx = Nginx(tempo_charm, tempo_cluster_provider, "lolcathost")
+
+    prepared_config = nginx.config()
+
+    for role, addresses in addresses.items():
+        for address in addresses:
+            if role == "all":
+                _assert_config_per_role(Tempo.all_ports, address, prepared_config)
+            if role == "distributor":
+                _assert_config_per_role(Tempo.receiver_ports, address, prepared_config)
+            if role == "query-frontend":
+                _assert_config_per_role(Tempo.server_ports, address, prepared_config)
+
+
+def _assert_config_per_role(source_dict, address, prepared_config):
+    # as entire config is in a format that's hard to parse (and crossplane returns a string), we look for servers,
+    # upstreams and correct proxy/grpc_pass instructions.
+    for port in source_dict.values():
+        assert f"server {address}:{port};" in prepared_config
+        assert f"listen {port}" in prepared_config
+        assert f"listen [::]:{port}" in prepared_config
+    for protocol in source_dict.keys():
+        sanitised_protocol = protocol.replace("_", "-")
+        assert f"upstream {sanitised_protocol}" in prepared_config
+        if "grpc" in protocol:
+            assert f"grpc_pass grpcs://{sanitised_protocol}" in prepared_config
+        else:
+            assert f"proxy_pass https://{sanitised_protocol}" in prepared_config