github.com/nhannv/mattermost-server@v5.11.1+incompatible/store/storetest/cluster_discovery_store.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package storetest

import (
	"testing"

	"time"

	"github.com/mattermost/mattermost-server/model"
	"github.com/mattermost/mattermost-server/store"
)

func TestClusterDiscoveryStore(t *testing.T, ss store.Store) {
	t.Run("", func(t *testing.T) { testClusterDiscoveryStore(t, ss) })
	t.Run("Delete", func(t *testing.T) { testClusterDiscoveryStoreDelete(t, ss) })
	t.Run("LastPing", func(t *testing.T) { testClusterDiscoveryStoreLastPing(t, ss) })
	t.Run("Exists", func(t *testing.T) { testClusterDiscoveryStoreExists(t, ss) })
	t.Run("ClusterDiscoveryGetStore", func(t *testing.T) { testClusterDiscoveryGetStore(t, ss) })
}

func testClusterDiscoveryStore(t *testing.T, ss store.Store) {
	discovery := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname" + model.NewId(),
		Type:        "test_test",
	}

	if result := <-ss.ClusterDiscovery().Save(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	if result := <-ss.ClusterDiscovery().Cleanup(); result.Err != nil {
		t.Fatal(result.Err)
	}
}

func testClusterDiscoveryStoreDelete(t *testing.T, ss store.Store) {
	discovery := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname" + model.NewId(),
		Type:        "test_test",
	}

	if result := <-ss.ClusterDiscovery().Save(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	if result := <-ss.ClusterDiscovery().Delete(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}
}

func testClusterDiscoveryStoreLastPing(t *testing.T, ss store.Store) {
	discovery := &model.ClusterDiscovery{
		ClusterName: "cluster_name_lastPing",
		Hostname:    "hostname" + model.NewId(),
		Type:        "test_test_lastPing" + model.NewId(),
	}

	if result := <-ss.ClusterDiscovery().Save(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	if result := <-ss.ClusterDiscovery().SetLastPingAt(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	ttime := model.GetMillis()

	time.Sleep(1 * time.Second)

	if result := <-ss.ClusterDiscovery().SetLastPingAt(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	if result := <-ss.ClusterDiscovery().GetAll(discovery.Type, "cluster_name_lastPing"); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		list := result.Data.([]*model.ClusterDiscovery)

		if len(list) != 1 {
			t.Fatal("should only be 1 items")
			return
		}

		if list[0].LastPingAt-ttime < 500 {
			t.Fatal("failed to set time")
		}
	}

	discovery2 := &model.ClusterDiscovery{
		ClusterName: "cluster_name_missing",
		Hostname:    "hostname" + model.NewId(),
		Type:        "test_test_missing",
	}

	if result := <-ss.ClusterDiscovery().SetLastPingAt(discovery2); result.Err != nil {
		t.Fatal(result.Err)
	}
}

func testClusterDiscoveryStoreExists(t *testing.T, ss store.Store) {
	discovery := &model.ClusterDiscovery{
		ClusterName: "cluster_name_Exists",
		Hostname:    "hostname" + model.NewId(),
		Type:        "test_test_Exists" + model.NewId(),
	}

	if result := <-ss.ClusterDiscovery().Save(discovery); result.Err != nil {
		t.Fatal(result.Err)
	}

	if result := <-ss.ClusterDiscovery().Exists(discovery); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		val := result.Data.(bool)
		if !val {
			t.Fatal("should be true")
		}
	}

	discovery.ClusterName = "cluster_name_Exists2"

	if result := <-ss.ClusterDiscovery().Exists(discovery); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		val := result.Data.(bool)
		if val {
			t.Fatal("should be true")
		}
	}
}

func testClusterDiscoveryGetStore(t *testing.T, ss store.Store) {
	testType1 := model.NewId()

	discovery1 := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname1",
		Type:        testType1,
	}
	store.Must(ss.ClusterDiscovery().Save(discovery1))

	discovery2 := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname2",
		Type:        testType1,
	}
	store.Must(ss.ClusterDiscovery().Save(discovery2))

	discovery3 := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname3",
		Type:        testType1,
		CreateAt:    1,
		LastPingAt:  1,
	}
	store.Must(ss.ClusterDiscovery().Save(discovery3))

	testType2 := model.NewId()

	discovery4 := &model.ClusterDiscovery{
		ClusterName: "cluster_name",
		Hostname:    "hostname1",
		Type:        testType2,
	}
	store.Must(ss.ClusterDiscovery().Save(discovery4))

	if result := <-ss.ClusterDiscovery().GetAll(testType1, "cluster_name"); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		list := result.Data.([]*model.ClusterDiscovery)

		if len(list) != 2 {
			t.Fatal("Should only have returned 2")
		}
	}

	if result := <-ss.ClusterDiscovery().GetAll(testType2, "cluster_name"); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		list := result.Data.([]*model.ClusterDiscovery)

		if len(list) != 1 {
			t.Fatal("Should only have returned 1")
		}
	}

	if result := <-ss.ClusterDiscovery().GetAll(model.NewId(), "cluster_name"); result.Err != nil {
		t.Fatal(result.Err)
	} else {
		list := result.Data.([]*model.ClusterDiscovery)

		if len(list) != 0 {
			t.Fatal("shouldn't be any")
		}
	}
}