本文整理汇总了Golang中github.com/golang/mock/gomock.NewController函数的典型用法代码示例。如果您正苦于以下问题:Golang NewController函数的具体用法?Golang NewController怎么用?Golang NewController使用的例子?那么恭喜您, 这里精选的函数代码示例或许可以为您提供帮助。
在下文中一共展示了NewController函数的20个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于我们的系统推荐出更棒的Golang代码示例。
示例1: TestListVpcOfferingReturnVpcsIfSuccess
func TestListVpcOfferingReturnVpcsIfSuccess(t *testing.T) {
//given
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockEntityService := services_mocks.NewMockEntityService(ctrl)
vpcOfferingService := VpcOfferingApi{
entityService: mockEntityService,
}
expectedVpcOffering1 := VpcOffering{Id: TEST_VPC_OFFERING_ID + "1",
Name: TEST_VPC_OFFERING_NAME + "1",
State: TEST_VPC_OFFERING_STATE + "1",
}
expectedVpcOfferings := []VpcOffering{expectedVpcOffering1}
mockEntityService.EXPECT().List(gomock.Any()).Return(buildListTestVpcOfferingJsonResponse(expectedVpcOfferings), nil)
//when
vpcOfferings, _ := vpcOfferingService.List()
//then
if assert.NotNil(t, vpcOfferings) {
assert.Equal(t, expectedVpcOfferings, vpcOfferings)
}
}
开发者ID:cloud-ca,项目名称:go-cloudca,代码行数:28,代码来源:vpc_offering_test.go
示例2: TestStartAgentNoEnvFile
func TestStartAgentNoEnvFile(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
containerID := "container id"
mockFS := NewMockfileSystem(mockCtrl)
mockDocker := NewMockdockerclient(mockCtrl)
mockFS.EXPECT().ReadFile(config.AgentConfigFile()).Return(nil, errors.New("test error"))
mockDocker.EXPECT().CreateContainer(gomock.Any()).Do(func(opts godocker.CreateContainerOptions) {
validateCommonCreateContainerOptions(opts, t)
}).Return(&godocker.Container{
ID: containerID,
}, nil)
mockDocker.EXPECT().StartContainer(containerID, nil)
mockDocker.EXPECT().WaitContainer(containerID)
client := &Client{
docker: mockDocker,
fs: mockFS,
}
_, err := client.StartAgent()
if err != nil {
t.Error("Error should not be returned")
}
}
开发者ID:chanshunli,项目名称:amazon-ecs-init,代码行数:28,代码来源:docker_test.go
示例3: TestStopAgent
func TestStopAgent(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
mockDocker := NewMockdockerclient(mockCtrl)
mockDocker.EXPECT().ListContainers(godocker.ListContainersOptions{
All: true,
Filters: map[string][]string{
"status": []string{},
},
}).Return([]godocker.APIContainers{
godocker.APIContainers{
Names: []string{"/" + config.AgentContainerName},
ID: "id",
},
}, nil)
mockDocker.EXPECT().StopContainer("id", uint(10))
client := &Client{
docker: mockDocker,
}
err := client.StopAgent()
if err != nil {
t.Error("Error should not be returned")
}
}
开发者ID:chanshunli,项目名称:amazon-ecs-init,代码行数:28,代码来源:docker_test.go
示例4: TestGetComputeOfferingReturnComputeOfferingIfSuccess
func TestGetComputeOfferingReturnComputeOfferingIfSuccess(t *testing.T) {
//given
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockEntityService := services_mocks.NewMockEntityService(ctrl)
computeOfferingService := ComputeOfferingApi{
entityService: mockEntityService,
}
expectedComputeOffering := ComputeOffering{Id: TEST_COMPUTE_OFFERING_ID,
Name: TEST_COMPUTE_OFFERING_NAME,
MemoryInMB: TEST_COMPUTE_OFFERING_MEMORY,
CpuCount: TEST_COMPUTE_OFFERING_CPU_NUMBER,
Custom: TEST_COMPUTE_OFFERING_CUSTOM}
mockEntityService.EXPECT().Get(TEST_COMPUTE_OFFERING_ID, gomock.Any()).Return(buildComputeOfferingJsonResponse(&expectedComputeOffering), nil)
//when
computeOffering, _ := computeOfferingService.Get(TEST_COMPUTE_OFFERING_ID)
//then
if assert.NotNil(t, computeOffering) {
assert.Equal(t, expectedComputeOffering, *computeOffering)
}
}
开发者ID:cloud-ca,项目名称:go-cloudca,代码行数:27,代码来源:compute_offering_test.go
示例5: TestPullImageECRSuccess
func TestPullImageECRSuccess(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockDocker := mock_dockeriface.NewMockClient(ctrl)
mockDocker.EXPECT().Ping().AnyTimes().Return(nil)
factory := mock_dockerclient.NewMockFactory(ctrl)
factory.EXPECT().GetDefaultClient().AnyTimes().Return(mockDocker, nil)
client, _ := NewDockerGoClient(factory, "", config.NewSensitiveRawMessage([]byte{}), false)
goClient, _ := client.(*dockerGoClient)
ecrClientFactory := mock_ecr.NewMockECRFactory(ctrl)
ecrClient := mock_ecr.NewMockECRSDK(ctrl)
goClient.ecrClientFactory = ecrClientFactory
testTime := ttime.NewTestTime()
ttime.SetTime(testTime)
registryId := "123456789012"
region := "eu-west-1"
endpointOverride := "my.endpoint"
authData := &api.RegistryAuthenticationData{
Type: "ecr",
ECRAuthData: &api.ECRAuthData{
RegistryId: registryId,
Region: region,
EndpointOverride: endpointOverride,
},
}
imageEndpoint := "registry.endpoint"
image := imageEndpoint + "/myimage:tag"
username := "username"
password := "password"
dockerAuthConfiguration := docker.AuthConfiguration{
Username: username,
Password: password,
ServerAddress: "https://" + imageEndpoint,
}
getAuthorizationTokenInput := &ecrapi.GetAuthorizationTokenInput{
RegistryIds: []*string{aws.String(registryId)},
}
ecrClientFactory.EXPECT().GetClient(region, endpointOverride).Return(ecrClient)
ecrClient.EXPECT().GetAuthorizationToken(getAuthorizationTokenInput).Return(
&ecrapi.GetAuthorizationTokenOutput{
AuthorizationData: []*ecrapi.AuthorizationData{
&ecrapi.AuthorizationData{
ProxyEndpoint: aws.String("https://" + imageEndpoint),
AuthorizationToken: aws.String(base64.StdEncoding.EncodeToString([]byte(username + ":" + password))),
},
},
}, nil)
mockDocker.EXPECT().PullImage(
&pullImageOptsMatcher{image},
dockerAuthConfiguration,
).Return(nil)
metadata := client.PullImage(image, authData)
if metadata.Error != nil {
t.Error("Expected pull to succeed")
}
}
开发者ID:bmanas,项目名称:amazon-ecs-agent,代码行数:60,代码来源:docker_container_engine_test.go
示例6: TestCallOnewayFailure
func TestCallOnewayFailure(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
ctx := context.Background()
caller := "caller"
service := "service"
procedure := "procedure"
body := []byte{1, 2, 3}
outbound := transporttest.NewMockOnewayOutbound(mockCtrl)
client := New(channel.MultiOutbound(caller, service,
transport.Outbounds{
Oneway: outbound,
}))
outbound.EXPECT().CallOneway(gomock.Any(),
transporttest.NewRequestMatcher(t,
&transport.Request{
Service: service,
Caller: caller,
Procedure: procedure,
Encoding: Encoding,
Body: bytes.NewReader(body),
}),
).Return(nil, errors.New("some error"))
_, err := client.CallOneway(
ctx,
yarpc.NewReqMeta().Procedure(procedure),
body)
assert.Error(t, err)
}
开发者ID:yarpc,项目名称:yarpc-go,代码行数:35,代码来源:outbound_test.go
示例7: TestBasicSuccessful
func TestBasicSuccessful(t *testing.T) {
pub, leaf, k := setup(t)
ctrl := gomock.NewController(t)
defer ctrl.Finish()
scope := mock_metrics.NewMockScope(ctrl)
pub.stats = scope
server := logSrv(leaf.Raw, k)
defer server.Close()
port, err := getPort(server)
test.AssertNotError(t, err, "Failed to get test server port")
addLog(t, pub, port, &k.PublicKey)
statName := pub.ctLogs[0].statName
log.Clear()
scope.EXPECT().NewScope(statName).Return(scope)
scope.EXPECT().Inc("Submits", int64(1)).Return(nil)
scope.EXPECT().TimingDuration("SubmitLatency", gomock.Any()).Return(nil)
err = pub.SubmitToCT(ctx, leaf.Raw)
test.AssertNotError(t, err, "Certificate submission failed")
test.AssertEquals(t, len(log.GetAllMatching("Failed to.*")), 0)
// No Intermediate
pub.issuerBundle = []ct.ASN1Cert{}
log.Clear()
scope.EXPECT().NewScope(statName).Return(scope)
scope.EXPECT().Inc("Submits", int64(1)).Return(nil)
scope.EXPECT().TimingDuration("SubmitLatency", gomock.Any()).Return(nil)
err = pub.SubmitToCT(ctx, leaf.Raw)
test.AssertNotError(t, err, "Certificate submission failed")
test.AssertEquals(t, len(log.GetAllMatching("Failed to.*")), 0)
}
开发者ID:jfrazelle,项目名称:boulder,代码行数:33,代码来源:publisher_test.go
示例8: mocks
func mocks(t *testing.T, cfg *config.Config) (*gomock.Controller, *MockDockerClient, TaskEngine) {
ctrl := gomock.NewController(t)
client := NewMockDockerClient(ctrl)
taskEngine := NewTaskEngine(cfg, false)
taskEngine.(*DockerTaskEngine).SetDockerClient(client)
return ctrl, client, taskEngine
}
开发者ID:appuri,项目名称:amazon-ecs-agent,代码行数:7,代码来源:docker_task_engine_test.go
示例9: TestHelpRun2
func TestHelpRun2(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
cmd := NewMockCommander(ctrl)
rem := remote.NewMockRemoter(ctrl)
srv := server.NewMockServer(ctrl)
cfg := core.NewMockConfig(ctrl)
mmi := ui.NewMockUserInteraction(ctrl)
help := &cmd_help{cmd, rem, srv, cfg, mmi}
cmd_foo := NewMockCommand(ctrl)
cmd.EXPECT().Command("foo").Return(cmd_foo)
cmd_foo.EXPECT().Help([]string{"help", "foo"}).Return("\nhelp for foo\n", nil)
mmi.EXPECT().Pager(`
[1;32mHelp for command foo[0m
help for foo
[1m--------[0m
[32mGate Copyright (C) 2012-2015 Cyril Adrian <[email protected]>[0m
[32mThis program comes with ABSOLUTELY NO WARRANTY; for details type [33mshow w[32m.[0m
[32mThis is free software, and you are welcome to redistribute it[0m
[32munder certain conditions; type [33mshow c[32m for details.[0m
`)
err := help.Run([]string{"help", "foo"})
if err != nil {
t.Error(err)
}
}
开发者ID:cadrian,项目名称:gate,代码行数:34,代码来源:cmd_help_test.go
示例10: TestAcsWsUrl
func TestAcsWsUrl(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
taskEngine := engine.NewMockTaskEngine(ctrl)
taskEngine.EXPECT().Version().Return("Docker version result", nil)
wsurl := handler.AcsWsUrl("http://endpoint.tld", "myCluster", "myContainerInstance", taskEngine)
parsed, err := url.Parse(wsurl)
if err != nil {
t.Fatal("Should be able to parse url")
}
if parsed.Path != "/ws" {
t.Fatal("Wrong path")
}
if parsed.Query().Get("clusterArn") != "myCluster" {
t.Fatal("Wrong cluster")
}
if parsed.Query().Get("containerInstanceArn") != "myContainerInstance" {
t.Fatal("Wrong cluster")
}
if parsed.Query().Get("agentVersion") != version.Version {
t.Fatal("Wrong cluster")
}
if parsed.Query().Get("agentHash") != version.GitHashString() {
t.Fatal("Wrong cluster")
}
if parsed.Query().Get("dockerVersion") != "Docker version result" {
t.Fatal("Wrong docker version")
}
}
开发者ID:bmanas,项目名称:amazon-ecs-agent,代码行数:35,代码来源:acs_handler_test.go
示例11: TestSeekAndReadWithRetries
// Testing retry logic for Read()
func TestSeekAndReadWithRetries(t *testing.T) {
mockCtrl := gomock.NewController(t)
hdfsReader := NewMockReadSeekCloser(mockCtrl)
hdfsAccessor := NewMockHdfsAccessor(mockCtrl)
ftHdfsReader := NewFaultTolerantHdfsReader("/path/to/file", hdfsReader, hdfsAccessor, atMost2Attempts())
var err error
var nr int
// Performing succesfull read of 60 bytes of requested 100 at offset 1000
hdfsReader.EXPECT().Seek(int64(1000)).Return(nil)
err = ftHdfsReader.Seek(1000)
assert.Nil(t, err)
hdfsReader.EXPECT().Read(gomock.Any()).Return(60, nil)
nr, err = ftHdfsReader.Read(make([]byte, 100))
assert.Nil(t, err)
assert.Equal(t, 60, nr)
// Now the stream should be at position 160
// Requesting one more read of 200 bytes, but this time it will fail
hdfsReader.EXPECT().Read(gomock.Any()).Return(0, errors.New("Injected failure"))
// As a result, ftHdfsReader should close the stream...
hdfsReader.EXPECT().Close().Return(nil)
// ...and invoke an OpenRead() to get new HdfsReader
newHdfsReader := NewMockReadSeekCloser(mockCtrl)
hdfsAccessor.EXPECT().OpenRead("/path/to/file").Return(newHdfsReader, nil)
// It should seek at corret position (1060), and repeat the read
newHdfsReader.EXPECT().Seek(int64(1060)).Return(nil)
newHdfsReader.EXPECT().Read(gomock.Any()).Return(150, nil)
nr, err = ftHdfsReader.Read(make([]byte, 200))
assert.Nil(t, err)
assert.Equal(t, 150, nr)
}
开发者ID:Microsoft,项目名称:hdfs-mount,代码行数:33,代码来源:FaultTolerantHdfsReader_test.go
示例12: TestWriterInsert
func TestWriterInsert(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
cli := mock_cassandra.NewMockCassandra(ctrl)
//expectingBatch := thrift.NewTMap(k, v, s)
cli.EXPECT().BatchMutate(gomock.Any(), ConsistencyLevel_ONE)
conn := &connection{
socket: nil,
transport: nil,
client: cli,
node: &node{node: "node"},
}
cp := &stubTransactionRunner{conn: conn}
w := newWriter(cp, CONSISTENCY_ONE)
row := &Row{
Key: []byte("rowkey"),
Columns: []*Column{
&Column{Name: []byte("name1"), Value: []byte("value1")},
&Column{Name: []byte("name2"), Value: []byte("value2")},
},
}
w.Insert("cf", row)
e := w.Run()
if e != nil {
t.Error("Error", e)
}
}
开发者ID:apesternikov,项目名称:gossie,代码行数:27,代码来源:writer_test.go
示例13: TestFunction_Rollback_UpdateAlias_failed
func TestFunction_Rollback_UpdateAlias_failed(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
serviceMock := mock_lambdaiface.NewMockLambdaAPI(mockCtrl)
serviceMock.EXPECT().GetAlias(gomock.Any()).Return(&lambda.AliasConfiguration{FunctionVersion: aws.String("1")}, nil)
serviceMock.EXPECT().ListVersionsByFunction(gomock.Any()).Return(&lambda.ListVersionsByFunctionOutput{
Versions: []*lambda.FunctionConfiguration{
&lambda.FunctionConfiguration{Version: aws.String("$LATEST")},
&lambda.FunctionConfiguration{Version: aws.String("1")},
&lambda.FunctionConfiguration{Version: aws.String("2")},
},
}, nil)
serviceMock.EXPECT().UpdateAlias(gomock.Any()).Return(nil, errors.New("API err"))
fn := &function.Function{
FunctionName: "testfn",
Alias: "current",
Service: serviceMock,
Log: log.Log,
}
err := fn.Rollback()
assert.EqualError(t, err, "API err")
}
开发者ID:Robh081956,项目名称:apex,代码行数:25,代码来源:function_test.go
示例14: mocks
func mocks(t *testing.T, cfg *config.Config) (*gomock.Controller, *mock_engine.MockDockerClient, engine.TaskEngine) {
ctrl := gomock.NewController(t)
client := mock_engine.NewMockDockerClient(ctrl)
taskEngine := engine.NewTaskEngine(cfg)
taskEngine.(*engine.DockerTaskEngine).SetDockerClient(client)
return ctrl, client, taskEngine
}
开发者ID:nickschuch,项目名称:amazon-ecs-agent,代码行数:7,代码来源:docker_task_engine_test.go
示例15: TestSubmitContainerStateChangeLongReason
func TestSubmitContainerStateChangeLongReason(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
client, _, mockSubmitStateClient := NewMockClient(mockCtrl)
exitCode := 20
trimmedReason := strings.Repeat("a", api.EcsMaxReasonLength)
reason := strings.Repeat("a", api.EcsMaxReasonLength+1)
mockSubmitStateClient.EXPECT().SubmitContainerStateChange(&containerSubmitInputMatcher{
ecs.SubmitContainerStateChangeInput{
Cluster: strptr(configuredCluster),
Task: strptr("arn"),
ContainerName: strptr("cont"),
Status: strptr("STOPPED"),
ExitCode: int64ptr(&exitCode),
Reason: strptr(trimmedReason),
NetworkBindings: []*ecs.NetworkBinding{},
},
})
err := client.SubmitContainerStateChange(api.ContainerStateChange{
TaskArn: "arn",
ContainerName: "cont",
Status: api.ContainerStopped,
ExitCode: &exitCode,
Reason: reason,
})
if err != nil {
t.Errorf("Unable to submit container state change: %v", err)
}
}
开发者ID:rafkhan,项目名称:amazon-ecs-agent,代码行数:30,代码来源:api_client_test.go
示例16: TestClusterScale
func TestClusterScale(t *testing.T) {
newCliParams = func(context *cli.Context, rdwr config.ReadWriter) (*config.CliParams, error) {
return &config.CliParams{
Cluster: clusterName,
}, nil
}
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockEcs := mock_ecs.NewMockECSClient(ctrl)
mockCloudformation := mock_cloudformation.NewMockCloudformationClient(ctrl)
mockEcs.EXPECT().Initialize(gomock.Any())
mockEcs.EXPECT().IsActiveCluster(gomock.Any()).Return(true, nil)
mockCloudformation.EXPECT().Initialize(gomock.Any())
mockCloudformation.EXPECT().ValidateStackExists(gomock.Any()).Return(nil)
mockCloudformation.EXPECT().UpdateStack(gomock.Any(), gomock.Any()).Return("", nil)
mockCloudformation.EXPECT().WaitUntilUpdateComplete(gomock.Any()).Return(nil)
globalSet := flag.NewFlagSet("ecs-cli", 0)
globalSet.String("region", "us-west-1", "")
globalContext := cli.NewContext(nil, globalSet, nil)
flagSet := flag.NewFlagSet("ecs-cli-down", 0)
flagSet.Bool(capabilityIAMFlag, true, "")
flagSet.String(asgMaxSizeFlag, "1", "")
context := cli.NewContext(nil, flagSet, globalContext)
err := scaleCluster(context, &mockReadWriter{}, mockEcs, mockCloudformation)
if err != nil {
t.Fatal("Error scaling cluster: ", err)
}
}
开发者ID:jloper3,项目名称:amazon-ecs-cli,代码行数:33,代码来源:cluster_app_test.go
示例17: TestTranscribeWithIBM
func TestTranscribeWithIBM(t *testing.T) {
assert := assert.New(t)
ctrl := gomock.NewController(t)
defer ctrl.Finish()
// Setup the mock package
websocket.MOCK().SetController(ctrl)
mockos.MOCK().SetController(ctrl)
bufio.MOCK().SetController(ctrl)
mockWS := new(websocket.Conn)
mockFile := new(mockos.File)
mockReader := new(bufio.Reader)
websocket.DefaultDialer.EXPECT().Dial(gomock.Any(), gomock.Any()).Return(mockWS, nil, nil).Times(1)
mockWS.EXPECT().WriteJSON(gomock.Any()).MinTimes(1)
mockos.EXPECT().Open("file.flac").Times(1).Return(mockFile, nil).Times(1)
bufio.EXPECT().NewReader(mockFile).Return(mockReader).Times(1)
mockReader.EXPECT().Read(gomock.Any()).Return(0, nil).Times(1)
mockWS.EXPECT().WriteMessage(websocket.BinaryMessage, gomock.Any()).MinTimes(1)
mockWS.EXPECT().ReadJSON(gomock.Any()).Do(func(res **IBMResult) {
(*res).Results = []ibmResultField{
ibmResultField{},
ibmResultField{},
}
}).Times(1)
mockWS.EXPECT().Close().Times(1) // could run many times
_, err := TranscribeWithIBM("file.flac", "", "")
assert.NoError(err)
}
开发者ID:hack4impact,项目名称:transcribe4all,代码行数:31,代码来源:ibm_test.go
示例18: TestClusterPSTaskGetInfoFail
func TestClusterPSTaskGetInfoFail(t *testing.T) {
newCliParams = func(context *cli.Context, rdwr config.ReadWriter) (*config.CliParams, error) {
return &config.CliParams{
Cluster: clusterName,
}, nil
}
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockEcs := mock_ecs.NewMockECSClient(ctrl)
mockEcs.EXPECT().Initialize(gomock.Any())
mockEcs.EXPECT().IsActiveCluster(gomock.Any()).Return(true, nil)
mockEcs.EXPECT().GetTasksPages(gomock.Any(), gomock.Any()).Do(func(x, y interface{}) {
}).Return(errors.New("error"))
globalSet := flag.NewFlagSet("ecs-cli", 0)
globalSet.String("region", "us-west-1", "")
globalContext := cli.NewContext(nil, globalSet, nil)
flagSet := flag.NewFlagSet("ecs-cli-down", 0)
context := cli.NewContext(nil, flagSet, globalContext)
_, err := clusterPS(context, &mockReadWriter{}, mockEcs)
if err == nil {
t.Fatal("Expected error in cluster ps")
}
}
开发者ID:jloper3,项目名称:amazon-ecs-cli,代码行数:27,代码来源:cluster_app_test.go
示例19: TestFunction_Rollback_latestVersion
func TestFunction_Rollback_latestVersion(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
serviceMock := mock_lambdaiface.NewMockLambdaAPI(mockCtrl)
deployedVersions := []*lambda.FunctionConfiguration{
&lambda.FunctionConfiguration{Version: aws.String("$LATEST")},
&lambda.FunctionConfiguration{Version: aws.String("1")},
&lambda.FunctionConfiguration{Version: aws.String("2")},
}
currentVersion := aws.String("1")
afterRollbackVersion := aws.String("2")
serviceMock.EXPECT().GetAlias(gomock.Any()).Return(&lambda.AliasConfiguration{FunctionVersion: currentVersion}, nil)
serviceMock.EXPECT().ListVersionsByFunction(gomock.Any()).Return(&lambda.ListVersionsByFunctionOutput{Versions: deployedVersions}, nil)
serviceMock.EXPECT().UpdateAlias(&lambda.UpdateAliasInput{
FunctionName: aws.String("testfn"),
Name: aws.String("current"),
FunctionVersion: afterRollbackVersion,
})
fn := &function.Function{
FunctionName: "testfn",
Service: serviceMock,
Log: log.Log,
}
err := fn.Rollback()
assert.Nil(t, err)
}
开发者ID:kujohn,项目名称:apex,代码行数:29,代码来源:function_test.go
示例20: TestClusterUp
func TestClusterUp(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockEcs := mock_ecs.NewMockECSClient(ctrl)
mockCloudformation := mock_cloudformation.NewMockCloudformationClient(ctrl)
mockEcs.EXPECT().Initialize(gomock.Any())
mockEcs.EXPECT().CreateCluster(gomock.Any()).Do(func(in interface{}) {
if in.(string) != clusterName {
t.Fatal("Expected to be called with " + clusterName + " not " + in.(string))
}
}).Return(clusterName, nil)
mockCloudformation.EXPECT().Initialize(gomock.Any())
mockCloudformation.EXPECT().ValidateStackExists(gomock.Any()).Return(errors.New("error"))
mockCloudformation.EXPECT().CreateStack(gomock.Any(), gomock.Any(), gomock.Any()).Return("", nil)
mockCloudformation.EXPECT().WaitUntilCreateComplete(gomock.Any()).Return(nil)
globalSet := flag.NewFlagSet("ecs-cli", 0)
globalSet.String("region", "us-west-1", "")
globalContext := cli.NewContext(nil, globalSet, nil)
flagSet := flag.NewFlagSet("ecs-cli-up", 0)
flagSet.Bool(capabilityIAMFlag, true, "")
flagSet.String(keypairNameFlag, "default", "")
context := cli.NewContext(nil, flagSet, globalContext)
err := createCluster(context, &mockReadWriter{}, mockEcs, mockCloudformation, ami.NewStaticAmiIds())
if err != nil {
t.Fatal("Error bringing up cluster: ", err)
}
}
开发者ID:jloper3,项目名称:amazon-ecs-cli,代码行数:32,代码来源:cluster_app_test.go
注:本文中的github.com/golang/mock/gomock.NewController函数示例整理自Github/MSDocs等源码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。 |
请发表评论