func (s *environSuite) TestGetEnvironment(c *gc.C) {
env, err := utils.GetEnvironment(s.State)
c.Assert(err, jc.ErrorIsNil)
config, err := s.State.EnvironConfig()
c.Assert(err, jc.ErrorIsNil)
c.Check(env.Config(), jc.DeepEquals, config)
c.Check(env, gc.Not(gc.Equals), s.Environ)
}
// stateStepsFor126 returns upgrade steps for Juju 1.26 that manipulate state directly.
func stateStepsFor126() []Step {
return []Step{
&upgradeStep{
description: "add the version field to all settings docs",
targets: []Target{DatabaseMaster},
run: func(context Context) error {
return state.MigrateSettingsSchema(context.State())
},
},
&upgradeStep{
description: "add status to filesystem",
targets: []Target{DatabaseMaster},
run: func(context Context) error {
return state.AddFilesystemStatus(context.State())
},
},
&upgradeStep{
description: "upgrade environment config",
targets: []Target{DatabaseMaster},
run: func(context Context) error {
// TODO(axw) updateEnvironConfig should be
// called for all upgrades, to decouple this
// package from provider-specific upgrades.
st := context.State()
return upgradeEnvironConfig(st, st, environs.GlobalProviderRegistry())
},
},
//TODO(perrito666) make this an unconditional upgrade step.
// it would be ideal not to have to modify this package whenever we add provider upgrade steps.
&upgradeStep{
description: "provider side upgrades",
targets: []Target{DatabaseMaster},
run: func(context Context) error {
st := context.State()
env, err := utils.GetEnvironment(st)
if err != nil {
return errors.Annotate(err, "getting provider for upgrade")
}
return upgradeProviderChanges(env, st, version.Number{Major: 1, Minor: 26})
},
},
&upgradeStep{
description: "update machine preferred addresses",
targets: []Target{DatabaseMaster},
run: func(context Context) error {
return state.AddPreferredAddressesToMachines(context.State())
},
},
}
}