// 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)
}
