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="") == []