Migrate tests from store/storetest/cluster_discovery_store.go to use testify (#12757)
Этот коммит содержится в:
коммит произвёл
Miguel de la Cruz
родитель
0ba6a791ef
Коммит
d7d7216b0d
@@ -29,13 +29,11 @@ func testClusterDiscoveryStore(t *testing.T, ss store.Store) {
|
|||||||
Type: "test_test",
|
Type: "test_test",
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().Save(discovery); err != nil {
|
err := ss.ClusterDiscovery().Save(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().Cleanup(); err != nil {
|
err = ss.ClusterDiscovery().Cleanup()
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func testClusterDiscoveryStoreDelete(t *testing.T, ss store.Store) {
|
func testClusterDiscoveryStoreDelete(t *testing.T, ss store.Store) {
|
||||||
@@ -45,13 +43,11 @@ func testClusterDiscoveryStoreDelete(t *testing.T, ss store.Store) {
|
|||||||
Type: "test_test",
|
Type: "test_test",
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().Save(discovery); err != nil {
|
err := ss.ClusterDiscovery().Save(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := ss.ClusterDiscovery().Delete(discovery); err != nil {
|
_, err = ss.ClusterDiscovery().Delete(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func testClusterDiscoveryStoreLastPing(t *testing.T, ss store.Store) {
|
func testClusterDiscoveryStoreLastPing(t *testing.T, ss store.Store) {
|
||||||
@@ -61,29 +57,24 @@ func testClusterDiscoveryStoreLastPing(t *testing.T, ss store.Store) {
|
|||||||
Type: "test_test_lastPing" + model.NewId(),
|
Type: "test_test_lastPing" + model.NewId(),
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().Save(discovery); err != nil {
|
err := ss.ClusterDiscovery().Save(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().SetLastPingAt(discovery); err != nil {
|
err = ss.ClusterDiscovery().SetLastPingAt(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
ttime := model.GetMillis()
|
ttime := model.GetMillis()
|
||||||
|
|
||||||
time.Sleep(1 * time.Second)
|
time.Sleep(1 * time.Second)
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().SetLastPingAt(discovery); err != nil {
|
err = ss.ClusterDiscovery().SetLastPingAt(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
list, err := ss.ClusterDiscovery().GetAll(discovery.Type, "cluster_name_lastPing")
|
list, err := ss.ClusterDiscovery().GetAll(discovery.Type, "cluster_name_lastPing")
|
||||||
require.Nil(t, err)
|
require.Nil(t, err)
|
||||||
assert.Len(t, list, 1)
|
assert.Len(t, list, 1)
|
||||||
|
|
||||||
if list[0].LastPingAt-ttime < 500 {
|
require.Less(t, int64(500), list[0].LastPingAt-ttime)
|
||||||
t.Fatal("failed to set time")
|
|
||||||
}
|
|
||||||
|
|
||||||
discovery2 := &model.ClusterDiscovery{
|
discovery2 := &model.ClusterDiscovery{
|
||||||
ClusterName: "cluster_name_missing",
|
ClusterName: "cluster_name_missing",
|
||||||
@@ -91,9 +82,8 @@ func testClusterDiscoveryStoreLastPing(t *testing.T, ss store.Store) {
|
|||||||
Type: "test_test_missing",
|
Type: "test_test_missing",
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().SetLastPingAt(discovery2); err != nil {
|
err = ss.ClusterDiscovery().SetLastPingAt(discovery2)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func testClusterDiscoveryStoreExists(t *testing.T, ss store.Store) {
|
func testClusterDiscoveryStoreExists(t *testing.T, ss store.Store) {
|
||||||
@@ -103,9 +93,8 @@ func testClusterDiscoveryStoreExists(t *testing.T, ss store.Store) {
|
|||||||
Type: "test_test_Exists" + model.NewId(),
|
Type: "test_test_Exists" + model.NewId(),
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := ss.ClusterDiscovery().Save(discovery); err != nil {
|
err := ss.ClusterDiscovery().Save(discovery)
|
||||||
t.Fatal(err)
|
require.Nil(t, err)
|
||||||
}
|
|
||||||
|
|
||||||
val, err := ss.ClusterDiscovery().Exists(discovery)
|
val, err := ss.ClusterDiscovery().Exists(discovery)
|
||||||
require.Nil(t, err)
|
require.Nil(t, err)
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user