From 3a7e71ef22c7fc224203c24331ac9f491310255a Mon Sep 17 00:00:00 2001
From: Kentaro Hayashi <hayashi@clear-code.com>
Date: Tue, 23 Jul 2024 11:40:43 +0900
Subject: [PATCH] rpm: add confluent (kafka) test case

test packaged rdkafka usability.

Signed-off-by: Kentaro Hayashi <hayashi@clear-code.com>
---
 .github/workflows/yum.yml            |  9 +++
 fluent-package/yum/confluent-test.sh | 99 ++++++++++++++++++++++++++++
 2 files changed, 108 insertions(+)
 create mode 100755 fluent-package/yum/confluent-test.sh

diff --git a/.github/workflows/yum.yml b/.github/workflows/yum.yml
index a64575ca4..14ba70058 100644
--- a/.github/workflows/yum.yml
+++ b/.github/workflows/yum.yml
@@ -75,6 +75,15 @@ jobs:
           --env CENTOS_STREAM=${{ matrix.centos-stream }} \
           ${{ matrix.test-docker-image }} \
           /fluentd/fluent-package/yum/serverspec-test.sh
+      - name: Confluent Test
+        run: |
+          mkdir -p .bundle
+          docker run \
+          --rm \
+          --tty \
+          --volume ${PWD}:/fluentd:ro \
+          ${{ matrix.test-docker-image }} \
+          /fluentd/fluent-package/yum/confluent-test.sh
       - name: Binstubs Test
         run: |
           mkdir -p .bundle
diff --git a/fluent-package/yum/confluent-test.sh b/fluent-package/yum/confluent-test.sh
new file mode 100755
index 000000000..39f844a9b
--- /dev/null
+++ b/fluent-package/yum/confluent-test.sh
@@ -0,0 +1,99 @@
+#!/bin/bash
+
+set -exu
+
+# Amazon Linux 2 system-release-cpe is:
+# cpe:2.3:o:amazon:amazon_linux:2
+# CentOS 7 system-release-cpe is:
+# cpe:/o:centos:centos:7
+# This means that column glitch exists.
+# So, we should remove before "o" character.
+
+distribution=$(cat /etc/system-release-cpe | awk '{print substr($0, index($1, "o"))}' | cut -d: -f2)
+version=$(cat /etc/system-release-cpe | awk '{print substr($0, index($1, "o"))}' | cut -d: -f4)
+
+case ${distribution} in
+  amazon)
+    case ${version} in
+      2)
+        DNF=yum
+        DISTRIBUTION_VERSION=${version}
+	${DNF} install -y java-17-amazon-corretto-headless
+	;;
+      2023)
+        DNF=dnf
+        DISTRIBUTION_VERSION=${version}
+	${DNF} install -y java-21-amazon-corretto-headless
+        ;;
+    esac
+    ;;
+  rocky|almalinux)
+    DNF=dnf
+    DISTRIBUTION_VERSION=$(echo ${version} | cut -d. -f1)
+    ${DNF} install -y java-21-openjdk-headless
+    ;;
+esac
+
+repositories_dir=/fluentd/fluent-package/yum/repositories
+ARCH=$(rpm --eval "%{_arch}")
+${DNF} install -y \
+  ${repositories_dir}/${distribution}/${DISTRIBUTION_VERSION}/${ARCH}/Packages/*.rpm
+
+fluentd --version
+
+/usr/sbin/fluent-gem install --no-document serverspec
+rpm --import https://packages.confluent.io/rpm/7.6/archive.key
+
+cat <<EOF > /etc/yum.repos.d/confluent.repo
+[Confluent]
+name=Confluent repository
+baseurl=https://packages.confluent.io/rpm/7.6
+gpgcheck=1
+gpgkey=https://packages.confluent.io/rpm/7.6/archive.key
+enabled=1
+
+[Confluent-Clients]
+name=Confluent Clients repository
+baseurl=https://packages.confluent.io/clients/rpm/centos/\$releasever/\$basearch
+gpgcheck=1
+gpgkey=https://packages.confluent.io/clients/rpm/archive.key
+enabled=1
+EOF
+
+${DNF} update && ${DNF} install -y confluent-platform nmap-ncat
+
+export KAFKA_OPTS=-Dzookeeper.4lw.commands.whitelist=ruok
+/usr/bin/zookeeper-server-start /etc/kafka/zookeeper.properties  &
+N_POLLING=30
+n=1
+while true ; do
+    sleep 1
+    status=$(echo ruok | nc localhost 2181)
+    if [ "$status" = "imok" ]; then
+	break
+    fi
+    n=$((n + 1))
+    if [ $n -ge $N_POLLING ]; then
+	echo "failed to get response from zookeeper-server"
+	exit 1
+    fi
+done
+/usr/bin/kafka-server-start /etc/kafka/server.properties &
+n=1
+while true ; do
+    sleep 1
+    status=$(/usr/bin/zookeeper-shell localhost:2181 ls /brokers/ids | sed -n 6p)
+    if [ "$status" = "[0]" ]; then
+	break
+    fi
+    n=$((n + 1))
+    if [ $n -ge $N_POLLING ]; then
+	echo "failed to get response from kafka-server"
+	exit 1
+    fi
+done
+/usr/bin/kafka-topics --create --bootstrap-server localhost:9092 --replication-factor 1 --partitions 1 --topic test
+export PATH=/opt/fluent/bin:$PATH
+export INSTALLATION_TEST=true
+cd /fluentd && rake serverspec:kafka
+/usr/sbin/fluentd -c /fluentd/serverspec/test.conf &