def test_plugins(self):
proxy_services = {"tensorboard": {"port": 6006}, "notebook": {"port": 8888}}
assert (
len(
get_plugins_location_config(
resolver="", auth="", proxy_services=proxy_services
)
)
== 2
)
def test_plugins_dns_prefix(self):
proxy_services = {
"tensorboard": {
"port": 6006
},
"notebook": {
"port": 8888
}
}
settings.PROXIES_CONFIG.auth_enabled = True
settings.PROXIES_CONFIG.dns_use_resolver = True
expected = r"""
location ~ /tensorboard/proxy/([-_.:\w]+)/(.*) {
auth_request /auth/v1/;
auth_request_set $auth_status $upstream_status;
resolver coredns.kube-system.svc.cluster.local valid=5s;
rewrite_log on;
rewrite ^/tensorboard/proxy/([-_.:\w]+)/(.*) /tensorboard/proxy/$1/$2 break;
proxy_pass http://$1:6006;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_hide_header X-Frame-Options;
proxy_set_header Origin "";
proxy_buffering off;
}
location ~ /notebook/proxy/([-_.:\w]+)/(.*) {
auth_request /auth/v1/;
auth_request_set $auth_status $upstream_status;
resolver coredns.kube-system.svc.cluster.local valid=5s;
rewrite_log on;
rewrite ^/notebook/proxy/([-_.:\w]+)/(.*) /notebook/proxy/$1/$2 break;
proxy_pass http://$1:8888;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_hide_header X-Frame-Options;
proxy_set_header Origin "";
proxy_buffering off;
}
""" # noqa
settings.PROXIES_CONFIG.dns_prefix = "coredns.kube-system"
settings.PROXIES_CONFIG.dns_custom_cluster = "cluster.local"
assert get_dns_config() == "coredns.kube-system.svc.cluster.local"
resolver = get_resolver()
assert ("\n".join(
get_plugins_location_config(
resolver=resolver,
auth=get_auth_config(),
proxy_services=proxy_services,
)) == expected)
expected = r"""
location ~ /tensorboard/proxy/([-_.:\w]+)/(.*) {
auth_request /auth/v1/;
auth_request_set $auth_status $upstream_status;
resolver kube-dns.new-system.svc.new-dns valid=5s;
rewrite_log on;
rewrite ^/tensorboard/proxy/([-_.:\w]+)/(.*) /tensorboard/proxy/$1/$2 break;
proxy_pass http://$1:6006;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_hide_header X-Frame-Options;
proxy_set_header Origin "";
proxy_buffering off;
}
location ~ /notebook/proxy/([-_.:\w]+)/(.*) {
auth_request /auth/v1/;
auth_request_set $auth_status $upstream_status;
resolver kube-dns.new-system.svc.new-dns valid=5s;
rewrite_log on;
rewrite ^/notebook/proxy/([-_.:\w]+)/(.*) /notebook/proxy/$1/$2 break;
proxy_pass http://$1:8888;
proxy_http_version 1.1;
proxy_set_header Upgrade $http_upgrade;
proxy_set_header Connection "upgrade";
proxy_hide_header X-Frame-Options;
proxy_set_header Origin "";
proxy_buffering off;
}
""" # noqa
settings.PROXIES_CONFIG.dns_prefix = "kube-dns.new-system"
settings.PROXIES_CONFIG.dns_custom_cluster = "new-dns"
assert get_dns_config() == "kube-dns.new-system.svc.new-dns"
resolver = get_resolver()
assert ("\n".join(
get_plugins_location_config(
resolver=resolver,
auth=get_auth_config(),
proxy_services=proxy_services,
)) == expected)
def test_no_plugins(self):
assert get_plugins_location_config(resolver="", auth="") == []
