104 lines
2.6 KiB
Nix
104 lines
2.6 KiB
Nix
# Primarily reference the implementation of <nixos/tests/prometheus/remote-write.nix>
|
|||
import ../make-test-python.nix (
|
|||
{
|
|||
lib,
|
|||
pkgs,
|
|||
...
|
|||
}:
|
|||
let
|
|||
username = "vmtest";
|
|||
password = "fsddfy8233rb"; # random string
|
|||
passwordFile = pkgs.writeText "password-file" password;
|
|||
in
|
|||
{
|
|||
name = "victoriametrics-remote-write";
|
|||
meta = with pkgs.lib.maintainers; {
|
|||
maintainers = [
|
|||
yorickvp
|
|||
ryan4yin
|
|||
];
|
|||
};
|
|||
|
|||
nodes = {
|
|||
victoriametrics =
|
|||
{
|
|||
config,
|
|||
pkgs,
|
|||
...
|
|||
}:
|
|||
{
|
|||
environment.systemPackages = [ pkgs.jq ];
|
|||
networking.firewall.allowedTCPPorts = [ 8428 ];
|
|||
services.victoriametrics = {
|
|||
enable = true;
|
|||
extraOptions = [
|
|||
"-httpAuth.username=${username}"
|
|||
"-httpAuth.password=file://${toString passwordFile}"
|
|||
];
|
|||
};
|
|||
};
|
|||
|
|||
vmagent =
|
|||
{
|
|||
config,
|
|||
pkgs,
|
|||
...
|
|||
}:
|
|||
{
|
|||
environment.systemPackages = [ pkgs.jq ];
|
|||
services.vmagent = {
|
|||
enable = true;
|
|||
remoteWrite = {
|
|||
url = "http://victoriametrics:8428/api/v1/write";
|
|||
basicAuthUsername = username;
|
|||
basicAuthPasswordFile = toString passwordFile;
|
|||
};
|
|||
|
|||
prometheusConfig = {
|
|||
global = {
|
|||
scrape_interval = "2s";
|
|||
};
|
|||
scrape_configs = [
|
|||
{
|
|||
job_name = "node";
|
|||
static_configs = [
|
|||
{
|
|||
targets = [
|
|||
"node:${toString config.services.prometheus.exporters.node.port}"
|
|||
];
|
|||
}
|
|||
];
|
|||
}
|
|||
];
|
|||
};
|
|||
};
|
|||
};
|
|||
|
|||
node =
|
|||
{ ... }:
|
|||
{
|
|||
services.prometheus.exporters.node = {
|
|||
enable = true;
|
|||
openFirewall = true;
|
|||
};
|
|||
};
|
|||
};
|
|||
|
|||
testScript = ''
|
|||
node.wait_for_unit("prometheus-node-exporter")
|
|||
node.wait_for_open_port(9100)
|
|||
|
|||
victoriametrics.wait_for_unit("victoriametrics")
|
|||
victoriametrics.wait_for_open_port(8428)
|
|||
|
|||
vmagent.wait_for_unit("vmagent")
|
|||
|
|||
# check remote write
|
|||
victoriametrics.wait_until_succeeds(
|
|||
"curl --user '${username}:${password}' -sf 'http://localhost:8428/api/v1/query?query=node_exporter_build_info\{instance=\"node:9100\"\}' | "
|
|||
+ "jq '.data.result[0].value[1]' | grep '\"1\"'"
|
|||
)
|
|||
'';
|
|||
}
|
|||
)
|
