mockAgentClient = mock_agentclient.NewMockAgentClient(mockCtrl) mockAgentClientFactory.EXPECT().NewAgentClient(gomock.Any(), gomock.Any()).Return(mockAgentClient).AnyTimes() mockVMManagerFactory = mock_vm.NewMockManagerFactory(mockCtrl) fakeVMManager = fakebivm.NewFakeManager() mockVMManagerFactory.EXPECT().NewManager(cloud, mockAgentClient).Return(fakeVMManager).AnyTimes() fakeSSHTunnelFactory = fakebisshtunnel.NewFakeFactory() fakeSSHTunnel = fakebisshtunnel.NewFakeTunnel() fakeSSHTunnelFactory.SSHTunnel = fakeSSHTunnel fakeSSHTunnel.SetStartBehavior(nil, nil) fakeVM = fakebivm.NewFakeVM("fake-vm-cid") fakeVMManager.CreateVM = fakeVM fakeVM.AgentClientReturn = mockAgentClient logger := boshlog.NewLogger(boshlog.LevelNone) fakeStage = fakebiui.NewFakeStage() fakeStemcellRepo := fakebiconfig.NewFakeStemcellRepo() stemcellRecord := biconfig.StemcellRecord{ ID: "fake-stemcell-id", Name: "fake-stemcell-name", Version: "fake-stemcell-version", CID: "fake-stemcell-cid", } err := fakeStemcellRepo.SetFindBehavior("fake-stemcell-name", "fake-stemcell-version", stemcellRecord, true, nil) Expect(err).ToNot(HaveOccurred()) cloudStemcell = bistemcell.NewCloudStemcell(stemcellRecord, fakeStemcellRepo, cloud)
{ Name: "fake-job-name", PersistentDiskPool: "fake-persistent-disk-pool-name", Instances: 1, }, }, } fakeCloudStemcell = fakebistemcell.NewFakeCloudStemcell("fake-stemcell-cid", "fake-stemcell-name", "fake-stemcell-version") registry = biinstallmanifest.Registry{} fakeVM = fakebivm.NewFakeVM("fake-vm-cid") fakeVMManager.CreateVM = fakeVM mockAgentClient = mock_agentclient.NewMockAgentClient(mockCtrl) fakeVM.AgentClientReturn = mockAgentClient expectedInstance = NewInstance( "fake-job-name", 0, fakeVM, fakeVMManager, fakeSSHTunnelFactory, mockStateBuilder, logger, ) expectedDisk = fakebidisk.NewFakeDisk("fake-disk-cid") fakeVM.UpdateDisksDisks = []bidisk.Disk{expectedDisk} })