registry_mock.go

package dao

import (
	"testing"
)

type MockDaoRegistry struct {
	RepositoryConfig MockRepositoryConfigDao
	Rpm              MockRpmDao
	Repository       MockRepositoryDao
	Metrics          MockMetricsDao
	Snapshot         MockSnapshotDao
	TaskInfo         MockTaskInfoDao
	AdminTask        MockAdminTaskDao
	Domain           MockDomainDao
	PackageGroup     MockPackageGroupDao
	Environment      MockEnvironmentDao
	Template         MockTemplateDao
	ModuleStream     MockModuleStreamDao
}

func (m *MockDaoRegistry) ToDaoRegistry() *DaoRegistry {
	r := DaoRegistry{
		RepositoryConfig: &m.RepositoryConfig,
		Rpm:              &m.Rpm,
		Repository:       &m.Repository,
		Metrics:          &m.Metrics,
		Snapshot:         &m.Snapshot,
		TaskInfo:         &m.TaskInfo,
		AdminTask:        &m.AdminTask,
		Domain:           &m.Domain,
		PackageGroup:     &m.PackageGroup,
		Environment:      &m.Environment,
		Template:         &m.Template,
		ModuleStream:     &m.ModuleStream,
	}
	return &r
}

func GetMockDaoRegistry(t *testing.T) *MockDaoRegistry {
	reg := MockDaoRegistry{
		RepositoryConfig: *NewMockRepositoryConfigDao(t),
		Rpm:              *NewMockRpmDao(t),
		Repository:       *NewMockRepositoryDao(t),
		Metrics:          *NewMockMetricsDao(t),
		Snapshot:         *NewMockSnapshotDao(t),
		TaskInfo:         *NewMockTaskInfoDao(t),
		AdminTask:        *NewMockAdminTaskDao(t),
		Domain:           *NewMockDomainDao(t),
		PackageGroup:     *NewMockPackageGroupDao(t),
		Environment:      *NewMockEnvironmentDao(t),
		Template:         *NewMockTemplateDao(t),
		ModuleStream:     *NewMockModuleStreamDao(t),
	}
	return &reg
}