summary refs log tree commit diff
path: root/nixos/tests/kafka_0_10.nix
blob: 6e7820f64bc4b3d848c475ccd30c76013ec6fd68 (plain) (blame)
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
import ./make-test.nix ({ pkgs, lib, ... } :
let
  kafkaPackage = pkgs.apacheKafka_0_10;
in {
  name = "kafka_0_10";
  meta = with pkgs.stdenv.lib.maintainers; {
    maintainers = [ nequissimus ];
  };

  nodes = {
    zookeeper1 = { config, ... }: {
      services.zookeeper = {
        enable = true;
      };

      networking.firewall.allowedTCPPorts = [ 2181 ];
    };
    kafka = { config, ... }: {
      services.apache-kafka = {
        enable = true;
        extraProperties = ''
          offsets.topic.replication.factor = 1
        '';
        package = kafkaPackage;
        zookeeper = "zookeeper1:2181";
      };

      networking.firewall.allowedTCPPorts = [ 9092 ];
      virtualisation.memorySize = 2048;
    };
  };

  testScript = ''
    startAll;

    $zookeeper1->waitForUnit("zookeeper");
    $zookeeper1->waitForUnit("network.target");
    $zookeeper1->waitForOpenPort(2181);

    $kafka->waitForUnit("apache-kafka");
    $kafka->waitForUnit("network.target");
    $kafka->waitForOpenPort(9092);

    $kafka->waitUntilSucceeds("${kafkaPackage}/bin/kafka-topics.sh --create --zookeeper zookeeper1:2181 --partitions 1 --replication-factor 1 --topic testtopic");
    $kafka->mustSucceed("echo 'test 1' | ${kafkaPackage}/bin/kafka-console-producer.sh --broker-list localhost:9092 --topic testtopic");
    $kafka->mustSucceed("${kafkaPackage}/bin/kafka-console-consumer.sh --bootstrap-server localhost:9092 --topic testtopic --from-beginning --max-messages 1 | grep 'test 1'");
  '';
})