extend test struct with new fields

go-vela · Jun 5, 2024 · 2826456 · 2826456
1 parent 0a49261
commit 2826456
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/database/database_test.go b/database/database_test.go
@@ -32,6 +32,10 @@ func TestDatabase_New(t *testing.T) {
 				ConnectionOpen:   20,
 				EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 				SkipCreation:     false,
+				LogLevel:         "info",
+				LogSkipNotFound:  true,
+				LogSlowThreshold: 100 * time.Millisecond,
+				LogShowSQL:       false,
 			},
 		},
 		{
@@ -46,6 +50,10 @@ func TestDatabase_New(t *testing.T) {
 				ConnectionOpen:   20,
 				EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 				SkipCreation:     false,
+				LogLevel:         "info",
+				LogSkipNotFound:  true,
+				LogSlowThreshold: 100 * time.Millisecond,
+				LogShowSQL:       false,
 			},
 		},
 		{
@@ -60,6 +68,10 @@ func TestDatabase_New(t *testing.T) {
 				ConnectionOpen:   20,
 				EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 				SkipCreation:     false,
+				LogLevel:         "info",
+				LogSkipNotFound:  true,
+				LogSlowThreshold: 100 * time.Millisecond,
+				LogShowSQL:       false,
 			},
 		},
 		{
@@ -74,6 +86,10 @@ func TestDatabase_New(t *testing.T) {
 				ConnectionOpen:   20,
 				EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 				SkipCreation:     false,
+				LogLevel:         "info",
+				LogSkipNotFound:  true,
+				LogSlowThreshold: 100 * time.Millisecond,
+				LogShowSQL:       false,
 			},
 		},
 	}
@@ -88,6 +104,10 @@ func TestDatabase_New(t *testing.T) {
 				WithConnectionIdle(test.config.ConnectionIdle),
 				WithConnectionOpen(test.config.ConnectionOpen),
 				WithDriver(test.config.Driver),
+				WithLogLevel(test.config.LogLevel),
+				WithLogShowSQL(test.config.LogShowSQL),
+				WithLogSkipNotFound(test.config.LogSkipNotFound),
+				WithLogSlowThreshold(test.config.LogSlowThreshold),
 				WithEncryptionKey(test.config.EncryptionKey),
 				WithSkipCreation(test.config.SkipCreation),
 			)
@@ -119,6 +139,10 @@ func testPostgres(t *testing.T) (*engine, sqlmock.Sqlmock) {
 			Driver:           "postgres",
 			EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 			SkipCreation:     false,
+			LogLevel:         "info",
+			LogSkipNotFound:  true,
+			LogSlowThreshold: 100 * time.Millisecond,
+			LogShowSQL:       false,
 		},
 		logger: logrus.NewEntry(logrus.StandardLogger()),
 	}
@@ -161,6 +185,10 @@ func testSqlite(t *testing.T) *engine {
 			Driver:           "sqlite3",
 			EncryptionKey:    "A1B2C3D4E5G6H7I8J9K0LMNOPQRSTUVW",
 			SkipCreation:     false,
+			LogLevel:         "info",
+			LogSkipNotFound:  true,
+			LogSlowThreshold: 100 * time.Millisecond,
+			LogShowSQL:       false,
 		},
 		logger: logrus.NewEntry(logrus.StandardLogger()),
 	}