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
|
{ lib, ... }:
{
name = "tdarr";
meta = with lib.maintainers; {
maintainers = [ mistyttm ];
};
nodes.machine =
{ pkgs, ... }:
{
services.tdarr = {
enable = true;
server = {
serverPort = 9266;
webUIPort = 9265;
};
nodes = {
main = {
type = "mapped";
priority = -1;
pollInterval = 2000;
startPaused = false;
maxLogSizeMB = 10;
workers = {
transcodeCPU = 1;
healthcheckCPU = 1;
};
};
secondary = {
enable = false;
};
};
};
};
testScript = ''
import json
machine.wait_for_unit("tdarr-server.service")
machine.wait_for_unit("tdarr-node-main.service")
with subtest("disabled node should not have a service"):
machine.fail("systemctl is-enabled tdarr-node-secondary.service")
with subtest("data directories created with correct ownership"):
machine.succeed("test -d /var/lib/tdarr")
machine.succeed("stat -c '%U:%G' /var/lib/tdarr | grep -q 'tdarr:tdarr'")
with subtest("disabled node directory should not exist"):
machine.fail("test -d /var/lib/tdarr/nodes/secondary")
with subtest("server environment variables are set correctly"):
env = machine.succeed(
"systemctl show tdarr-server.service --property=Environment"
)
assert "serverPort=9266" in env, f"serverPort not found in: {env}"
assert "webUIPort=9265" in env, f"webUIPort not found in: {env}"
assert "serverIP=0.0.0.0" in env, f"serverIP not found in: {env}"
assert "serverBindIP=false" in env, f"serverBindIP not found in: {env}"
assert "serverDualStack=false" in env, f"serverDualStack not found in: {env}"
assert "openBrowser=false" in env, f"openBrowser not found in: {env}"
assert "auth=false" in env, f"auth not found in: {env}"
assert "maxLogSizeMB=10" in env, f"maxLogSizeMB not found in: {env}"
assert "cronPluginUpdate=" in env, f"cronPluginUpdate not found in: {env}"
with subtest("node environment variables are set correctly"):
env = machine.succeed(
"systemctl show tdarr-node-main.service --property=Environment"
)
assert "serverURL=http://127.0.0.1:9266" in env, f"serverURL not found in: {env}"
assert "nodeType=mapped" in env, f"nodeType not found in: {env}"
assert "priority=-1" in env, f"priority not found in: {env}"
assert "pollInterval=2000" in env, f"pollInterval not found in: {env}"
assert "startPaused=false" in env, f"startPaused not found in: {env}"
assert "maxLogSizeMB=10" in env, f"maxLogSizeMB not found in: {env}"
assert "transcodecpuWorkers=1" in env, f"transcodecpuWorkers not found in: {env}"
assert "healthcheckcpuWorkers=1" in env, f"healthcheckcpuWorkers not found in: {env}"
assert "transcodegpuWorkers=0" in env, f"transcodegpuWorkers not found in: {env}"
assert "healthcheckgpuWorkers=0" in env, f"healthcheckgpuWorkers not found in: {env}"
with subtest("custom ports are listening"):
machine.wait_for_open_port(9265)
machine.wait_for_open_port(9266)
with subtest("server reports healthy status"):
status = json.loads(machine.succeed("curl -sf http://localhost:9266/api/v2/status"))
assert "version" in status, f"version missing from status: {status}"
assert status.get("uptime", -1) >= 0, f"unexpected uptime in status: {status}"
with subtest("web UI serves HTML"):
html = machine.succeed("curl --fail http://localhost:9265/")
assert "<!DOCTYPE html>" in html or "<html" in html, f"web UI did not return HTML: {html[:200]}"
with subtest("node registers with server and reports correct config"):
machine.wait_until_succeeds(
"curl -sf http://localhost:9266/api/v2/get-nodes | grep -q 'main'"
)
response = machine.succeed("curl -sf http://localhost:9266/api/v2/get-nodes")
nodes = json.loads(response)
node = next((v for v in nodes.values() if "main" in v.get("nodeName", "")), None)
assert node is not None, f"node 'main' not found in: {nodes}"
node_config = node["config"]
node_worker_limits = node["workerLimits"]
# Worker limits
assert node_worker_limits.get("transcodecpu") == 1, f"unexpected transcodecpu worker count: {node}"
assert node_worker_limits.get("healthcheckcpu") == 1, f"unexpected healthcheckcpu worker count: {node}"
assert node_worker_limits.get("transcodegpu") == 0, f"unexpected transcodegpu worker count: {node}"
assert node_worker_limits.get("healthcheckgpu") == 0, f"unexpected healthcheckgpu worker count: {node}"
# Node config as registered with the server
assert "pathTranslators" in node_config, f"pathTranslators missing from node config: {node_config}"
assert node_config.get("nodeType") == "mapped", f"unexpected nodeType: {node_config}"
assert node_config.get("priority") == -1, f"unexpected priority: {node_config}"
assert node_config.get("pollInterval") == 2000, f"unexpected pollInterval: {node_config}"
assert node_config.get("startPaused") == False, f"unexpected startPaused: {node_config}"
assert node_config.get("maxLogSizeMB") == "10", f"unexpected maxLogSizeMB: {node_config}"
assert node_config.get("cronPluginUpdate") == "", f"unexpected cronPluginUpdate: {node_config}"
# Top-level node state
assert node.get("nodePaused") == False, f"unexpected nodePaused: {node}"
assert node.get("nodeTags") == "mapped", f"unexpected nodeTags: {node}"
'';
}
|
