func (f *factory) loadTarballProvider() bitarball.Provider { if f.tarballProvider != nil { return f.tarballProvider } tarballCacheBasePath := filepath.Join(f.workspaceRootPath, "downloads") tarballCache := bitarball.NewCache(tarballCacheBasePath, f.fs, f.logger) httpClient := bihttpclient.NewHTTPClient(bitarball.HTTPClient, f.logger) sha1Calculator := bicrypto.NewSha1Calculator(f.fs) f.tarballProvider = bitarball.NewProvider(tarballCache, f.fs, httpClient, sha1Calculator, 3, 500*time.Millisecond, f.logger) return f.tarballProvider }
func (f *agentClientFactory) NewAgentClient(directorID, mbusURL string) agentclient.AgentClient { httpClient := httpclient.NewHTTPClient(httpclient.DefaultClient, f.logger) return NewAgentClient(mbusURL, directorID, f.getTaskDelay, 10, httpClient, f.logger) }
mockCloud.EXPECT().AttachDisk(newVMCID, newDiskCID), mockAgentClient.EXPECT().MountDisk(newDiskCID), mockAgentClient.EXPECT().MigrateDisk(), mockCloud.EXPECT().DetachDisk(newVMCID, oldDiskCID), mockCloud.EXPECT().DeleteDisk(oldDiskCID), // start jobs & wait for running mockAgentClient.EXPECT().Stop(), mockAgentClient.EXPECT().Apply(applySpec), mockAgentClient.EXPECT().Start(), mockAgentClient.EXPECT().GetState().Return(agentRunningState, nil), ) } var expectRegistryToWork = func() { httpClient := bihttp.NewHTTPClient(bihttp.DefaultClient, logger) endpoint := "http://*****:*****@127.0.0.1:6901/instances/fake-director-id/settings" settingsBytes := []byte("fake-registry-contents") //usually json, but not required to be response, err := httpClient.Put(endpoint, settingsBytes) Expect(err).ToNot(HaveOccurred()) Expect(response.StatusCode).To(Equal(http.StatusCreated)) response, err = httpClient.Get(endpoint) Expect(err).ToNot(HaveOccurred()) Expect(response.StatusCode).To(Equal(http.StatusOK)) responseBytes, err := ioutil.ReadAll(response.Body) Expect(err).ToNot(HaveOccurred()) Expect(responseBytes).To(Equal([]byte("{\"settings\":\"fake-registry-contents\",\"status\":\"ok\"}")))