github.com/Lephar/snapd@v0.0.0-20210825215435-c7fba9cef4d2/cmd/snap/cmd_reboot_test.go

// -*- Mode: Go; indent-tabs-mode: t -*-

/*
 * Copyright (C) 2020 Canonical Ltd
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 3 as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 */

package main_test

import (
	"fmt"
	"io/ioutil"
	"net/http"
	"strings"

	. "gopkg.in/check.v1"

	snap "github.com/snapcore/snapd/cmd/snap"
)

func (s *SnapSuite) TestRebootHelp(c *C) {
	msg := `Usage:
  snap.test reboot [reboot-OPTIONS] [<label>]

The reboot command reboots the system into a particular mode of the selected
recovery system.

When called without a system label and without a mode it will just
trigger a regular reboot.

When called without a system label but with a mode it will use the
current system to enter the given mode.

Note that "recover" and "run" modes are only available for the
current system.

[reboot command options]
      --run        Boot into run mode
      --install    Boot into install mode
      --recover    Boot into recover mode

[reboot command arguments]
  <label>:         The recovery system label
`
	s.testSubCommandHelp(c, "reboot", msg)
}

func (s *SnapSuite) TestRebootHappy(c *C) {

	for _, tc := range []struct {
		cmdline          []string
		expectedEndpoint string
		expectedJSON     string
		expectedMsg      string
	}{
		{
			cmdline:          []string{"reboot"},
			expectedEndpoint: "/v2/systems",
			expectedJSON:     `{"action":"reboot","mode":""}`,
			expectedMsg:      `Reboot`,
		},
		{
			cmdline:          []string{"reboot", "--recover"},
			expectedEndpoint: "/v2/systems",
			expectedJSON:     `{"action":"reboot","mode":"recover"}`,
			expectedMsg:      `Reboot into "recover" mode.`,
		},
		{
			cmdline:          []string{"reboot", "20200101"},
			expectedEndpoint: "/v2/systems/20200101",
			expectedJSON:     `{"action":"reboot","mode":""}`,
			expectedMsg:      `Reboot into "20200101".`,
		},
		{
			cmdline:          []string{"reboot", "--recover", "20200101"},
			expectedEndpoint: "/v2/systems/20200101",
			expectedJSON:     `{"action":"reboot","mode":"recover"}`,
			expectedMsg:      `Reboot into "20200101" "recover" mode.`,
		},
	} {

		n := 0
		s.ResetStdStreams()

		s.RedirectClientToTestServer(func(w http.ResponseWriter, r *http.Request) {
			switch n {
			case 0:
				c.Check(r.Method, Equals, "POST")
				c.Check(r.URL.Path, Equals, tc.expectedEndpoint, Commentf("%v", tc.cmdline))
				c.Check(r.URL.RawQuery, Equals, "")
				body, err := ioutil.ReadAll(r.Body)
				c.Check(err, IsNil)
				c.Check(string(body), Equals, tc.expectedJSON+"\n")
				fmt.Fprintln(w, `{"type": "sync", "result": {}}`)
			default:
				c.Fatalf("expected to get 1 requests, now on %d", n+1)
			}

			n++
		})

		// The server side will work out if the request is valid
		rest, err := snap.Parser(snap.Client()).ParseArgs(tc.cmdline)
		c.Assert(err, IsNil)
		c.Assert(rest, DeepEquals, []string{})
		c.Check(s.Stdout(), Equals, tc.expectedMsg+"\n", Commentf("%v", tc.cmdline))
		c.Check(s.Stderr(), Equals, "")
	}
}

func (s *SnapSuite) TestRebootUnhappy(c *C) {
	s.RedirectClientToTestServer(func(w http.ResponseWriter, r *http.Request) {
		c.Fatalf("server should not be hit in this test")
	})

	var tc = []struct {
		args   []string
		errStr string
	}{
		{
			args:   []string{"reboot", "--run", "--recover", "20200101"},
			errStr: "Please specify a single mode",
		},
		{
			args:   []string{"reboot", "--unknown-mode", "20200101"},
			errStr: "unknown flag `unknown-mode'",
		},
	}

	for _, t := range tc {
		_, err := snap.Parser(snap.Client()).ParseArgs(t.args)
		c.Check(err, ErrorMatches, t.errStr, Commentf(strings.Join(t.args, " ")))
	}
}

func (s *SnapSuite) TestRebootAPIFail(c *C) {
	n := 0
	s.RedirectClientToTestServer(func(w http.ResponseWriter, r *http.Request) {
		switch n {
		case 0:
			c.Check(r.Method, Equals, "POST")
			c.Check(r.URL.Path, Equals, "/v2/systems/20200101")
			c.Check(r.URL.RawQuery, Equals, "")
			w.WriteHeader(404)
			fmt.Fprintln(w, `{"type": "error", "status-code":404, "result": {"message":"requested system does not exist"}}`)
		default:
			c.Fatalf("expected to get 1 requests, now on %d", n+1)
		}

		n++
	})
	_, err := snap.Parser(snap.Client()).ParseArgs([]string{"reboot", "--recover", "20200101"})
	c.Assert(err, ErrorMatches, `cannot request system reboot into "20200101": requested system does not exist`)
	c.Check(s.Stdout(), Equals, "")
	c.Check(s.Stderr(), Equals, "")
}