test: fix dsn will never be check with expectedDSN

Author: drivebyer

pkg/controller/mysqluser/mysqluser_controller_test.go

@@ -158,7 +158,7 @@ var _ = Describe("MySQL user controller", func() {
 					return nil
 				}
 
-				fakeSQL.AssertDSN(expectedDSN)
+				fakeSQL.AddExpectedDSN(expectedDSN)
 				// the create user runs twice
 				fakeSQL.AddExpectedCalls(expectedQueryRunnerCall)
 				fakeSQL.AddExpectedCalls(expectedQueryRunnerCall)
@@ -262,7 +262,7 @@ var _ = Describe("MySQL user controller", func() {
 					return nil
 				}
 
-				fakeSQL.AssertDSN(expectedDSN)
+				fakeSQL.AddExpectedDSN(expectedDSN)
 				fakeSQL.AddExpectedCalls(expectedQueryRunnerCall)
 				fakeSQL.AddExpectedCalls(expectedQueryRunnerCall)
 
@@ -490,7 +490,7 @@ var _ = Describe("MySQL user controller", func() {
 					return deletionResult
 				}
 
-				fakeSQL.AssertDSN(expectedDSN)
+				fakeSQL.AddExpectedDSN(expectedDSN)
 				fakeSQL.AddExpectedCalls(expectedQueryRunnerCall)
 
 				Expect(c.Delete(context.TODO(), user.Unwrap())).To(Succeed())

pkg/internal/mysql/fake/mysql.go

@@ -47,6 +47,13 @@ func (qr *SQLRunner) AddExpectedCalls(expectedCalls ...SQLCall) {
 	qr.expectedCalls = append(qr.expectedCalls, expectedCalls...)
 }
 
+// AddExpectedDSN add a expected DSN to the fake SQLRunner for later assert.
+func (qr *SQLRunner) AddExpectedDSN(expectedDSN string) {
+	qr.lock.Lock()
+	defer qr.lock.Unlock()
+	qr.expectedDSN = &expectedDSN
+}
+
 // PurgeExpectedCalls removes all the expected query runner calls
 func (qr *SQLRunner) PurgeExpectedCalls() {
 	qr.lock.Lock()