github.com/kyma-incubator/compass/components/director@v0.0.0-20230623144113-d764f56ff805/internal/domain/runtime_context/repository_test.go

package runtimectx_test

import (
	"context"
	"database/sql/driver"
	"regexp"
	"testing"

	"github.com/stretchr/testify/assert"

	"github.com/kyma-incubator/compass/components/director/pkg/pagination"

	"github.com/DATA-DOG/go-sqlmock"
	runtimectx "github.com/kyma-incubator/compass/components/director/internal/domain/runtime_context"
	"github.com/kyma-incubator/compass/components/director/internal/domain/runtime_context/automock"
	"github.com/kyma-incubator/compass/components/director/internal/labelfilter"
	"github.com/kyma-incubator/compass/components/director/internal/model"
	"github.com/kyma-incubator/compass/components/director/internal/repo/testdb"
	"github.com/kyma-incubator/compass/components/director/pkg/persistence"
	"github.com/stretchr/testify/require"
)

func TestPgRepository_GetByID(t *testing.T) {
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoGetTestSuite{
		Name: "Get Runtime Context By ID",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE id = $1 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2))`),
				Args:     []driver.Value{runtimeCtxID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value)}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ExpectedModelEntity:       runtimeCtxModel,
		ExpectedDBEntity:          runtimeCtxEntity,
		MethodArgs:                []interface{}{tenantID, runtimeCtxID},
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_GetGlobalByID(t *testing.T) {
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoGetTestSuite{
		Name: "Get Runtime Context Globally by ID",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE id = $1`),
				Args:     []driver.Value{runtimeCtxID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value)}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ExpectedModelEntity:       runtimeCtxModel,
		ExpectedDBEntity:          runtimeCtxEntity,
		MethodName:                "GetGlobalByID",
		MethodArgs:                []interface{}{runtimeCtxID},
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_GetByRuntimeID(t *testing.T) {
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoGetTestSuite{
		Name: "Get Runtime Context By runtime ID",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE runtime_id = $1 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2))`),
				Args:     []driver.Value{runtimeID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value)}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ExpectedModelEntity:       runtimeCtxModel,
		ExpectedDBEntity:          runtimeCtxEntity,
		MethodName:                "GetByRuntimeID",
		MethodArgs:                []interface{}{tenantID, runtimeID},
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_GetForRuntime(t *testing.T) {
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoGetTestSuite{
		Name: "Get RuntimeContext For Runtime",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE id = $1 AND runtime_id = $2 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $3))`),
				Args:     []driver.Value{runtimeCtxID, runtimeID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{
						sqlmock.NewRows(fixColumns).
							AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{
						sqlmock.NewRows(fixColumns),
					}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ExpectedModelEntity:       runtimeCtxModel,
		ExpectedDBEntity:          runtimeCtxEntity,
		MethodArgs:                []interface{}{tenantID, runtimeCtxID, runtimeID},
		MethodName:                "GetForRuntime",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_GetByFiltersAndID(t *testing.T) {
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoGetTestSuite{
		Name: "Get Runtime Context By Filters and ID",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE id = $1
												AND id IN (SELECT "runtime_context_id" FROM public.labels WHERE "runtime_context_id" IS NOT NULL AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $2)) AND "key" = $3 AND "value" ?| array[$4])
												AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $5))`),
				Args:     []driver.Value{runtimeCtxID, tenantID, model.ScenariosKey, "scenario", tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value)}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ExpectedModelEntity:       runtimeCtxModel,
		ExpectedDBEntity:          runtimeCtxEntity,
		MethodName:                "GetByFiltersAndID",
		MethodArgs:                []interface{}{tenantID, runtimeCtxID, []*labelfilter.LabelFilter{labelfilter.NewForKeyWithQuery(model.ScenariosKey, `$[*] ? ( @ == "scenario" )`)}},
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_GetByFiltersGlobal_ShouldReturnRuntimeContextModelForRuntimeContextEntity(t *testing.T) {
	// GIVEN
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	sqlxDB, sqlMock := testdb.MockDatabase(t)
	defer sqlMock.AssertExpectations(t)

	rows := sqlmock.NewRows(fixColumns).AddRow(runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value)

	sqlMock.ExpectQuery(`^SELECT (.+) FROM public.runtime_contexts WHERE id IN \(SELECT "runtime_context_id" FROM public\.labels WHERE "runtime_context_id" IS NOT NULL AND "key" = \$1\)$`).
		WithArgs("someKey").
		WillReturnRows(rows)

	ctx := persistence.SaveToContext(context.TODO(), sqlxDB)

	mockConverter := &automock.EntityConverter{}
	mockConverter.On("FromEntity", runtimeCtxEntity).Return(runtimeCtxModel, nil).Once()
	pgRepository := runtimectx.NewRepository(mockConverter)

	// WHEN
	filters := []*labelfilter.LabelFilter{labelfilter.NewForKey("someKey")}
	modelRuntimeCtx, err := pgRepository.GetByFiltersGlobal(ctx, filters)

	// THEN
	require.NoError(t, err)
	require.Equal(t, runtimeCtxModel, modelRuntimeCtx)
	mockConverter.AssertExpectations(t)
}

func TestPgRepository_List(t *testing.T) {
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithID(runtimeCtx2ID)

	runtimeCtxModel1 := fixModelRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithID(runtimeCtx2ID)

	suite := testdb.RepoListPageableTestSuite{
		Name: "List Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE (runtime_id = $1
												AND id IN (SELECT "runtime_context_id" FROM public.labels WHERE "runtime_context_id" IS NOT NULL AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $2)) AND "key" = $3 AND "value" ?| array[$4])
												AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $5))) ORDER BY id LIMIT 2 OFFSET 0`),
				Args:     []driver.Value{runtimeID, tenantID, model.ScenariosKey, "scenario", tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
			},
			{
				Query: regexp.QuoteMeta(`SELECT COUNT(*) FROM public.runtime_contexts WHERE (runtime_id = $1
												AND id IN (SELECT "runtime_context_id" FROM public.labels WHERE "runtime_context_id" IS NOT NULL AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $2)) AND "key" = $3 AND "value" ?| array[$4])
												AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $5)))`),
				Args:     []driver.Value{runtimeID, tenantID, model.ScenariosKey, "scenario", tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows([]string{"count"}).AddRow(2)}
				},
			},
		},
		Pages: []testdb.PageDetails{
			{
				ExpectedModelEntities: []interface{}{runtimeCtxModel1, runtimeCtxModel2},
				ExpectedDBEntities:    []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
				ExpectedPage: &model.RuntimeContextPage{
					Data: []*model.RuntimeContext{runtimeCtxModel1, runtimeCtxModel2},
					PageInfo: &pagination.Page{
						StartCursor: "",
						EndCursor:   "",
						HasNextPage: false,
					},
					TotalCount: 2,
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{runtimeID, tenantID, []*labelfilter.LabelFilter{labelfilter.NewForKeyWithQuery(model.ScenariosKey, `$[*] ? ( @ == "scenario" )`)}, 2, ""},
		MethodName:                "List",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_ListByRuntimeIDs(t *testing.T) {
	pageSize := 1
	cursor := ""

	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, onePageRuntimeID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, multiplePagesRuntimeID)

	runtimeCtxModel1 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, onePageRuntimeID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, multiplePagesRuntimeID)

	suite := testdb.RepoListPageableTestSuite{
		Name: "ListByRuntimeIDs Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`(SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $1)) AND runtime_id = $2 ORDER BY runtime_id ASC, id ASC LIMIT $3 OFFSET $4)
												UNION
												(SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $5)) AND runtime_id = $6 ORDER BY runtime_id ASC, id ASC LIMIT $7 OFFSET $8)
												UNION
												(SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $9)) AND runtime_id = $10 ORDER BY runtime_id ASC, id ASC LIMIT $11 OFFSET $12)`),
				Args:     []driver.Value{tenantID, emptyPageRuntimeID, pageSize, 0, tenantID, onePageRuntimeID, pageSize, 0, tenantID, multiplePagesRuntimeID, pageSize, 0},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
			},
			{
				Query:    regexp.QuoteMeta(`SELECT runtime_id AS id, COUNT(*) AS total_count FROM public.runtime_contexts WHERE (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $1)) GROUP BY runtime_id ORDER BY runtime_id ASC`),
				Args:     []driver.Value{tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows([]string{"id", "total_count"}).AddRow(emptyPageRuntimeID, 0).AddRow(onePageRuntimeID, 1).AddRow(multiplePagesRuntimeID, 2)}
				},
			},
		},
		Pages: []testdb.PageDetails{
			{
				ExpectedModelEntities: nil,
				ExpectedDBEntities:    nil,
				ExpectedPage: &model.RuntimeContextPage{
					Data: nil,
					PageInfo: &pagination.Page{
						StartCursor: "",
						EndCursor:   "",
						HasNextPage: false,
					},
					TotalCount: 0,
				},
			},
			{
				ExpectedModelEntities: []interface{}{runtimeCtxModel1},
				ExpectedDBEntities:    []interface{}{runtimeCtxEntity1},
				ExpectedPage: &model.RuntimeContextPage{
					Data: []*model.RuntimeContext{runtimeCtxModel1},
					PageInfo: &pagination.Page{
						StartCursor: "",
						EndCursor:   "",
						HasNextPage: false,
					},
					TotalCount: 1,
				},
			},
			{
				ExpectedModelEntities: []interface{}{runtimeCtxModel2},
				ExpectedDBEntities:    []interface{}{runtimeCtxEntity2},
				ExpectedPage: &model.RuntimeContextPage{
					Data: []*model.RuntimeContext{runtimeCtxModel2},
					PageInfo: &pagination.Page{
						StartCursor: "",
						EndCursor:   pagination.EncodeNextOffsetCursor(0, pageSize),
						HasNextPage: true,
					},
					TotalCount: 2,
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID, []string{emptyPageRuntimeID, onePageRuntimeID, multiplePagesRuntimeID}, pageSize, cursor},
		MethodName:                "ListByRuntimeIDs",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_ListByScenariosAndRuntimeIDs(t *testing.T) {
	scenario1 := "scenario-1"
	scenario2 := "scenario-2"
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	runtimeCtxModel1 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	suite := testdb.RepoListTestSuite{
		Name: "ListByScenariosAndRuntimeIDs Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts 
											  WHERE id IN (SELECT "runtime_context_id" FROM public.labels 
											  WHERE "runtime_context_id" IS NOT NULL 
											  AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $1)) 
											  AND "key" = $2 AND "value" ?| array[$3] UNION SELECT "runtime_context_id" FROM public.labels WHERE "runtime_context_id" IS NOT NULL 
											  AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $4)) AND "key" = $5 
											  AND "value" ?| array[$6]) AND runtime_id IN ($7, $8) 
											  AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $9))`),
				Args:     []driver.Value{tenantID, model.ScenariosKey, scenario1, tenantID, model.ScenariosKey, scenario2, runtimeID, runtimeID2, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ExpectedModelEntities: []interface{}{runtimeCtxModel1, runtimeCtxModel2},
		ExpectedDBEntities:    []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID, []string{scenario1, scenario2}, []string{runtimeID, runtimeID2}},
		MethodName:                "ListByScenariosAndRuntimeIDs",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)

	// Additional test - empty slice because test suite returns empty result given valid query
	t.Run("returns empty slice given no scenarios", func(t *testing.T) {
		// GIVEN
		ctx := context.TODO()
		repository := runtimectx.NewRepository(nil)

		// WHEN
		actual, err := repository.ListByScenariosAndRuntimeIDs(ctx, tenantID, []string{}, []string{})

		// THEN
		assert.NoError(t, err)
		assert.Nil(t, actual)
	})
}

func TestPgRepository_ListByScenarios(t *testing.T) {
	scenario1 := "scenario-1"
	scenario2 := "scenario-2"
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	runtimeCtxModel1 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	suite := testdb.RepoListTestSuite{
		Name: "ListByScenarios Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts 
											  WHERE id IN (SELECT "runtime_context_id" FROM public.labels 
											  WHERE "runtime_context_id" IS NOT NULL 
											  AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $1)) 
											  AND "key" = $2 AND "value" ?| array[$3] UNION SELECT "runtime_context_id" FROM public.labels WHERE "runtime_context_id" IS NOT NULL 
											  AND (id IN (SELECT id FROM runtime_contexts_labels_tenants WHERE tenant_id = $4)) AND "key" = $5 
											  AND "value" ?| array[$6]) 
											  AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $7))`),
				Args:     []driver.Value{tenantID, model.ScenariosKey, scenario1, tenantID, model.ScenariosKey, scenario2, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ExpectedModelEntities: []interface{}{runtimeCtxModel1, runtimeCtxModel2},
		ExpectedDBEntities:    []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID, []string{scenario1, scenario2}},
		MethodName:                "ListByScenarios",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)

	// Additional test - empty slice because test suite returns empty result given valid query
	t.Run("returns empty slice given no scenarios", func(t *testing.T) {
		// GIVEN
		ctx := context.TODO()
		repository := runtimectx.NewRepository(nil)

		// WHEN
		actual, err := repository.ListByScenarios(ctx, tenantID, []string{})

		// THEN
		assert.NoError(t, err)
		assert.Nil(t, actual)
	})
}

func TestPgRepository_ListByIDs(t *testing.T) {
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	runtimeCtxModel1 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx1ID, runtimeID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithIDAndRuntimeID(runtimeCtx2ID, runtimeID2)

	suite := testdb.RepoListTestSuite{
		Name: "ListByIDs Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE id IN ($1, $2) AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $3))`),
				Args:     []driver.Value{runtimeCtx1ID, runtimeCtx2ID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ExpectedModelEntities: []interface{}{runtimeCtxModel1, runtimeCtxModel2},
		ExpectedDBEntities:    []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID, []string{runtimeCtx1ID, runtimeCtx2ID}},
		MethodName:                "ListByIDs",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)

	// Additional test - empty slice because test suite returns empty result given valid query
	t.Run("returns empty slice given no scenarios", func(t *testing.T) {
		// GIVEN
		ctx := context.TODO()
		repository := runtimectx.NewRepository(nil)

		// WHEN
		actual, err := repository.ListByIDs(ctx, tenantID, []string{})

		// THEN
		assert.NoError(t, err)
		assert.Nil(t, actual)
	})
}
func TestPgRepository_ListAll(t *testing.T) {
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithID(runtimeCtx2ID)

	runtimeCtxModel1 := fixModelRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithID(runtimeCtx2ID)

	suite := testdb.RepoListTestSuite{
		Name: "List Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $1))`),
				Args:     []driver.Value{tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		ExpectedDBEntities:        []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
		ExpectedModelEntities:     []interface{}{runtimeCtxModel1, runtimeCtxModel2},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID},
		MethodName:                "ListAll",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_ListAllForRuntime(t *testing.T) {
	runtimeCtx1ID := "aec0e9c5-06da-4625-9f8a-bda17ab8c3b9"
	runtimeCtx2ID := "ccdbef8f-b97a-490c-86e2-2bab2862a6e4"
	runtimeCtxEntity1 := fixEntityRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxEntity2 := fixEntityRuntimeCtxWithID(runtimeCtx2ID)

	runtimeCtxModel1 := fixModelRuntimeCtxWithID(runtimeCtx1ID)
	runtimeCtxModel2 := fixModelRuntimeCtxWithID(runtimeCtx2ID)

	suite := testdb.RepoListTestSuite{
		Name: "List Runtime Contexts",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query: regexp.QuoteMeta(`SELECT id, runtime_id, key, value FROM public.runtime_contexts WHERE runtime_id = $1 AND
												(id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2))`),
				Args:     []driver.Value{runtimeID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns).
						AddRow(runtimeCtxEntity1.ID, runtimeCtxEntity1.RuntimeID, runtimeCtxEntity1.Key, runtimeCtxEntity1.Value).
						AddRow(runtimeCtxEntity2.ID, runtimeCtxEntity2.RuntimeID, runtimeCtxEntity2.Key, runtimeCtxEntity2.Value),
					}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{sqlmock.NewRows(fixColumns)}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		ExpectedDBEntities:        []interface{}{runtimeCtxEntity1, runtimeCtxEntity2},
		ExpectedModelEntities:     []interface{}{runtimeCtxModel1, runtimeCtxModel2},
		RepoConstructorFunc:       runtimectx.NewRepository,
		MethodArgs:                []interface{}{tenantID, runtimeID},
		MethodName:                "ListAllForRuntime",
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_Create(t *testing.T) {
	var nilRuntimeCtxModel *model.RuntimeContext
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoCreateTestSuite{
		Name: "Create Runtime Context",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:       `^INSERT INTO public.runtime_contexts \(.+\) VALUES \(.+\)$`,
				Args:        []driver.Value{runtimeCtxModel.ID, runtimeCtxModel.RuntimeID, runtimeCtxModel.Key, runtimeCtxModel.Value},
				ValidResult: sqlmock.NewResult(-1, 1),
			},
			{
				Query:       regexp.QuoteMeta(`WITH RECURSIVE parents AS (SELECT t1.id, t1.parent FROM business_tenant_mappings t1 WHERE id = ? UNION ALL SELECT t2.id, t2.parent FROM business_tenant_mappings t2 INNER JOIN parents t on t2.id = t.parent) INSERT INTO tenant_runtime_contexts ( tenant_id, id, owner ) (SELECT parents.id AS tenant_id, ? as id, ? AS owner FROM parents)`),
				Args:        []driver.Value{tenantID, runtimeCtxModel.ID, true},
				ValidResult: sqlmock.NewResult(-1, 1),
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ModelEntity:               runtimeCtxModel,
		DBEntity:                  runtimeCtxEntity,
		NilModelEntity:            nilRuntimeCtxModel,
		TenantID:                  tenantID,
		DisableConverterErrorTest: true,
		IsTopLevelEntity:          true,
	}

	suite.Run(t)
}

func TestPgRepository_Update(t *testing.T) {
	var nilRuntimeCtxModel *model.RuntimeContext
	runtimeCtxModel := fixModelRuntimeCtx()
	runtimeCtxEntity := fixEntityRuntimeCtx()

	suite := testdb.RepoUpdateTestSuite{
		Name: "Update Runtime Context",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:         regexp.QuoteMeta(`UPDATE public.runtime_contexts SET key = ?, value = ? WHERE id = ? AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = ? AND owner = true))`),
				Args:          []driver.Value{runtimeCtxModel.Key, runtimeCtxModel.Value, runtimeCtxModel.ID, tenantID},
				ValidResult:   sqlmock.NewResult(-1, 1),
				InvalidResult: sqlmock.NewResult(-1, 0),
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc:       runtimectx.NewRepository,
		ModelEntity:               runtimeCtxModel,
		DBEntity:                  runtimeCtxEntity,
		NilModelEntity:            nilRuntimeCtxModel,
		TenantID:                  tenantID,
		DisableConverterErrorTest: true,
	}

	suite.Run(t)
}

func TestPgRepository_Delete(t *testing.T) {
	suite := testdb.RepoDeleteTestSuite{
		Name: "Runtime Context Delete",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:         regexp.QuoteMeta(`DELETE FROM public.runtime_contexts WHERE id = $1 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2 AND owner = true))`),
				Args:          []driver.Value{runtimeCtxID, tenantID},
				ValidResult:   sqlmock.NewResult(-1, 1),
				InvalidResult: sqlmock.NewResult(-1, 2),
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc: runtimectx.NewRepository,
		MethodArgs:          []interface{}{tenantID, runtimeCtxID},
	}

	suite.Run(t)
}

func TestPgRepository_Exist(t *testing.T) {
	suite := testdb.RepoExistTestSuite{
		Name: "Runtime Context Exists",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT 1 FROM public.runtime_contexts WHERE id = $1 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2))`),
				Args:     []driver.Value{runtimeCtxID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{testdb.RowWhenObjectExist()}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{testdb.RowWhenObjectDoesNotExist()}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc: runtimectx.NewRepository,
		TargetID:            runtimeCtxID,
		TenantID:            tenantID,
		MethodName:          "Exists",
		MethodArgs:          []interface{}{tenantID, runtimeCtxID},
	}

	suite.Run(t)
}

func TestPgRepository_ExistsByRuntimeID(t *testing.T) {
	suite := testdb.RepoExistTestSuite{
		Name: "Runtime Context Exists By Runtime ID",
		SQLQueryDetails: []testdb.SQLQueryDetails{
			{
				Query:    regexp.QuoteMeta(`SELECT 1 FROM public.runtime_contexts WHERE runtime_id = $1 AND (id IN (SELECT id FROM tenant_runtime_contexts WHERE tenant_id = $2))`),
				Args:     []driver.Value{runtimeID, tenantID},
				IsSelect: true,
				ValidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{testdb.RowWhenObjectExist()}
				},
				InvalidRowsProvider: func() []*sqlmock.Rows {
					return []*sqlmock.Rows{testdb.RowWhenObjectDoesNotExist()}
				},
			},
		},
		ConverterMockProvider: func() testdb.Mock {
			return &automock.EntityConverter{}
		},
		RepoConstructorFunc: runtimectx.NewRepository,
		TargetID:            runtimeCtxID,
		TenantID:            tenantID,
		MethodName:          "ExistsByRuntimeID",
		MethodArgs:          []interface{}{tenantID, runtimeID},
	}

	suite.Run(t)
}