From b09c6d56832eb1718c07d74abf3bc6ae3fe4e030 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Sun, 28 Apr 2024 14:36:04 +0200
Subject: Adding upstream version 1.1.0.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 tests/environment_test.go | 120 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 120 insertions(+)
 create mode 100644 tests/environment_test.go

(limited to 'tests/environment_test.go')

diff --git a/tests/environment_test.go b/tests/environment_test.go
new file mode 100644
index 0000000..b4cedbc
--- /dev/null
+++ b/tests/environment_test.go
@@ -0,0 +1,120 @@
+package icingadb_test
+
+import (
+	"encoding/hex"
+	"encoding/json"
+	"github.com/icinga/icinga-testing/services"
+	"github.com/icinga/icinga-testing/utils/eventually"
+	"github.com/jmoiron/sqlx"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"golang.org/x/sync/errgroup"
+	"sort"
+	"testing"
+	"time"
+)
+
+func TestMultipleEnvironments(t *testing.T) {
+	rdb := getDatabase(t)
+
+	numEnvs := 3
+	icinga2Instances := make([]services.Icinga2, numEnvs)
+
+	// Start numEnvs icinga2 instances with an icingadb instance each, all writing to the same SQL database.
+	var g errgroup.Group
+	for i := range icinga2Instances {
+		i := i
+
+		g.Go(func() error {
+			r := it.RedisServerT(t)
+			icinga2Instances[i] = it.Icinga2NodeT(t, "master")
+			icinga2Instances[i].EnableIcingaDb(r)
+			icinga2Instances[i].Reload()
+			it.IcingaDbInstanceT(t, r, rdb)
+			return nil
+		})
+	}
+	_ = g.Wait()
+
+	// Query the IcingaDB environment_id from each icinga2 instance.
+	var expectedEnvs []string
+	for _, instance := range icinga2Instances {
+		res, err := instance.ApiClient().GetJson("/v1/objects/icingadbs")
+		require.NoError(t, err, "requesting IcingaDB objects from API should succeed")
+		var objects ObjectsIcingaDBsResponse
+		err = json.NewDecoder(res.Body).Decode(&objects)
+		require.NoError(t, err, "requesting IcingaDB objects from API should succeed")
+		require.NotEmpty(t, objects.Results, "API response should return an IcingaDB object")
+		expectedEnvs = append(expectedEnvs, objects.Results[0].Attrs.EnvironmentId)
+	}
+
+	sort.Strings(expectedEnvs)
+	for i := 0; i < len(expectedEnvs)-1; i++ {
+		require.NotEqual(t, expectedEnvs[i], expectedEnvs[i+1], "all environment IDs should be distinct")
+	}
+
+	db, err := sqlx.Open(rdb.Driver(), rdb.DSN())
+	require.NoError(t, err, "SQL database open")
+	t.Cleanup(func() { _ = db.Close() })
+
+	t.Run("Table", func(t *testing.T) {
+		t.Parallel()
+
+		eventually.Assert(t, func(t require.TestingT) {
+			var query string
+			switch rdb.Driver() {
+			case "mysql":
+				query = "SELECT LOWER(HEX(id)), name FROM environment ORDER BY id"
+			case "postgres":
+				query = "SELECT LOWER(ENCODE(id, 'hex')), name FROM environment ORDER BY id"
+			default:
+				panic("unknown database driver")
+			}
+			rows, err := db.Query(query)
+			require.NoError(t, err, "SQL query")
+			defer rows.Close()
+
+			var gotEnvs []string
+			for rows.Next() {
+				var id, name string
+				err := rows.Scan(&id, &name)
+				require.NoError(t, err, "SQL scan")
+				require.Equal(t, id, name, "name should be initialized to the environment id")
+				gotEnvs = append(gotEnvs, id)
+			}
+
+			require.Equal(t, expectedEnvs, gotEnvs, "each environment should be present in the environments table")
+		}, 20*time.Second, 250*time.Millisecond)
+	})
+
+	t.Run("HA", func(t *testing.T) {
+		t.Parallel()
+
+		eventually.Assert(t, func(t require.TestingT) {
+			rows, err := db.Query("SELECT environment_id, COUNT(*) FROM icingadb_instance WHERE responsible = 'y' GROUP BY environment_id")
+			require.NoError(t, err, "SQL query")
+			defer rows.Close()
+
+			numRows := 0
+			for rows.Next() {
+				var env []byte
+				var count int
+				err := rows.Scan(&env, &count)
+				require.NoError(t, err, "SQL scan")
+
+				assert.LessOrEqualf(t, count, 1,
+					"environment %s must have at most one active instance", hex.EncodeToString(env))
+				numRows++
+			}
+			require.Equal(t, numEnvs, numRows, "each environment should have one active instance")
+		}, 20*time.Second, 250*time.Millisecond)
+	})
+}
+
+type ObjectsIcingaDBsResponse struct {
+	Results []struct {
+		Attrs struct {
+			EnvironmentId string `json:"environment_id"`
+		} `json:"attrs"`
+	} `json:"results"`
+}
-- 
cgit v1.2.3