multiapps-cli-plugin/cli/fakes/fake_cli_connection_builder.go at f647983d2dadfa2234832c5124cc516ba28a7fd3 · cloudfoundry/multiapps-cli-plugin · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
package fakes

import (
	plugin_models "code.cloudfoundry.org/cli/v8/plugin/models"
	fakes "code.cloudfoundry.org/cli/v8/plugin/pluginfakes"
)

// FakeCliConnectionBuilder is a builder of FakeCliConnection instances
type FakeCliConnectionBuilder struct {
	cliConn fakes.FakeCliConnection
}

// NewFakeCliConnectionBuilder creates a new builder
func NewFakeCliConnectionBuilder() *FakeCliConnectionBuilder {
	return &FakeCliConnectionBuilder{}
}

func (b *FakeCliConnectionBuilder) CurrentOrg(guid, name string, err error) *FakeCliConnectionBuilder {
	org := plugin_models.Organization{OrganizationFields: plugin_models.OrganizationFields{Guid: guid, Name: name}}
	b.cliConn.GetCurrentOrgReturns(org, err)
	return b
}

func (b *FakeCliConnectionBuilder) CurrentSpace(guid, name string, err error) *FakeCliConnectionBuilder {
	space := plugin_models.Space{SpaceFields: plugin_models.SpaceFields{Guid: guid, Name: name}}
	b.cliConn.GetCurrentSpaceReturns(space, err)
	return b
}

func (b *FakeCliConnectionBuilder) Username(username string, err error) *FakeCliConnectionBuilder {
	b.cliConn.UsernameReturns(username, err)
	return b
}

func (b *FakeCliConnectionBuilder) AccessToken(token string, err error) *FakeCliConnectionBuilder {
	b.cliConn.AccessTokenReturns(token, err)
	return b
}

func (b *FakeCliConnectionBuilder) APIEndpoint(apiURL string, err error) *FakeCliConnectionBuilder {
	b.cliConn.ApiEndpointReturns(apiURL, err)
	return b
}

func (b *FakeCliConnectionBuilder) GetApp(name string, app plugin_models.GetAppModel, err error) *FakeCliConnectionBuilder {
	b.cliConn.GetAppReturns(app, err) // TODO
	return b
}

func (b *FakeCliConnectionBuilder) GetApps(apps []plugin_models.GetAppsModel, err error) *FakeCliConnectionBuilder {
	b.cliConn.GetAppsReturns(apps, err)
	return b
}

func (b *FakeCliConnectionBuilder) GetService(name string, service plugin_models.GetService_Model, err error) *FakeCliConnectionBuilder {
	b.cliConn.GetServiceReturns(service, err) // TODO
	return b
}

func (b *FakeCliConnectionBuilder) GetServices(services []plugin_models.GetServices_Model, err error) *FakeCliConnectionBuilder {
	b.cliConn.GetServicesReturns(services, err)
	return b
}

func (b *FakeCliConnectionBuilder) GetSpace(name string, space plugin_models.GetSpace_Model, err error) *FakeCliConnectionBuilder {
	b.cliConn.GetSpaceReturns(space, err) // TODO
	return b
}

// Build builds a FakeCliConnection instance
func (b *FakeCliConnectionBuilder) Build() *fakes.FakeCliConnection {
	return &b.cliConn
}