datastoretest

type StorerMock ¶

type StorerMock struct {
	// CloseFunc mocks the Close method.
	CloseFunc func(ctx context.Context) error

	// GetHierarchyCodelistFunc mocks the GetHierarchyCodelist method.
	GetHierarchyCodelistFunc func(ctx context.Context, instanceID string, dimension string) (string, error)

	// GetHierarchyElementFunc mocks the GetHierarchyElement method.
	GetHierarchyElementFunc func(ctx context.Context, instanceID string, dimension string, code string) (*models.HierarchyResponse, error)

	// GetHierarchyRootFunc mocks the GetHierarchyRoot method.
	GetHierarchyRootFunc func(ctx context.Context, instanceID string, dimension string) (*models.HierarchyResponse, error)
	// contains filtered or unexported fields
}

StorerMock is a mock implementation of datastore.Storer.

    func TestSomethingThatUsesStorer(t *testing.T) {

        // make and configure a mocked datastore.Storer
        mockedStorer := &StorerMock{
            CloseFunc: func(ctx context.Context) error {
	               panic("mock out the Close method")
            },
            GetHierarchyCodelistFunc: func(ctx context.Context, instanceID string, dimension string) (string, error) {
	               panic("mock out the GetHierarchyCodelist method")
            },
            GetHierarchyElementFunc: func(ctx context.Context, instanceID string, dimension string, code string) (*models.HierarchyResponse, error) {
	               panic("mock out the GetHierarchyElement method")
            },
            GetHierarchyRootFunc: func(ctx context.Context, instanceID string, dimension string) (*models.HierarchyResponse, error) {
	               panic("mock out the GetHierarchyRoot method")
            },
        }

        // use mockedStorer in code that requires datastore.Storer
        // and then make assertions.

    }

func (mock *StorerMock) Close(ctx context.Context) error

func (mock *StorerMock) CloseCalls() []struct {
	Ctx context.Context
}

len(mockedStorer.CloseCalls())

func (mock *StorerMock) GetHierarchyCodelist(ctx context.Context, instanceID string, dimension string) (string, error)

func (mock *StorerMock) GetHierarchyCodelistCalls() []struct {
	Ctx        context.Context
	InstanceID string
	Dimension  string
}

len(mockedStorer.GetHierarchyCodelistCalls())

func (mock *StorerMock) GetHierarchyElement(ctx context.Context, instanceID string, dimension string, code string) (*models.HierarchyResponse, error)

func (mock *StorerMock) GetHierarchyElementCalls() []struct {
	Ctx        context.Context
	InstanceID string
	Dimension  string
	Code       string
}

len(mockedStorer.GetHierarchyElementCalls())

func (mock *StorerMock) GetHierarchyRoot(ctx context.Context, instanceID string, dimension string) (*models.HierarchyResponse, error)

func (mock *StorerMock) GetHierarchyRootCalls() []struct {
	Ctx        context.Context
	InstanceID string
	Dimension  string
}

len(mockedStorer.GetHierarchyRootCalls())