func (s *RemoveServiceSuite) SetUpTest(c *gc.C) {
s.RepoSuite.SetUpTest(c)
s.CmdBlockHelper = testing.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
s.stub = &jutesting.Stub{}
s.budgetAPIClient = &mockBudgetAPIClient{Stub: s.stub}
s.PatchValue(&getBudgetAPIClient, func(*httpbakery.Client) budgetAPIClient { return s.budgetAPIClient })
}
func (s *UpgradeJujuSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
s.resources = common.NewResources()
s.authoriser = apiservertesting.FakeAuthorizer{
Tag: s.AdminUserTag(c),
}
s.CmdBlockHelper = coretesting.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
}
func (s *UpgradeCharmSuccessStateSuite) SetUpTest(c *gc.C) {
s.RepoSuite.SetUpTest(c)
s.path = testcharms.Repo.ClonedDirPath(s.CharmsPath, "riak")
err := runDeploy(c, s.path, "--series", "quantal")
c.Assert(err, jc.ErrorIsNil)
s.riak, err = s.State.Application("riak")
c.Assert(err, jc.ErrorIsNil)
ch, forced, err := s.riak.Charm()
c.Assert(err, jc.ErrorIsNil)
c.Assert(ch.Revision(), gc.Equals, 7)
c.Assert(forced, jc.IsFalse)
s.CmdBlockHelper = coretesting.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
}
// JujuConnSuite very helpfully uploads some default
// tools to the environment's storage. We don't want
// 'em there; but we do want a consistent default-series
// in the environment state.
func (s *UpgradeJujuSuite) Reset(c *gc.C) {
s.JujuConnSuite.Reset(c)
envtesting.RemoveTools(c, s.DefaultToolsStorage, s.Environ.Config().AgentStream())
updateAttrs := map[string]interface{}{
"default-series": "raring",
"agent-version": "1.2.3",
}
err := s.State.UpdateModelConfig(updateAttrs, nil, nil)
c.Assert(err, jc.ErrorIsNil)
s.PatchValue(&sync.BuildAgentTarball, toolstesting.GetMockBuildTools(c))
// Set API host ports so FindTools works.
hostPorts := [][]network.HostPort{
network.NewHostPorts(1234, "0.1.2.3"),
}
err = s.State.SetAPIHostPorts(hostPorts)
c.Assert(err, jc.ErrorIsNil)
s.CmdBlockHelper = coretesting.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
}
func (s *UnexposeSuite) SetUpTest(c *gc.C) {
s.RepoSuite.SetUpTest(c)
s.CmdBlockHelper = testing.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
}
func (s *keySuiteBase) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
s.CmdBlockHelper = coretesting.NewCmdBlockHelper(s.APIState)
c.Assert(s.CmdBlockHelper, gc.NotNil)
s.AddCleanup(func(*gc.C) { s.CmdBlockHelper.Close() })
}