// useTestImageData causes the given content to be served when published metadata is requested. func useTestImageData(c *gc.C, files map[string]string) { if files != nil { sstesting.SetRoundTripperFiles(sstesting.AddSignedFiles(c, files), nil) } else { sstesting.SetRoundTripperFiles(nil, nil) } }
func (s *signedSuite) SetUpSuite(c *gc.C) { s.BaseSuite.SetUpSuite(c) var imageData = map[string]string{ "/unsigned/streams/v1/index.json": unsignedIndex, "/unsigned/streams/v1/tools_metadata.json": unsignedProduct, } // Set up some signed data from the unsigned data. // Overwrite the product path to use the sjson suffix. rawUnsignedIndex := strings.Replace( unsignedIndex, "streams/v1/tools_metadata.json", "streams/v1/tools_metadata.sjson", -1) r := bytes.NewReader([]byte(rawUnsignedIndex)) signedData, err := simplestreams.Encode( r, sstesting.SignedMetadataPrivateKey, sstesting.PrivateKeyPassphrase) c.Assert(err, jc.ErrorIsNil) imageData["/signed/streams/v1/index.sjson"] = string(signedData) // Replace the tools path in the unsigned data with a different one so we can test that the right // tools path is used. rawUnsignedProduct := strings.Replace( unsignedProduct, "juju-1.13.0", "juju-1.13.1", -1) r = bytes.NewReader([]byte(rawUnsignedProduct)) signedData, err = simplestreams.Encode( r, sstesting.SignedMetadataPrivateKey, sstesting.PrivateKeyPassphrase) c.Assert(err, jc.ErrorIsNil) imageData["/signed/streams/v1/tools_metadata.sjson"] = string(signedData) sstesting.SetRoundTripperFiles(imageData, map[string]int{"signedtest://unauth": http.StatusUnauthorized}) s.PatchValue(&juju.JujuPublicKey, sstesting.SignedMetadataPublicKey) }
func (s *signedSuite) SetUpSuite(c *gc.C) { var imageData = map[string]string{ "/unsigned/streams/v1/index.json": unsignedIndex, "/unsigned/streams/v1/image_metadata.json": unsignedProduct, } // Set up some signed data from the unsigned data. // Overwrite the product path to use the sjson suffix. rawUnsignedIndex := strings.Replace( unsignedIndex, "streams/v1/image_metadata.json", "streams/v1/image_metadata.sjson", -1) r := bytes.NewReader([]byte(rawUnsignedIndex)) signedData, err := simplestreams.Encode( r, sstesting.SignedMetadataPrivateKey, sstesting.PrivateKeyPassphrase) c.Assert(err, jc.ErrorIsNil) imageData["/signed/streams/v1/index.sjson"] = string(signedData) // Replace the image id in the unsigned data with a different one so we can test that the right // image id is used. rawUnsignedProduct := strings.Replace( unsignedProduct, "ami-26745463", "ami-123456", -1) r = bytes.NewReader([]byte(rawUnsignedProduct)) signedData, err = simplestreams.Encode( r, sstesting.SignedMetadataPrivateKey, sstesting.PrivateKeyPassphrase) c.Assert(err, jc.ErrorIsNil) imageData["/signed/streams/v1/image_metadata.sjson"] = string(signedData) sstesting.SetRoundTripperFiles(imageData, map[string]int{"test://unauth": http.StatusUnauthorized}) s.origKey = imagemetadata.SetSigningPublicKey(sstesting.SignedMetadataPublicKey) }
// Set Metadata requests to be served by the filecontent supplied. func UseExternalTestImageMetadata(c *gc.C, creds *auth.Credentials) { metadata := parseIndexData(creds) files := map[string]string{ "/streams/v1/index.json": metadata.String(), "/streams/v1/com.ubuntu.cloud:released:joyent.json": imagesData, } sstesting.SetRoundTripperFiles(sstesting.AddSignedFiles(c, files), nil) }
func (s *signedSuite) TearDownSuite(c *gc.C) { sstesting.SetRoundTripperFiles(nil, nil) imagemetadata.SetSigningPublicKey(s.origKey) }
func (s *signedSuite) TearDownSuite(c *gc.C) { sstesting.SetRoundTripperFiles(nil, nil) s.BaseSuite.TearDownSuite(c) }
// UnregisterMachinesEndpoint resets the machines endpoint. func UnregisterMachinesEndpoint() { sstesting.SetRoundTripperFiles(nil, nil) }
// RegisterMachinesEndpoint creates a fake endpoint so that // machines api calls succeed. func RegisterMachinesEndpoint() { files := map[string]string{ "/test/machines": "", } sstesting.SetRoundTripperFiles(files, nil) }
func UnregisterExternalTestImageMetadata() { sstesting.SetRoundTripperFiles(nil, nil) }