From 199784f7116cd1949aacb6af0b3e1cd473227d75 Mon Sep 17 00:00:00 2001 From: Maxim Trofimov Date: Sat, 26 Aug 2023 20:09:06 +0300 Subject: [PATCH] fix conflict Signed-off-by: Maxim Trofimov --- pkg/storage/driver/sql_test.go | 36 ++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) diff --git a/pkg/storage/driver/sql_test.go b/pkg/storage/driver/sql_test.go index 4c0c7b668..6caeed77f 100644 --- a/pkg/storage/driver/sql_test.go +++ b/pkg/storage/driver/sql_test.go @@ -21,6 +21,7 @@ import ( "time" sqlmock "github.com/DATA-DOG/go-sqlmock" + migrate "github.com/rubenv/sql-migrate" rspb "helm.sh/helm/v3/pkg/release" ) @@ -530,3 +531,38 @@ func mockGetReleaseCustomLabels(mock sqlmock.Sqlmock, key string, namespace stri } eq.WillReturnRows(returnRows).RowsWillBeClosed() } + +func TestCheckAlreadyAppliedFind(t *testing.T) { + sqlDriver, mock := newTestFixtureSQL(t) + initID := "init" + testMigrations := []*migrate.Migration{{Id: initID}} + mock. + ExpectQuery(""). + WillReturnRows( + sqlmock.NewRows([]string{"id", "applied_at"}). + AddRow(initID, time.Time{})) + mock.ExpectCommit() + + if !sqlDriver.checkAlreadyApplied(testMigrations) { + t.Errorf("Did not find init id: %v", initID) + } + +} + +func TestCheckAlreadyAppliedNotFind(t *testing.T) { + sqlDriver, mock := newTestFixtureSQL(t) + initID := "init" + testMigrations := []*migrate.Migration{{Id: initID}} + mock. + ExpectQuery(""). + WillReturnRows( + sqlmock.NewRows([]string{"id", "applied_at"}). + AddRow("1", time.Time{}). + AddRow("2", time.Time{})) + mock.ExpectCommit() + + if sqlDriver.checkAlreadyApplied(testMigrations) { + t.Errorf("Did find init id: %v, that not exists", initID) + } + +}