func PatchCreateContainer(s *testing.CleanupSuite, c *gc.C, expectedURL string) startInstanceFunc {
mockFunc := func(_ *localEnviron, args environs.StartInstanceParams) (instance.Instance, *instance.HardwareCharacteristics, error) {
c.Assert(args.Tools, gc.HasLen, 1)
c.Assert(args.Tools[0].URL, gc.Equals, expectedURL)
return &mockInstance{id: "mock"}, nil, nil
}
s.PatchValue(&createContainer, mockFunc)
return mockFunc
}
func PatchGenerateCertificate(s *testing.CleanupSuite, cert, key string) {
s.PatchValue(&generateCertificate, func() ([]byte, []byte, error) {
return []byte(cert), []byte(key), nil
})
}
// PatchOfficialDataSources is used by tests.
// We replace one of the urls with the supplied value
// and prevent the other from being used.
func PatchOfficialDataSources(s *testing.CleanupSuite, url string) {
s.PatchValue(&imagemetadata.DefaultUbuntuBaseURL, "")
s.PatchValue(&imagemetadata.DefaultJujuBaseURL, url)
}
