summaryrefslogtreecommitdiffstats
path: root/nixos/tests/prometheus/alertmanager.nix
blob: 794bbb8aab90224f208e833c8f79b8ddd71307bb (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
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
{ pkgs, ... }:

{
  name = "prometheus-alertmanager";

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

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

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

          alertmanagers = [
            {
              scheme = "http";
              static_configs = [
                { targets = [ "alertmanager:${toString config.services.prometheus.alertmanager.port}" ]; }
              ];
            }
          ];

          rules = [
            ''
              groups:
                - name: test
                  rules:
                    - alert: InstanceDown
                      expr: up == 0
                      for: 5s
                      labels:
                        severity: page
                      annotations:
                        summary: "Instance {{ $labels.instance }} down"
            ''
          ];

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

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

          configuration = {
            global = {
              resolve_timeout = "1m";
            };

            route = {
              # Root route node
              receiver = "test";
              group_by = [ "..." ];
              continue = false;
              group_wait = "1s";
              group_interval = "15s";
              repeat_interval = "24h";
            };

            receivers = [
              {
                name = "test";
                webhook_configs = [
                  {
                    url = "http://logger:6725";
                    send_resolved = true;
                    max_alerts = 0;
                  }
                ];
              }
            ];
          };
        };
      };

    logger =
      { config, pkgs, ... }:
      {
        networking.firewall.allowedTCPPorts = [ 6725 ];

        services.prometheus.alertmanagerWebhookLogger.enable = true;
      };
  };

  testScript = ''
    alertmanager.wait_for_unit("alertmanager")
    alertmanager.wait_for_open_port(9093)
    alertmanager.wait_until_succeeds("curl -s http://127.0.0.1:9093/-/ready")
    #alertmanager.wait_until_succeeds("journalctl -o cat -u alertmanager.service | grep 'version=${pkgs.prometheus-alertmanager.version}'")

    logger.wait_for_unit("alertmanager-webhook-logger")
    logger.wait_for_open_port(6725)

    prometheus.wait_for_unit("prometheus")
    prometheus.wait_for_open_port(9090)

    prometheus.wait_until_succeeds(
      "curl -sf 'http://127.0.0.1:9090/api/v1/query?query=count(up\{job=\"alertmanager\"\}==1)' | "
      + "jq '.data.result[0].value[1]' | grep '\"1\"'"
    )

    prometheus.wait_until_succeeds(
      "curl -sf 'http://127.0.0.1:9090/api/v1/query?query=sum(alertmanager_build_info)%20by%20(version)' | "
      + "jq '.data.result[0].metric.version' | grep '\"${pkgs.prometheus-alertmanager.version}\"'"
    )

    prometheus.wait_until_succeeds(
      "curl -sf 'http://127.0.0.1:9090/api/v1/query?query=count(up\{job=\"node\"\}!=1)' | "
      + "jq '.data.result[0].value[1]' | grep '\"1\"'"
    )

    prometheus.wait_until_succeeds(
      "curl -sf 'http://127.0.0.1:9090/api/v1/query?query=alertmanager_notifications_total\{integration=\"webhook\"\}' | "
      + "jq '.data.result[0].value[1]' | grep -v '\"0\"'"
    )

    logger.wait_until_succeeds(
      "journalctl -o cat -u alertmanager-webhook-logger.service | grep '\"alertname\":\"InstanceDown\"'"
    )

    logger.log(logger.succeed("systemd-analyze security alertmanager-webhook-logger.service | grep -v '✓'"))

    alertmanager.log(alertmanager.succeed("systemd-analyze security alertmanager.service | grep -v '✓'"))
  '';
}