mirror of
https://github.com/helm/helm.git
synced 2026-02-14 00:05:36 -05:00
e711488970
This is part of HIP 20 which provides a means to have v3 charts that live alongside v2 charts while having breaking changes. The plan is to have a different release object for v3 chart instances for at least a couple reasons: 1. So that the chart object on the release can be fundamentally different. 2. So that Helm v3 does not detect or try to work with instances of charts whose apiVersion it does not know about. Note: it is expected that Helm v3 usage will be used long after the Helm project no longer supports it. 5 years after Helm v2 had reached end-of-life there was still usage of it. Note: The release util package is separate from the versioned elements as it is planned to use generics to handle multiple release object versions. Signed-off-by: Matt Farina <matt.farina@suse.com>
592 lines
17 KiB
Go
592 lines
17 KiB
Go
/*
|
|
Copyright The Helm Authors.
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package driver
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"regexp"
|
|
"testing"
|
|
"time"
|
|
|
|
sqlmock "github.com/DATA-DOG/go-sqlmock"
|
|
migrate "github.com/rubenv/sql-migrate"
|
|
|
|
rspb "helm.sh/helm/v4/pkg/release/v1"
|
|
)
|
|
|
|
func TestSQLName(t *testing.T) {
|
|
sqlDriver, _ := newTestFixtureSQL(t)
|
|
if sqlDriver.Name() != SQLDriverName {
|
|
t.Errorf("Expected name to be %s, got %s", SQLDriverName, sqlDriver.Name())
|
|
}
|
|
}
|
|
|
|
func TestSQLGet(t *testing.T) {
|
|
vers := int(1)
|
|
name := "smug-pigeon"
|
|
namespace := "default"
|
|
key := testKey(name, vers)
|
|
rel := releaseStub(name, vers, namespace, rspb.StatusDeployed)
|
|
|
|
body, _ := encodeRelease(rel)
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
|
|
query := fmt.Sprintf(
|
|
regexp.QuoteMeta("SELECT %s FROM %s WHERE %s = $1 AND %s = $2"),
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.
|
|
ExpectQuery(query).
|
|
WithArgs(key, namespace).
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
}).AddRow(
|
|
body,
|
|
),
|
|
).RowsWillBeClosed()
|
|
|
|
mockGetReleaseCustomLabels(mock, key, namespace, rel.Labels)
|
|
|
|
got, err := sqlDriver.Get(key)
|
|
if err != nil {
|
|
t.Fatalf("Failed to get release: %v", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(rel, got) {
|
|
t.Errorf("Expected release {%v}, got {%v}", rel, got)
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSQLList(t *testing.T) {
|
|
releases := []*rspb.Release{}
|
|
releases = append(releases, releaseStub("key-1", 1, "default", rspb.StatusUninstalled))
|
|
releases = append(releases, releaseStub("key-2", 1, "default", rspb.StatusUninstalled))
|
|
releases = append(releases, releaseStub("key-3", 1, "default", rspb.StatusDeployed))
|
|
releases = append(releases, releaseStub("key-4", 1, "default", rspb.StatusDeployed))
|
|
releases = append(releases, releaseStub("key-5", 1, "default", rspb.StatusSuperseded))
|
|
releases = append(releases, releaseStub("key-6", 1, "default", rspb.StatusSuperseded))
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
query := fmt.Sprintf(
|
|
"SELECT %s, %s, %s FROM %s WHERE %s = $1 AND %s = $2",
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
rows := mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
})
|
|
for _, r := range releases {
|
|
body, _ := encodeRelease(r)
|
|
rows.AddRow(body)
|
|
}
|
|
mock.
|
|
ExpectQuery(regexp.QuoteMeta(query)).
|
|
WithArgs(sqlReleaseDefaultOwner, sqlDriver.namespace).
|
|
WillReturnRows(rows).RowsWillBeClosed()
|
|
|
|
for _, r := range releases {
|
|
mockGetReleaseCustomLabels(mock, "", r.Namespace, r.Labels)
|
|
}
|
|
}
|
|
|
|
// list all deleted releases
|
|
del, err := sqlDriver.List(func(rel *rspb.Release) bool {
|
|
return rel.Info.Status == rspb.StatusUninstalled
|
|
})
|
|
// check
|
|
if err != nil {
|
|
t.Errorf("Failed to list deleted: %v", err)
|
|
}
|
|
if len(del) != 2 {
|
|
t.Errorf("Expected 2 deleted, got %d:\n%v\n", len(del), del)
|
|
}
|
|
|
|
// list all deployed releases
|
|
dpl, err := sqlDriver.List(func(rel *rspb.Release) bool {
|
|
return rel.Info.Status == rspb.StatusDeployed
|
|
})
|
|
// check
|
|
if err != nil {
|
|
t.Errorf("Failed to list deployed: %v", err)
|
|
}
|
|
if len(dpl) != 2 {
|
|
t.Errorf("Expected 2 deployed, got %d:\n%v\n", len(dpl), dpl)
|
|
}
|
|
|
|
// list all superseded releases
|
|
ssd, err := sqlDriver.List(func(rel *rspb.Release) bool {
|
|
return rel.Info.Status == rspb.StatusSuperseded
|
|
})
|
|
// check
|
|
if err != nil {
|
|
t.Errorf("Failed to list superseded: %v", err)
|
|
}
|
|
if len(ssd) != 2 {
|
|
t.Errorf("Expected 2 superseded, got %d:\n%v\n", len(ssd), ssd)
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
|
|
// Check if release having both system and custom labels, this is needed to ensure that selector filtering would work.
|
|
rls := ssd[0]
|
|
_, ok := rls.Labels["name"]
|
|
if !ok {
|
|
t.Fatalf("Expected 'name' label in results, actual %v", rls.Labels)
|
|
}
|
|
_, ok = rls.Labels["key1"]
|
|
if !ok {
|
|
t.Fatalf("Expected 'key1' label in results, actual %v", rls.Labels)
|
|
}
|
|
}
|
|
|
|
func TestSqlCreate(t *testing.T) {
|
|
vers := 1
|
|
name := "smug-pigeon"
|
|
namespace := "default"
|
|
key := testKey(name, vers)
|
|
rel := releaseStub(name, vers, namespace, rspb.StatusDeployed)
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
body, _ := encodeRelease(rel)
|
|
|
|
query := fmt.Sprintf(
|
|
"INSERT INTO %s (%s,%s,%s,%s,%s,%s,%s,%s,%s) VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9)",
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableTypeColumn,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableNameColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
sqlReleaseTableVersionColumn,
|
|
sqlReleaseTableStatusColumn,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableCreatedAtColumn,
|
|
)
|
|
|
|
mock.ExpectBegin()
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(query)).
|
|
WithArgs(key, sqlReleaseDefaultType, body, rel.Name, rel.Namespace, int(rel.Version), rel.Info.Status.String(), sqlReleaseDefaultOwner, int(time.Now().Unix())).
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
|
|
labelsQuery := fmt.Sprintf(
|
|
"INSERT INTO %s (%s,%s,%s,%s) VALUES ($1,$2,$3,$4)",
|
|
sqlCustomLabelsTableName,
|
|
sqlCustomLabelsTableReleaseKeyColumn,
|
|
sqlCustomLabelsTableReleaseNamespaceColumn,
|
|
sqlCustomLabelsTableKeyColumn,
|
|
sqlCustomLabelsTableValueColumn,
|
|
)
|
|
|
|
mock.MatchExpectationsInOrder(false)
|
|
for k, v := range filterSystemLabels(rel.Labels) {
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(labelsQuery)).
|
|
WithArgs(key, rel.Namespace, k, v).
|
|
WillReturnResult(sqlmock.NewResult(1, 1))
|
|
}
|
|
mock.ExpectCommit()
|
|
|
|
if err := sqlDriver.Create(key, rel); err != nil {
|
|
t.Fatalf("failed to create release with key %s: %v", key, err)
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSqlCreateAlreadyExists(t *testing.T) {
|
|
vers := 1
|
|
name := "smug-pigeon"
|
|
namespace := "default"
|
|
key := testKey(name, vers)
|
|
rel := releaseStub(name, vers, namespace, rspb.StatusDeployed)
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
body, _ := encodeRelease(rel)
|
|
|
|
insertQuery := fmt.Sprintf(
|
|
"INSERT INTO %s (%s,%s,%s,%s,%s,%s,%s,%s,%s) VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9)",
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableTypeColumn,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableNameColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
sqlReleaseTableVersionColumn,
|
|
sqlReleaseTableStatusColumn,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableCreatedAtColumn,
|
|
)
|
|
|
|
// Insert fails (primary key already exists)
|
|
mock.ExpectBegin()
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(insertQuery)).
|
|
WithArgs(key, sqlReleaseDefaultType, body, rel.Name, rel.Namespace, int(rel.Version), rel.Info.Status.String(), sqlReleaseDefaultOwner, int(time.Now().Unix())).
|
|
WillReturnError(fmt.Errorf("dialect dependent SQL error"))
|
|
|
|
selectQuery := fmt.Sprintf(
|
|
regexp.QuoteMeta("SELECT %s FROM %s WHERE %s = $1 AND %s = $2"),
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
// Let's check that we do make sure the error is due to a release already existing
|
|
mock.
|
|
ExpectQuery(selectQuery).
|
|
WithArgs(key, namespace).
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableKeyColumn,
|
|
}).AddRow(
|
|
key,
|
|
),
|
|
).RowsWillBeClosed()
|
|
mock.ExpectRollback()
|
|
|
|
if err := sqlDriver.Create(key, rel); err == nil {
|
|
t.Fatalf("failed to create release with key %s: %v", key, err)
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSqlUpdate(t *testing.T) {
|
|
vers := 1
|
|
name := "smug-pigeon"
|
|
namespace := "default"
|
|
key := testKey(name, vers)
|
|
rel := releaseStub(name, vers, namespace, rspb.StatusDeployed)
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
body, _ := encodeRelease(rel)
|
|
|
|
query := fmt.Sprintf(
|
|
"UPDATE %s SET %s = $1, %s = $2, %s = $3, %s = $4, %s = $5, %s = $6 WHERE %s = $7 AND %s = $8",
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableNameColumn,
|
|
sqlReleaseTableVersionColumn,
|
|
sqlReleaseTableStatusColumn,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableModifiedAtColumn,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(query)).
|
|
WithArgs(body, rel.Name, int(rel.Version), rel.Info.Status.String(), sqlReleaseDefaultOwner, int(time.Now().Unix()), key, namespace).
|
|
WillReturnResult(sqlmock.NewResult(0, 1))
|
|
|
|
if err := sqlDriver.Update(key, rel); err != nil {
|
|
t.Fatalf("failed to update release with key %s: %v", key, err)
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSqlQuery(t *testing.T) {
|
|
// Reflect actual use cases in ../storage.go
|
|
labelSetUnknown := map[string]string{
|
|
"name": "smug-pigeon",
|
|
"owner": sqlReleaseDefaultOwner,
|
|
"status": "unknown",
|
|
}
|
|
labelSetDeployed := map[string]string{
|
|
"name": "smug-pigeon",
|
|
"owner": sqlReleaseDefaultOwner,
|
|
"status": "deployed",
|
|
}
|
|
labelSetAll := map[string]string{
|
|
"name": "smug-pigeon",
|
|
"owner": sqlReleaseDefaultOwner,
|
|
}
|
|
|
|
supersededRelease := releaseStub("smug-pigeon", 1, "default", rspb.StatusSuperseded)
|
|
supersededReleaseBody, _ := encodeRelease(supersededRelease)
|
|
deployedRelease := releaseStub("smug-pigeon", 2, "default", rspb.StatusDeployed)
|
|
deployedReleaseBody, _ := encodeRelease(deployedRelease)
|
|
|
|
// Let's actually start our test
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
|
|
query := fmt.Sprintf(
|
|
"SELECT %s, %s, %s FROM %s WHERE %s = $1 AND %s = $2 AND %s = $3 AND %s = $4",
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableNameColumn,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableStatusColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.
|
|
ExpectQuery(regexp.QuoteMeta(query)).
|
|
WithArgs("smug-pigeon", sqlReleaseDefaultOwner, "unknown", "default").
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
}),
|
|
).RowsWillBeClosed()
|
|
|
|
mock.
|
|
ExpectQuery(regexp.QuoteMeta(query)).
|
|
WithArgs("smug-pigeon", sqlReleaseDefaultOwner, "deployed", "default").
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
}).AddRow(
|
|
deployedReleaseBody,
|
|
),
|
|
).RowsWillBeClosed()
|
|
|
|
mockGetReleaseCustomLabels(mock, "", deployedRelease.Namespace, deployedRelease.Labels)
|
|
|
|
query = fmt.Sprintf(
|
|
"SELECT %s, %s, %s FROM %s WHERE %s = $1 AND %s = $2 AND %s = $3",
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableNameColumn,
|
|
sqlReleaseTableOwnerColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.
|
|
ExpectQuery(regexp.QuoteMeta(query)).
|
|
WithArgs("smug-pigeon", sqlReleaseDefaultOwner, "default").
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
}).AddRow(
|
|
supersededReleaseBody,
|
|
).AddRow(
|
|
deployedReleaseBody,
|
|
),
|
|
).RowsWillBeClosed()
|
|
|
|
mockGetReleaseCustomLabels(mock, "", supersededRelease.Namespace, supersededRelease.Labels)
|
|
mockGetReleaseCustomLabels(mock, "", deployedRelease.Namespace, deployedRelease.Labels)
|
|
|
|
_, err := sqlDriver.Query(labelSetUnknown)
|
|
if err == nil {
|
|
t.Errorf("Expected error {%v}, got nil", ErrReleaseNotFound)
|
|
} else if err != ErrReleaseNotFound {
|
|
t.Fatalf("failed to query for unknown smug-pigeon release: %v", err)
|
|
}
|
|
|
|
results, err := sqlDriver.Query(labelSetDeployed)
|
|
if err != nil {
|
|
t.Fatalf("failed to query for deployed smug-pigeon release: %v", err)
|
|
}
|
|
|
|
for _, res := range results {
|
|
if !reflect.DeepEqual(res, deployedRelease) {
|
|
t.Errorf("Expected release {%v}, got {%v}", deployedRelease, res)
|
|
}
|
|
}
|
|
|
|
results, err = sqlDriver.Query(labelSetAll)
|
|
if err != nil {
|
|
t.Fatalf("failed to query release history for smug-pigeon: %v", err)
|
|
}
|
|
|
|
if len(results) != 2 {
|
|
t.Errorf("expected a resultset of size 2, got %d", len(results))
|
|
}
|
|
|
|
for _, res := range results {
|
|
if !reflect.DeepEqual(res, deployedRelease) && !reflect.DeepEqual(res, supersededRelease) {
|
|
t.Errorf("Expected release {%v} or {%v}, got {%v}", deployedRelease, supersededRelease, res)
|
|
}
|
|
}
|
|
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestSqlDelete(t *testing.T) {
|
|
vers := 1
|
|
name := "smug-pigeon"
|
|
namespace := "default"
|
|
key := testKey(name, vers)
|
|
rel := releaseStub(name, vers, namespace, rspb.StatusDeployed)
|
|
|
|
body, _ := encodeRelease(rel)
|
|
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
|
|
selectQuery := fmt.Sprintf(
|
|
"SELECT %s FROM %s WHERE %s = $1 AND %s = $2",
|
|
sqlReleaseTableBodyColumn,
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.ExpectBegin()
|
|
mock.
|
|
ExpectQuery(regexp.QuoteMeta(selectQuery)).
|
|
WithArgs(key, namespace).
|
|
WillReturnRows(
|
|
mock.NewRows([]string{
|
|
sqlReleaseTableBodyColumn,
|
|
}).AddRow(
|
|
body,
|
|
),
|
|
).RowsWillBeClosed()
|
|
|
|
deleteQuery := fmt.Sprintf(
|
|
"DELETE FROM %s WHERE %s = $1 AND %s = $2",
|
|
sqlReleaseTableName,
|
|
sqlReleaseTableKeyColumn,
|
|
sqlReleaseTableNamespaceColumn,
|
|
)
|
|
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(deleteQuery)).
|
|
WithArgs(key, namespace).
|
|
WillReturnResult(sqlmock.NewResult(0, 1))
|
|
|
|
mockGetReleaseCustomLabels(mock, key, namespace, rel.Labels)
|
|
|
|
deleteLabelsQuery := fmt.Sprintf(
|
|
"DELETE FROM %s WHERE %s = $1 AND %s = $2",
|
|
sqlCustomLabelsTableName,
|
|
sqlCustomLabelsTableReleaseKeyColumn,
|
|
sqlCustomLabelsTableReleaseNamespaceColumn,
|
|
)
|
|
mock.
|
|
ExpectExec(regexp.QuoteMeta(deleteLabelsQuery)).
|
|
WithArgs(key, namespace).
|
|
WillReturnResult(sqlmock.NewResult(0, 1))
|
|
|
|
mock.ExpectCommit()
|
|
|
|
deletedRelease, err := sqlDriver.Delete(key)
|
|
if err := mock.ExpectationsWereMet(); err != nil {
|
|
t.Errorf("sql expectations weren't met: %v", err)
|
|
}
|
|
if err != nil {
|
|
t.Fatalf("failed to delete release with key %q: %v", key, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(rel, deletedRelease) {
|
|
t.Errorf("Expected release {%v}, got {%v}", rel, deletedRelease)
|
|
}
|
|
}
|
|
|
|
func mockGetReleaseCustomLabels(mock sqlmock.Sqlmock, key string, namespace string, labels map[string]string) {
|
|
query := fmt.Sprintf(
|
|
regexp.QuoteMeta("SELECT %s, %s FROM %s WHERE %s = $1 AND %s = $2"),
|
|
sqlCustomLabelsTableKeyColumn,
|
|
sqlCustomLabelsTableValueColumn,
|
|
sqlCustomLabelsTableName,
|
|
sqlCustomLabelsTableReleaseKeyColumn,
|
|
sqlCustomLabelsTableReleaseNamespaceColumn,
|
|
)
|
|
|
|
eq := mock.ExpectQuery(query).
|
|
WithArgs(key, namespace)
|
|
|
|
returnRows := mock.NewRows([]string{
|
|
sqlCustomLabelsTableKeyColumn,
|
|
sqlCustomLabelsTableValueColumn,
|
|
})
|
|
for k, v := range labels {
|
|
returnRows.AddRow(k, v)
|
|
}
|
|
eq.WillReturnRows(returnRows).RowsWillBeClosed()
|
|
}
|
|
|
|
func TestSqlCheckAppliedMigrations(t *testing.T) {
|
|
cases := []struct {
|
|
migrationsToApply []*migrate.Migration
|
|
appliedMigrationsIDs []string
|
|
expectedResult bool
|
|
errorExplanation string
|
|
}{
|
|
{
|
|
migrationsToApply: []*migrate.Migration{{Id: "init1"}, {Id: "init2"}, {Id: "init3"}},
|
|
appliedMigrationsIDs: []string{"1", "2", "init1", "3", "init2", "4", "5"},
|
|
expectedResult: false,
|
|
errorExplanation: "Has found one migration id \"init3\" as applied, that was not applied",
|
|
},
|
|
{
|
|
migrationsToApply: []*migrate.Migration{{Id: "init1"}, {Id: "init2"}, {Id: "init3"}},
|
|
appliedMigrationsIDs: []string{"1", "2", "init1", "3", "init2", "4", "init3", "5"},
|
|
expectedResult: true,
|
|
errorExplanation: "Has not found one or more migration ids, that was applied",
|
|
},
|
|
{
|
|
migrationsToApply: []*migrate.Migration{{Id: "init"}},
|
|
appliedMigrationsIDs: []string{"1", "2", "3", "inits", "4", "tinit", "5"},
|
|
expectedResult: false,
|
|
errorExplanation: "Has found single \"init\", that was not applied",
|
|
},
|
|
{
|
|
migrationsToApply: []*migrate.Migration{{Id: "init"}},
|
|
appliedMigrationsIDs: []string{"1", "2", "init", "3", "init2", "4", "init3", "5"},
|
|
expectedResult: true,
|
|
errorExplanation: "Has not found single migration id \"init\", that was applied",
|
|
},
|
|
}
|
|
for i, c := range cases {
|
|
sqlDriver, mock := newTestFixtureSQL(t)
|
|
rows := sqlmock.NewRows([]string{"id", "applied_at"})
|
|
for _, id := range c.appliedMigrationsIDs {
|
|
rows.AddRow(id, time.Time{})
|
|
}
|
|
mock.
|
|
ExpectQuery("").
|
|
WillReturnRows(rows)
|
|
mock.ExpectCommit()
|
|
if sqlDriver.checkAlreadyApplied(c.migrationsToApply) != c.expectedResult {
|
|
t.Errorf("Test case: %v, Expected: %v, Have: %v, Explanation: %v", i, c.expectedResult, !c.expectedResult, c.errorExplanation)
|
|
}
|
|
}
|
|
}
|