Commit 889b407e authored by Saketh Pai's avatar Saketh Pai 🏍️ Committed by Timo Furrer
Browse files

feat(mocks): add streaming methods for various services

parent d5e76a7b

testing/attestations_mock.go

+45 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ 
package testing



import (

	io "io"

	reflect "reflect"



	gitlab "gitlab.com/gitlab-org/api/client-go"
@@ -128,3 +129,47 @@ func (c *MockAttestationsServiceInterfaceListAttestationsCall) DoAndReturn(f fun 
	c.Call = c.Call.DoAndReturn(f)

	return c

}



// StreamDownloadAttestation mocks base method.

func (m *MockAttestationsServiceInterface) StreamDownloadAttestation(pid any, attestationIID int64, w io.Writer, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, attestationIID, w}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamDownloadAttestation", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamDownloadAttestation indicates an expected call of StreamDownloadAttestation.

func (mr *MockAttestationsServiceInterfaceMockRecorder) StreamDownloadAttestation(pid, attestationIID, w any, options ...any) *MockAttestationsServiceInterfaceStreamDownloadAttestationCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, attestationIID, w}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamDownloadAttestation", reflect.TypeOf((*MockAttestationsServiceInterface)(nil).StreamDownloadAttestation), varargs...)

	return &MockAttestationsServiceInterfaceStreamDownloadAttestationCall{Call: call}

}



// MockAttestationsServiceInterfaceStreamDownloadAttestationCall wrap *gomock.Call

type MockAttestationsServiceInterfaceStreamDownloadAttestationCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockAttestationsServiceInterfaceStreamDownloadAttestationCall) Return(arg0 *gitlab.Response, arg1 error) *MockAttestationsServiceInterfaceStreamDownloadAttestationCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockAttestationsServiceInterfaceStreamDownloadAttestationCall) Do(f func(any, int64, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockAttestationsServiceInterfaceStreamDownloadAttestationCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockAttestationsServiceInterfaceStreamDownloadAttestationCall) DoAndReturn(f func(any, int64, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockAttestationsServiceInterfaceStreamDownloadAttestationCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}

testing/generic_packages_mock.go

+44 −0
Original line number Diff line number Diff line
@@ -168,3 +168,47 @@ func (c *MockGenericPackagesServiceInterfacePublishPackageFileCall) DoAndReturn( 
	c.Call = c.Call.DoAndReturn(f)

	return c

}



// StreamDownloadPackageFile mocks base method.

func (m *MockGenericPackagesServiceInterface) StreamDownloadPackageFile(pid any, packageName, packageVersion, fileName string, w io.Writer, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, packageName, packageVersion, fileName, w}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamDownloadPackageFile", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamDownloadPackageFile indicates an expected call of StreamDownloadPackageFile.

func (mr *MockGenericPackagesServiceInterfaceMockRecorder) StreamDownloadPackageFile(pid, packageName, packageVersion, fileName, w any, options ...any) *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, packageName, packageVersion, fileName, w}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamDownloadPackageFile", reflect.TypeOf((*MockGenericPackagesServiceInterface)(nil).StreamDownloadPackageFile), varargs...)

	return &MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall{Call: call}

}



// MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall wrap *gomock.Call

type MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall) Return(arg0 *gitlab.Response, arg1 error) *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall) Do(f func(any, string, string, string, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall) DoAndReturn(f func(any, string, string, string, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockGenericPackagesServiceInterfaceStreamDownloadPackageFileCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}

testing/merge_requests_mock.go

+45 −0
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ 
package testing



import (

	io "io"

	reflect "reflect"



	gitlab "gitlab.com/gitlab-org/api/client-go"
@@ -1431,6 +1432,50 @@ func (c *MockMergeRequestsServiceInterfaceShowMergeRequestRawDiffsCall) DoAndRet 
	return c

}



// StreamMergeRequestRawDiffs mocks base method.

func (m *MockMergeRequestsServiceInterface) StreamMergeRequestRawDiffs(pid any, mergeRequest int64, w io.Writer, opt *gitlab.ShowMergeRequestRawDiffsOptions, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, mergeRequest, w, opt}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamMergeRequestRawDiffs", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamMergeRequestRawDiffs indicates an expected call of StreamMergeRequestRawDiffs.

func (mr *MockMergeRequestsServiceInterfaceMockRecorder) StreamMergeRequestRawDiffs(pid, mergeRequest, w, opt any, options ...any) *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, mergeRequest, w, opt}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamMergeRequestRawDiffs", reflect.TypeOf((*MockMergeRequestsServiceInterface)(nil).StreamMergeRequestRawDiffs), varargs...)

	return &MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall{Call: call}

}



// MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall wrap *gomock.Call

type MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall) Return(arg0 *gitlab.Response, arg1 error) *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall) Do(f func(any, int64, io.Writer, *gitlab.ShowMergeRequestRawDiffsOptions, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall) DoAndReturn(f func(any, int64, io.Writer, *gitlab.ShowMergeRequestRawDiffsOptions, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockMergeRequestsServiceInterfaceStreamMergeRequestRawDiffsCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}



// SubscribeToMergeRequest mocks base method.

func (m *MockMergeRequestsServiceInterface) SubscribeToMergeRequest(pid any, mergeRequest int64, options ...gitlab.RequestOptionFunc) (*gitlab.MergeRequest, *gitlab.Response, error) {

	m.ctrl.T.Helper()

testing/project_import_export_mock.go

+44 −0
Original line number Diff line number Diff line
@@ -263,3 +263,47 @@ func (c *MockProjectImportExportServiceInterfaceScheduleExportCall) DoAndReturn( 
	c.Call = c.Call.DoAndReturn(f)

	return c

}



// StreamExportDownload mocks base method.

func (m *MockProjectImportExportServiceInterface) StreamExportDownload(pid any, w io.Writer, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, w}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamExportDownload", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamExportDownload indicates an expected call of StreamExportDownload.

func (mr *MockProjectImportExportServiceInterfaceMockRecorder) StreamExportDownload(pid, w any, options ...any) *MockProjectImportExportServiceInterfaceStreamExportDownloadCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, w}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamExportDownload", reflect.TypeOf((*MockProjectImportExportServiceInterface)(nil).StreamExportDownload), varargs...)

	return &MockProjectImportExportServiceInterfaceStreamExportDownloadCall{Call: call}

}



// MockProjectImportExportServiceInterfaceStreamExportDownloadCall wrap *gomock.Call

type MockProjectImportExportServiceInterfaceStreamExportDownloadCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockProjectImportExportServiceInterfaceStreamExportDownloadCall) Return(arg0 *gitlab.Response, arg1 error) *MockProjectImportExportServiceInterfaceStreamExportDownloadCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockProjectImportExportServiceInterfaceStreamExportDownloadCall) Do(f func(any, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectImportExportServiceInterfaceStreamExportDownloadCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockProjectImportExportServiceInterfaceStreamExportDownloadCall) DoAndReturn(f func(any, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectImportExportServiceInterfaceStreamExportDownloadCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}

testing/project_markdown_uploads_mock.go

+88 −0
Original line number Diff line number Diff line
@@ -263,6 +263,94 @@ func (c *MockProjectMarkdownUploadsServiceInterfaceListProjectMarkdownUploadsCal 
	return c

}



// StreamDownloadProjectMarkdownUploadByID mocks base method.

func (m *MockProjectMarkdownUploadsServiceInterface) StreamDownloadProjectMarkdownUploadByID(pid any, uploadID int64, w io.Writer, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, uploadID, w}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamDownloadProjectMarkdownUploadByID", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamDownloadProjectMarkdownUploadByID indicates an expected call of StreamDownloadProjectMarkdownUploadByID.

func (mr *MockProjectMarkdownUploadsServiceInterfaceMockRecorder) StreamDownloadProjectMarkdownUploadByID(pid, uploadID, w any, options ...any) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, uploadID, w}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamDownloadProjectMarkdownUploadByID", reflect.TypeOf((*MockProjectMarkdownUploadsServiceInterface)(nil).StreamDownloadProjectMarkdownUploadByID), varargs...)

	return &MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall{Call: call}

}



// MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall wrap *gomock.Call

type MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall) Return(arg0 *gitlab.Response, arg1 error) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall) Do(f func(any, int64, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall) DoAndReturn(f func(any, int64, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadByIDCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}



// StreamDownloadProjectMarkdownUploadBySecretAndFilename mocks base method.

func (m *MockProjectMarkdownUploadsServiceInterface) StreamDownloadProjectMarkdownUploadBySecretAndFilename(pid any, secret, filename string, w io.Writer, options ...gitlab.RequestOptionFunc) (*gitlab.Response, error) {

	m.ctrl.T.Helper()

	varargs := []any{pid, secret, filename, w}

	for _, a := range options {

		varargs = append(varargs, a)

	}

	ret := m.ctrl.Call(m, "StreamDownloadProjectMarkdownUploadBySecretAndFilename", varargs...)

	ret0, _ := ret[0].(*gitlab.Response)

	ret1, _ := ret[1].(error)

	return ret0, ret1

}



// StreamDownloadProjectMarkdownUploadBySecretAndFilename indicates an expected call of StreamDownloadProjectMarkdownUploadBySecretAndFilename.

func (mr *MockProjectMarkdownUploadsServiceInterfaceMockRecorder) StreamDownloadProjectMarkdownUploadBySecretAndFilename(pid, secret, filename, w any, options ...any) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall {

	mr.mock.ctrl.T.Helper()

	varargs := append([]any{pid, secret, filename, w}, options...)

	call := mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StreamDownloadProjectMarkdownUploadBySecretAndFilename", reflect.TypeOf((*MockProjectMarkdownUploadsServiceInterface)(nil).StreamDownloadProjectMarkdownUploadBySecretAndFilename), varargs...)

	return &MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall{Call: call}

}



// MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall wrap *gomock.Call

type MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall struct {

	*gomock.Call

}



// Return rewrite *gomock.Call.Return

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall) Return(arg0 *gitlab.Response, arg1 error) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall {

	c.Call = c.Call.Return(arg0, arg1)

	return c

}



// Do rewrite *gomock.Call.Do

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall) Do(f func(any, string, string, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall {

	c.Call = c.Call.Do(f)

	return c

}



// DoAndReturn rewrite *gomock.Call.DoAndReturn

func (c *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall) DoAndReturn(f func(any, string, string, io.Writer, ...gitlab.RequestOptionFunc) (*gitlab.Response, error)) *MockProjectMarkdownUploadsServiceInterfaceStreamDownloadProjectMarkdownUploadBySecretAndFilenameCall {

	c.Call = c.Call.DoAndReturn(f)

	return c

}



// UploadProjectMarkdown mocks base method.

func (m *MockProjectMarkdownUploadsServiceInterface) UploadProjectMarkdown(pid any, content io.Reader, filename string, options ...gitlab.RequestOptionFunc) (*gitlab.ProjectMarkdownUploadedFile, *gitlab.Response, error) {

	m.ctrl.T.Helper()