func NewRebootWaiter(apistate *api.State, acfg agent.Config) (*Reboot, error) {
rebootState, err := apistate.Reboot()
if err != nil {
return nil, errors.Trace(err)
}
tag, ok := acfg.Tag().(names.MachineTag)
if !ok {
return nil, errors.Errorf("Expected names.MachineTag, got: %T --> %v", acfg.Tag(), acfg.Tag())
}
return &Reboot{
acfg: acfg,
st: rebootState,
tag: tag,
apistate: apistate,
}, nil
}