common/mock_Network.go

package common

import "github.com/stretchr/testify/mock"

import "io"

type MockNetwork struct {
	mock.Mock
}

func (m *MockNetwork) RegisterRunner(config RunnerCredentials, description string, tags string, runUntagged bool, locked bool) *RegisterRunnerResponse {
	ret := m.Called(config, description, tags, runUntagged, locked)

	var r0 *RegisterRunnerResponse
	if ret.Get(0) != nil {
		r0 = ret.Get(0).(*RegisterRunnerResponse)
	}

	return r0
}
func (m *MockNetwork) VerifyRunner(config RunnerCredentials) bool {
	ret := m.Called(config)

	r0 := ret.Get(0).(bool)

	return r0
}
func (m *MockNetwork) UnregisterRunner(config RunnerCredentials) bool {
	ret := m.Called(config)

	r0 := ret.Get(0).(bool)

	return r0
}
func (m *MockNetwork) RequestJob(config RunnerConfig) (*JobResponse, bool) {
	ret := m.Called(config)

	var r0 *JobResponse
	if ret.Get(0) != nil {
		r0 = ret.Get(0).(*JobResponse)
	}
	r1 := ret.Get(1).(bool)

	return r0, r1
}
func (m *MockNetwork) UpdateJob(config RunnerConfig, jobCredentials *JobCredentials, id int, state JobState, trace *string) UpdateState {
	ret := m.Called(config, jobCredentials, id, state, trace)

	r0 := ret.Get(0).(UpdateState)

	return r0
}
func (m *MockNetwork) PatchTrace(config RunnerConfig, jobCredentials *JobCredentials, tracePart JobTracePatch) UpdateState {
	ret := m.Called(config, jobCredentials, tracePart)

	r0 := ret.Get(0).(UpdateState)

	return r0
}
func (m *MockNetwork) DownloadArtifacts(config JobCredentials, artifactsFile string) DownloadState {
	ret := m.Called(config, artifactsFile)

	r0 := ret.Get(0).(DownloadState)

	return r0
}
func (m *MockNetwork) UploadRawArtifacts(config JobCredentials, reader io.Reader, baseName string, expireIn string) UploadState {
	ret := m.Called(config, reader, baseName, expireIn)

	r0 := ret.Get(0).(UploadState)

	return r0
}
func (m *MockNetwork) UploadArtifacts(config JobCredentials, artifactsFile string) UploadState {
	ret := m.Called(config, artifactsFile)

	r0 := ret.Get(0).(UploadState)

	return r0
}
func (m *MockNetwork) ProcessJob(config RunnerConfig, buildCredentials *JobCredentials) JobTrace {
	ret := m.Called(config, buildCredentials)

	r0 := ret.Get(0).(JobTrace)

	return r0
}