58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
#!/usr/bin/env python
|
|
from util import start_cloudflared
|
|
|
|
|
|
class TestConfig:
|
|
# tmp_path is a fixture provides a temporary directory unique to the test invocation
|
|
def test_validate_ingress_rules(self, tmp_path, component_tests_config):
|
|
extra_config = {
|
|
'ingress': [
|
|
{
|
|
"hostname": "example.com",
|
|
"service": "https://localhost:8000",
|
|
"originRequest": {
|
|
"originServerName": "test.example.com",
|
|
"caPool": "/etc/certs/ca.pem"
|
|
},
|
|
},
|
|
{
|
|
"hostname": "api.example.com",
|
|
"path": "login",
|
|
"service": "https://localhost:9000",
|
|
},
|
|
{
|
|
"hostname": "wss.example.com",
|
|
"service": "wss://localhost:8000",
|
|
},
|
|
{
|
|
"hostname": "ssh.example.com",
|
|
"service": "ssh://localhost:8000",
|
|
},
|
|
{"service": "http_status:404"}
|
|
],
|
|
}
|
|
config = component_tests_config(extra_config)
|
|
validate_args = ["ingress", "validate"]
|
|
_ = start_cloudflared(tmp_path, config, validate_args)
|
|
|
|
self.match_rule(tmp_path, config,
|
|
"http://example.com/index.html", 0)
|
|
self.match_rule(tmp_path, config,
|
|
"https://example.com/index.html", 0)
|
|
self.match_rule(tmp_path, config,
|
|
"https://api.example.com/login", 1)
|
|
self.match_rule(tmp_path, config,
|
|
"https://wss.example.com", 2)
|
|
self.match_rule(tmp_path, config,
|
|
"https://ssh.example.com", 3)
|
|
self.match_rule(tmp_path, config,
|
|
"https://api.example.com", 4)
|
|
|
|
# This is used to check that the command tunnel ingress url <url> matches rule number <rule_num>. Note that rule number uses 1-based indexing
|
|
|
|
def match_rule(self, tmp_path, config, url, rule_num):
|
|
args = ["ingress", "rule", url]
|
|
match_rule = start_cloudflared(tmp_path, config, args)
|
|
|
|
assert f"Matched rule #{rule_num}" .encode() in match_rule.stdout
|