Migrate tests from store/storetest/cluster_discovery_store.go to use testify (#12757)

Этот коммит содержится в:
Amine
2019-10-15 16:22:19 +02:00
коммит произвёл 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)