summaryrefslogtreecommitdiffstats
path: root/nixos/tests/kafka/base.nix
blob: f7bce7f79cf76d113f0e6f3b064c268e47e870bf (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
{
  pkgs,
  runTest,
  lib,
  ...
}:

with lib;

let
  makeKafkaTest =
    name:
    { kafkaPackage }:
    (runTest {
      inherit name;

      nodes = {
        kafka =
          { ... }:
          {
            services.apache-kafka = {
              enable = true;
              package = kafkaPackage;
              clusterId = "ak2fIHr4S8WWarOF_ODD0g";
              formatLogDirs = true;
              settings = {
                "offsets.topic.replication.factor" = 1;
                "log.dirs" = [
                  "/var/lib/kafka/logdir1"
                  "/var/lib/kafka/logdir2"
                ];
                "node.id" = 1;
                "process.roles" = [
                  "broker"
                  "controller"
                ];
                "listeners" = [
                  "PLAINTEXT://:9092"
                  "CONTROLLER://:9093"
                ];
                "listener.security.protocol.map" = [
                  "PLAINTEXT:PLAINTEXT"
                  "CONTROLLER:PLAINTEXT"
                ];
                "controller.quorum.voters" = [
                  "1@kafka:9093"
                ];
                "controller.listener.names" = [ "CONTROLLER" ];
              };
            };

            networking.firewall.allowedTCPPorts = [
              9092
              9093
            ];
            virtualisation.diskSize = 1024;
            # i686 tests: qemu-system-i386 can simulate max 2047MB RAM (not 2048)
            virtualisation.memorySize = 2047;
          };
      };

      testScript = ''
        start_all()

        kafka.wait_for_unit("default.target")
        kafka.wait_for_unit("apache-kafka.service")
        kafka.wait_for_open_port(9092)

        kafka.wait_until_succeeds(
            "${kafkaPackage}/bin/kafka-topics.sh --create "
            + "--bootstrap-server localhost:9092 --partitions 1 "
            + "--replication-factor 1 --topic testtopic"
        )
        kafka.succeed(
            "echo 'test 1' | "
            + "${kafkaPackage}/bin/kafka-console-producer.sh "
            + "--bootstrap-server localhost:9092 --topic testtopic"
        )
        assert "test 1" in kafka.succeed(
            "${kafkaPackage}/bin/kafka-console-consumer.sh "
            + "--bootstrap-server localhost:9092 --topic testtopic "
            + "--from-beginning --max-messages 1"
        )
      '';
    });

in
with pkgs;
{
  kafka_4_1 = makeKafkaTest "kafka_4_1" { kafkaPackage = apacheKafka_4_1; };
  kafka_4_2 = makeKafkaTest "kafka_4_2" { kafkaPackage = apacheKafka_4_2; };
  kafka_4_3 = makeKafkaTest "kafka_4_3" { kafkaPackage = apacheKafka_4_3; };
  kafka = makeKafkaTest "kafka" { kafkaPackage = apacheKafka; };
}