github.com/chipaca/snappy@v0.0.0-20210104084008-1f06296fe8ad/daemon/api_debug_seeding_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 daemon

import (
	"net/http"
	"time"

	. "gopkg.in/check.v1"

	"github.com/snapcore/snapd/overlord/state"
)

var _ = Suite(&seedingDebugSuite{})

type seedingDebugSuite struct {
	APIBaseSuite
}

func (s *seedingDebugSuite) SetUpTest(c *C) {
	s.APIBaseSuite.SetUpTest(c)
	s.daemonWithOverlordMock(c)
}

func (s *seedingDebugSuite) getSeedingDebug(c *C) interface{} {
	req, err := http.NewRequest("GET", "/v2/debug?aspect=seeding", nil)
	c.Assert(err, IsNil)

	rsp := getDebug(debugCmd, req, nil).(*resp)
	c.Assert(rsp.Type, Equals, ResponseTypeSync)
	return rsp.Result
}

func (s *seedingDebugSuite) TestNoData(c *C) {
	data := s.getSeedingDebug(c)
	c.Check(data, NotNil)
	c.Check(data, DeepEquals, &seedingInfo{})
}

func (s *seedingDebugSuite) TestSeedingDebug(c *C) {
	seeded := true
	preseeded := true
	key1 := "foo"
	key2 := "bar"

	preseedStartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:00Z")
	c.Assert(err, IsNil)
	preseedTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:01Z")
	c.Assert(err, IsNil)
	seedRestartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:03Z")
	c.Assert(err, IsNil)
	seedTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:07Z")
	c.Assert(err, IsNil)

	st := s.d.overlord.State()
	st.Lock()

	st.Set("preseeded", preseeded)
	st.Set("seeded", seeded)

	st.Set("preseed-system-key", key1)
	st.Set("seed-restart-system-key", key2)

	st.Set("preseed-start-time", preseedStartTime)
	st.Set("seed-restart-time", seedRestartTime)

	st.Set("preseed-time", preseedTime)
	st.Set("seed-time", seedTime)

	st.Unlock()

	data := s.getSeedingDebug(c)
	c.Check(data, DeepEquals, &seedingInfo{
		Seeded:               true,
		Preseeded:            true,
		PreseedSystemKey:     "foo",
		SeedRestartSystemKey: "bar",
		PreseedStartTime:     &preseedStartTime,
		PreseedTime:          &preseedTime,
		SeedRestartTime:      &seedRestartTime,
		SeedTime:             &seedTime,
	})
}

func (s *seedingDebugSuite) TestSeedingDebugSeededNoTimes(c *C) {
	seedTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:07Z")
	c.Assert(err, IsNil)

	st := s.d.overlord.State()
	st.Lock()

	// only set seed-time and seeded
	st.Set("seed-time", seedTime)
	st.Set("seeded", true)

	st.Unlock()

	data := s.getSeedingDebug(c)
	c.Check(data, DeepEquals, &seedingInfo{
		Seeded:   true,
		SeedTime: &seedTime,
	})
}

func (s *seedingDebugSuite) TestSeedingDebugPreseededStillSeeding(c *C) {
	preseedStartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:00Z")
	c.Assert(err, IsNil)
	preseedTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:01Z")
	c.Assert(err, IsNil)
	seedRestartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:03Z")
	c.Assert(err, IsNil)

	st := s.d.overlord.State()
	st.Lock()

	st.Set("preseeded", true)
	st.Set("seeded", false)

	st.Set("preseed-system-key", "foo")
	st.Set("seed-restart-system-key", "bar")

	st.Set("preseed-start-time", preseedStartTime)
	st.Set("seed-restart-time", seedRestartTime)

	st.Set("preseed-time", preseedTime)

	st.Unlock()

	data := s.getSeedingDebug(c)
	c.Check(data, DeepEquals, &seedingInfo{
		Seeded:               false,
		Preseeded:            true,
		PreseedSystemKey:     "foo",
		SeedRestartSystemKey: "bar",
		PreseedStartTime:     &preseedStartTime,
		PreseedTime:          &preseedTime,
		SeedRestartTime:      &seedRestartTime,
	})
}

func (s *seedingDebugSuite) TestSeedingDebugPreseededSeedError(c *C) {
	preseedStartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:00Z")
	c.Assert(err, IsNil)
	preseedTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:01Z")
	c.Assert(err, IsNil)
	seedRestartTime, err := time.Parse(time.RFC3339, "2020-01-01T10:00:03Z")
	c.Assert(err, IsNil)

	st := s.d.overlord.State()
	st.Lock()

	st.Set("preseeded", true)
	st.Set("seeded", false)

	st.Set("preseed-system-key", "foo")
	st.Set("seed-restart-system-key", "bar")

	st.Set("preseed-start-time", preseedStartTime)
	st.Set("seed-restart-time", seedRestartTime)

	st.Set("preseed-time", preseedTime)

	chg1 := st.NewChange("seed", "tentative 1")
	t11 := st.NewTask("seed task", "t11")
	t12 := st.NewTask("seed task", "t12")
	chg1.AddTask(t11)
	chg1.AddTask(t12)
	t11.SetStatus(state.UndoneStatus)
	t11.Errorf("t11: undone")
	t12.SetStatus(state.ErrorStatus)
	t12.Errorf("t12: fail")

	// ensure different spawn time
	time.Sleep(50 * time.Millisecond)
	chg2 := st.NewChange("seed", "tentative 2")
	t21 := st.NewTask("seed task", "t21")
	chg2.AddTask(t21)
	t21.SetStatus(state.ErrorStatus)
	t21.Errorf("t21: error")

	chg3 := st.NewChange("seed", "tentative 3")
	t31 := st.NewTask("seed task", "t31")
	chg3.AddTask(t31)
	t31.SetStatus(state.DoingStatus)

	st.Unlock()

	data := s.getSeedingDebug(c)
	c.Check(data, DeepEquals, &seedingInfo{
		Seeded:               false,
		Preseeded:            true,
		PreseedSystemKey:     "foo",
		SeedRestartSystemKey: "bar",
		PreseedStartTime:     &preseedStartTime,
		PreseedTime:          &preseedTime,
		SeedRestartTime:      &seedRestartTime,
		SeedError: `cannot perform the following tasks:
- t12 (t12: fail)`,
	})
}