nixos/tests/vector.nix


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

{ system ? builtins.currentSystem, config ? { }
, pkgs ? import ../.. { inherit system config; } }:

with import ../lib/testing-python.nix { inherit system pkgs; };
with pkgs.lib;

{
  test1 = makeTest {
    name = "vector-test1";
    meta.maintainers = [ pkgs.lib.maintainers.happysalada ];

    nodes.machine = { config, pkgs, ... }: {
      services.vector = {
        enable = true;
        journaldAccess = true;
        settings = {
          sources = {
            journald.type = "journald";

            vector_metrics.type = "internal_metrics";

            vector_logs.type = "internal_logs";
          };

          sinks = {
            file = {
              type = "file";
              inputs = [ "journald" "vector_logs" ];
              path = "/var/lib/vector/logs.log";
              encoding = { codec = "json"; };
            };

            prometheus_exporter = {
              type = "prometheus_exporter";
              inputs = [ "vector_metrics" ];
              address = "[::]:9598";
            };
          };
        };
      };
    };

    # ensure vector is forwarding the messages appropriately
    testScript = ''
      machine.wait_for_unit("vector.service")
      machine.wait_for_open_port(9598)
      machine.wait_until_succeeds("curl -sSf http://localhost:9598/metrics | grep vector_build_info")
      machine.wait_until_succeeds("curl -sSf http://localhost:9598/metrics | grep vector_component_received_bytes_total | grep journald")
      machine.wait_until_succeeds("curl -sSf http://localhost:9598/metrics | grep vector_utilization | grep prometheus_exporter")
      machine.wait_for_file("/var/lib/vector/logs.log")
    '';
  };
}