fix: reorder

gofiber · Sep 19, 2024 · 153c837 · 153c837
1 parent efbc89f
commit 153c837
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/clickhouse/clickhouse_test.go b/clickhouse/clickhouse_test.go
@@ -77,9 +77,9 @@ func Test_Connection(t *testing.T) {
 		Table:  "test_table",
 		Clean:  true,
 	})
-	defer client.Close()
-
 	require.NoError(t, err)
+
+	defer client.Close()
 }
 
 func Test_Set(t *testing.T) {
@@ -197,6 +197,17 @@ func Test_Reset(t *testing.T) {
 	assert.Equal(t, []byte{}, value)
 }
 
+func TestClose_ShouldReturn_NoError(t *testing.T) {
+	client, err := getTestConnection(t, Config{
+		Engine: Memory,
+		Table:  "test_table",
+		Clean:  true,
+	})
+	require.NoError(t, err)
+
+	require.NoError(t, client.Close())
+}
+
 func Benchmark_Clickhouse_Set(b *testing.B) {
 	b.ReportAllocs()
 	b.ResetTimer()