func (r *restoreRootSuite) TestFindAllowedMethodWhenPreparing(c *gc.C) {
root := apiserver.TestingAboutToRestoreRoot(nil)
caller, err := root.FindMethod("Backups", 1, "Restore")
c.Assert(err, jc.ErrorIsNil)
c.Assert(caller, gc.NotNil)
}
func (r *restoreRootSuite) TestFindDisallowedMethodWhenPreparing(c *gc.C) {
root := apiserver.TestingAboutToRestoreRoot(nil)
caller, err := root.FindMethod("Service", 3, "Deploy")
c.Assert(err, gc.ErrorMatches, "juju restore is in progress - Juju functionality is limited to avoid data loss")
c.Assert(caller, gc.IsNil)
}
func (r *restoreRootSuite) TestFindAllowedMethodWhenPreparing(c *gc.C) {
root := apiserver.TestingAboutToRestoreRoot(nil)
// TODO(perrito666): Uncomment when Restore lands and delete
// the following line.
//caller, err := root.FindMethod("Backups", 0, "Restore")
caller, err := root.FindMethod("Client", 0, "FullStatus")
c.Assert(err, jc.ErrorIsNil)
c.Assert(caller, gc.NotNil)
}
func (r *restrictRestoreSuite) TestNotAllowed(c *gc.C) {
root := apiserver.TestingAboutToRestoreRoot()
caller, err := root.FindMethod("Application", 1, "Deploy")
c.Assert(err, gc.ErrorMatches, "juju restore is in progress - functionality is limited to avoid data loss")
c.Assert(caller, gc.IsNil)
}
func (r *restrictRestoreSuite) TestAllowed(c *gc.C) {
root := apiserver.TestingAboutToRestoreRoot()
caller, err := root.FindMethod("Backups", 1, "Restore")
c.Assert(err, jc.ErrorIsNil)
c.Assert(caller, gc.NotNil)
}