From 6138e101aa4a468f4fdc169b1fda64b6cd32a111 Mon Sep 17 00:00:00 2001
From: Maxim Trofimov <qwerty65k@mail.ru>
Date: Mon, 24 Apr 2023 20:38:13 +0300
Subject: [PATCH] add big tests

Signed-off-by: Maxim Trofimov <qwerty65k@mail.ru>
---
 pkg/storage/driver/sql_test.go | 49 +++++++++++++++++++++++++++++++++-
 1 file changed, 48 insertions(+), 1 deletion(-)

diff --git a/pkg/storage/driver/sql_test.go b/pkg/storage/driver/sql_test.go
index 6caeed77f..088c17076 100644
--- a/pkg/storage/driver/sql_test.go
+++ b/pkg/storage/driver/sql_test.go
@@ -562,7 +562,54 @@ func TestCheckAlreadyAppliedNotFind(t *testing.T) {
 	mock.ExpectCommit()
 
 	if sqlDriver.checkAlreadyApplied(testMigrations) {
-		t.Errorf("Did find init id: %v, that not exists", initID)
+		t.Errorf("Did find init id: %v, that does not exist", initID)
+	}
+
+}
+
+func TestCheckAlreadyAppliedBigFind(t *testing.T) {
+	sqlDriver, mock := newTestFixtureSQL(t)
+
+	testMigrations := []*migrate.Migration{{Id: "init1"}, {Id: "init2"}, {Id: "init3"}}
+	mock.
+		ExpectQuery("").
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "applied_at"}).
+				AddRow("1", time.Time{}).
+				AddRow("2", time.Time{}).
+				AddRow("init1", time.Time{}).
+				AddRow("init2", time.Time{}).
+				AddRow("3", time.Time{}).
+				AddRow("init3", time.Time{}).
+				AddRow("4", time.Time{}).
+				AddRow("5", time.Time{}))
+	mock.ExpectCommit()
+
+	if !sqlDriver.checkAlreadyApplied(testMigrations) {
+		t.Errorf("Did not find init ids, that exist")
+	}
+
+}
+
+func TestCheckAlreadyAppliedBigNotFind(t *testing.T) {
+	sqlDriver, mock := newTestFixtureSQL(t)
+
+	testMigrations := []*migrate.Migration{{Id: "init1"}, {Id: "init2"}, {Id: "init3"}}
+	mock.
+		ExpectQuery("").
+		WillReturnRows(
+			sqlmock.NewRows([]string{"id", "applied_at"}).
+				AddRow("1", time.Time{}).
+				AddRow("2", time.Time{}).
+				AddRow("init1", time.Time{}).
+				AddRow("3", time.Time{}).
+				AddRow("init2", time.Time{}).
+				AddRow("4", time.Time{}).
+				AddRow("5", time.Time{}))
+	mock.ExpectCommit()
+
+	if sqlDriver.checkAlreadyApplied(testMigrations) {
+		t.Errorf("Did not find init id: init3, that does not exist")
 	}
 
 }