1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
import click
import asyncio
import sys
import pathlib
import dagger
import uuid
import httpx
from nd import Distribution, NetdataInstaller, FeatureFlags, Endpoint, AgentContext
def run_async(func):
def wrapper(*args, **kwargs):
return asyncio.run(func(*args, **kwargs))
return wrapper
@run_async
async def simple_test():
config = dagger.Config(log_output=sys.stdout)
async with dagger.Connection(config) as client:
platform = dagger.Platform("linux/x86_64")
distro = Distribution("debian10")
repo_root = pathlib.Path("/netdata")
prefix_path = pathlib.Path("/opt/netdata")
installer = NetdataInstaller(
platform, distro, repo_root, prefix_path, FeatureFlags.DBEngine
)
api_key = uuid.uuid4()
#
# parent
#
parent_endpoint = Endpoint("parent1", 22000)
parent_ctx = AgentContext(
client, platform, distro, installer, parent_endpoint, api_key, True
)
parent_cmd = installer.prefix / "usr/sbin/netdata"
parent_args = [
parent_cmd.as_posix(),
"-D",
"-i",
"0.0.0.0",
"-p",
str(parent_endpoint.port),
]
parent_ctr = parent_ctx.build_container()
parent_ctr = parent_ctr.with_exec(parent_args)
parent_svc = parent_ctr.as_service()
#
# child
#
child_endpoint = Endpoint("child1", 21000)
child_ctx = AgentContext(
client, platform, distro, installer, child_endpoint, api_key, False
)
child_ctx.add_parent(parent_ctx)
child_cmd = installer.prefix / "usr/sbin/netdata"
child_args = [
child_cmd.as_posix(),
"-D",
"-i",
"0.0.0.0",
"-p",
str(child_endpoint.port),
]
child_ctr = child_ctx.build_container()
child_ctr = child_ctr.with_service_binding(parent_endpoint.hostname, parent_svc)
child_ctr = child_ctr.with_exec(child_args)
child_svc = child_ctr.as_service()
#
# endpoints
#
parent_tunnel, child_tunnel = await asyncio.gather(
client.host().tunnel(parent_svc, native=True).start(),
client.host().tunnel(child_svc, native=True).start(),
)
parent_endpoint, child_endpoint = await asyncio.gather(
parent_tunnel.endpoint(),
child_tunnel.endpoint(),
)
await asyncio.sleep(10)
#
# run tests
#
async with httpx.AsyncClient() as http:
resp = await http.get(f"http://{parent_endpoint}/api/v1/info")
#
# Check that the child was connected
#
jd = resp.json()
assert (
"hosts-available" in jd
), "Could not find 'host-available' key in api/v1/info"
assert jd["hosts-available"] == 2, "Child did not connect to parent"
#
# Check bearer protection
#
forbidden_urls = [
f"http://{parent_endpoint}/api/v2/bearer_protection",
f"http://{parent_endpoint}/api/v2/bearer_get_token",
]
for url in forbidden_urls:
async with httpx.AsyncClient() as http:
resp = await http.get(url)
assert (
resp.status_code == httpx.codes.UNAVAILABLE_FOR_LEGAL_REASONS
), "Bearer protection is broken"
@click.command(help="Run a simple parent/child test")
def test():
simple_test()
|
