Test sqlx.NewCluster for invalid input

yandex · Nov 21, 2020 · fffa446 · fffa446
1 parent bed25ff
commit fffa446
Showing 1 changed file with 25 additions and 15 deletions.
diff --git a/sqlx/cluster_test.go b/sqlx/cluster_test.go
@@ -28,19 +28,29 @@ import (
 )
 
 func TestNewCluster(t *testing.T) {
-	db, mock, err := sqlmock.New(sqlmock.MonitorPingsOption(true))
-	require.NoError(t, err)
-	require.NotNil(t, db)
-
-	mock.ExpectClose()
-	defer func() { assert.NoError(t, mock.ExpectationsWereMet()) }()
-
-	node := NewNode("fake.addr", sqlx.NewDb(db, "sqlmock"))
-	cl, err := NewCluster([]Node{node}, func(_ context.Context, _ *sql.DB) (bool, error) { return false, nil })
-	require.NoError(t, err)
-	require.NotNil(t, cl)
-	defer func() { require.NoError(t, cl.Close()) }()
-
-	require.Len(t, cl.Nodes(), 1)
-	require.Equal(t, node, cl.Nodes()[0])
+	checker := func(_ context.Context, _ *sql.DB) (bool, error) { return false, nil }
+
+	t.Run("Works", func(t *testing.T) {
+		db, mock, err := sqlmock.New(sqlmock.MonitorPingsOption(true))
+		require.NoError(t, err)
+		require.NotNil(t, db)
+
+		mock.ExpectClose()
+		defer func() { assert.NoError(t, mock.ExpectationsWereMet()) }()
+
+		node := NewNode("fake.addr", sqlx.NewDb(db, "sqlmock"))
+		cl, err := NewCluster([]Node{node}, checker)
+		require.NoError(t, err)
+		require.NotNil(t, cl)
+		defer func() { require.NoError(t, cl.Close()) }()
+
+		require.Len(t, cl.Nodes(), 1)
+		require.Equal(t, node, cl.Nodes()[0])
+	})
+
+	t.Run("Fails", func(t *testing.T) {
+		cl, err := NewCluster(nil, checker)
+		require.Error(t, err)
+		require.Nil(t, cl)
+	})
 }