summaryrefslogtreecommitdiffstats
path: root/nixos/tests/prometheus/remote-write.nix
blob: 23458c737e1acf5d16d420fea95fbda306e60113 (plain)
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
{
  name = "prometheus-remote-write";

  nodes = {
    receiver =
      { config, pkgs, ... }:
      {
        environment.systemPackages = [ pkgs.jq ];

        networking.firewall.allowedTCPPorts = [ config.services.prometheus.port ];

        services.prometheus = {
          enable = true;
          globalConfig.scrape_interval = "2s";

          extraFlags = [ "--web.enable-remote-write-receiver" ];
        };
      };

    prometheus =
      { config, pkgs, ... }:
      {
        environment.systemPackages = [ pkgs.jq ];

        networking.firewall.allowedTCPPorts = [ config.services.prometheus.port ];

        services.prometheus = {
          enable = true;
          globalConfig.scrape_interval = "2s";

          remoteWrite = [ { url = "http://receiver:9090/api/v1/write"; } ];

          scrapeConfigs = [
            {
              job_name = "node";
              static_configs = [
                { targets = [ "node:${toString config.services.prometheus.exporters.node.port}" ]; }
              ];
            }
          ];
        };
      };

    node =
      { config, pkgs, ... }:
      {
        services.prometheus.exporters.node = {
          enable = true;
          openFirewall = true;
        };
      };
  };

  testScript = ''
    node.wait_for_unit("prometheus-node-exporter")
    node.wait_for_open_port(9100)

    for machine in prometheus, receiver:
      machine.wait_for_unit("prometheus")
      machine.wait_for_open_port(9090)

    # Verify both servers got the same data from the exporter
    for machine in prometheus, receiver:
      machine.wait_until_succeeds(
        "curl -sf 'http://127.0.0.1:9090/api/v1/query?query=node_exporter_build_info\{instance=\"node:9100\"\}' | "
        + "jq '.data.result[0].value[1]' | grep '\"1\"'"
      )
  '';
}