func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigProviderNotConfigUpgrader(c *gc.C) {
s.registry.providers["someprovider"] = &mockEnvironProvider{}
err := upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, jc.ErrorIsNil)
s.registry.CheckCalls(c, []testing.StubCall{{
FuncName: "Provider", Args: []interface{}{"someprovider"},
}})
s.stub.CheckCallNames(c, "EnvironConfig")
}
func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigUpgradeConfigError(c *gc.C) {
s.registry.providers["someprovider"] = &mockEnvironConfigUpgrader{
upgradeConfig: func(cfg *config.Config) (*config.Config, error) {
return nil, errors.New("cannot upgrade config")
},
}
err := upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, gc.ErrorMatches, "upgrading config: cannot upgrade config")
s.stub.CheckCallNames(c, "EnvironConfig")
}
func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigUpdateConfigError(c *gc.C) {
s.stub.SetErrors(nil, errors.New("cannot update environ config"))
s.registry.providers["someprovider"] = &mockEnvironConfigUpgrader{
upgradeConfig: func(cfg *config.Config) (*config.Config, error) {
return cfg, nil
},
}
err := upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, gc.ErrorMatches, "updating config in state: cannot update environ config")
s.stub.CheckCallNames(c, "EnvironConfig", "UpdateEnvironConfig")
updateCall := s.stub.Calls()[1]
c.Assert(updateCall.Args, gc.HasLen, 3)
c.Assert(updateCall.Args[0], jc.DeepEquals, s.cfg.AllAttrs())
c.Assert(updateCall.Args[1], gc.IsNil)
c.Assert(updateCall.Args[2], gc.IsNil)
}
func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigProviderConfigUpgrader(c *gc.C) {
var err error
s.cfg, err = s.cfg.Apply(map[string]interface{}{"test-key": "test-value"})
c.Assert(err, jc.ErrorIsNil)
s.registry.providers["someprovider"] = &mockEnvironConfigUpgrader{
upgradeConfig: func(cfg *config.Config) (*config.Config, error) {
return cfg.Remove([]string{"test-key"})
},
}
err = upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, jc.ErrorIsNil)
s.stub.CheckCallNames(c, "EnvironConfig", "UpdateEnvironConfig")
updateCall := s.stub.Calls()[1]
expectedAttrs := s.cfg.AllAttrs()
delete(expectedAttrs, "test-key")
c.Assert(updateCall.Args, gc.HasLen, 3)
c.Assert(updateCall.Args[0], jc.DeepEquals, expectedAttrs)
c.Assert(updateCall.Args[1], jc.SameContents, []string{"test-key"})
c.Assert(updateCall.Args[2], gc.IsNil)
}
func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigProviderNotRegistered(c *gc.C) {
s.registry.SetErrors(errors.New(`no registered provider for "someprovider"`))
err := upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, gc.ErrorMatches, `getting provider: no registered provider for "someprovider"`)
s.stub.CheckCallNames(c, "EnvironConfig")
}
func (s *upgradeEnvironConfigSuite) TestUpgradeEnvironConfigEnvironConfigError(c *gc.C) {
s.stub.SetErrors(errors.New("cannot read environ config"))
err := upgrades.UpgradeEnvironConfig(s.reader, s.updater, s.registry)
c.Assert(err, gc.ErrorMatches, "reading environment config: cannot read environ config")
s.stub.CheckCallNames(c, "EnvironConfig")
}