github.com/juju/juju@v0.0.0-20240327075706-a90865de2538/worker/lease/manager_cross_test.go

// Copyright 2018 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package lease_test

import (
	"time"

	"github.com/juju/clock/testclock"
	"github.com/juju/errors"
	"github.com/juju/loggo"
	"github.com/juju/testing"
	jc "github.com/juju/testing/checkers"
	gc "gopkg.in/check.v1"

	corelease "github.com/juju/juju/core/lease"
	"github.com/juju/juju/worker/lease"
)

// Tests that check the manager handles leases across namespaces and
// models correctly.

type CrossSuite struct {
	testing.IsolationSuite
}

var _ = gc.Suite(&CrossSuite{})

func (s *CrossSuite) testClaims(c *gc.C, lease1, lease2 corelease.Key) {
	fix := Fixture{
		expectCalls: []call{{
			method: "ClaimLease",
			args: []interface{}{
				lease1,
				corelease.Request{Holder: "sgt-howie", Duration: time.Minute},
			},
			callback: func(leases map[corelease.Key]corelease.Info) {
				leases[lease1] = corelease.Info{
					Holder: "sgt-howie",
					Expiry: offset(time.Second),
				}
			},
		}, {
			method: "ClaimLease",
			args: []interface{}{
				lease2,
				corelease.Request{Holder: "rowan", Duration: time.Minute},
			},
			callback: func(leases map[corelease.Key]corelease.Info) {
				leases[lease2] = corelease.Info{
					Holder: "rowan",
					Expiry: offset(time.Second),
				}
			},
		}},
	}
	fix.RunTest(c, func(manager *lease.Manager, _ *testclock.Clock) {
		claimer1, err := manager.Claimer(lease1.Namespace, lease1.ModelUUID)
		c.Assert(err, jc.ErrorIsNil)
		claimer2, err := manager.Claimer(lease2.Namespace, lease2.ModelUUID)
		c.Assert(err, jc.ErrorIsNil)

		err = claimer1.Claim(lease1.Lease, "sgt-howie", time.Minute)
		c.Assert(err, jc.ErrorIsNil)
		err = claimer2.Claim(lease2.Lease, "rowan", time.Minute)
		c.Assert(err, jc.ErrorIsNil)

		err = claimer1.Claim(lease1.Lease, "lord-summerisle", time.Minute)
		c.Assert(err, gc.Equals, corelease.ErrClaimDenied)
	})
}

func (s *CrossSuite) TestClaimAcrossNamespaces(c *gc.C) {
	s.testClaims(c,
		key("ns1", "model", "summerisle"),
		key("ns2", "model", "summerisle"),
	)
}

func (s *CrossSuite) TestClaimAcrossModels(c *gc.C) {
	s.testClaims(c,
		key("ns", "model1", "summerisle"),
		key("ns", "model2", "summerisle"),
	)
}

func (s *CrossSuite) testWaits(c *gc.C, lease1, lease2 corelease.Key) {
	fix := Fixture{
		leases: map[corelease.Key]corelease.Info{
			lease1: {
				Holder: "sgt-howie",
				Expiry: offset(time.Second),
			},
			lease2: {
				Holder: "willow",
				Expiry: offset(time.Minute),
			},
		},
	}
	fix.RunTest(c, func(manager *lease.Manager, clock *testclock.Clock) {
		b1 := newBlockTest(c, manager, lease1)
		b2 := newBlockTest(c, manager, lease2)

		b1.assertBlocked(c)
		b2.assertBlocked(c)

		clock.Advance(2 * time.Second)

		err := b1.assertUnblocked(c)
		c.Assert(err, jc.ErrorIsNil)
		b2.assertBlocked(c)
	})
}

func (s *CrossSuite) TestWaitAcrossNamespaces(c *gc.C) {
	s.testWaits(c,
		key("ns1", "model", "summerisle"),
		key("ns2", "model", "summerisle"),
	)
}

func (s *CrossSuite) TestWaitAcrossModels(c *gc.C) {
	s.testWaits(c,
		key("ns", "model1", "summerisle"),
		key("ns", "model2", "summerisle"),
	)
}

func (s *CrossSuite) testChecks(c *gc.C, lease1, lease2 corelease.Key) {
	fix := Fixture{
		leases: map[corelease.Key]corelease.Info{
			lease1: {
				Holder: "sgt-howie",
				Expiry: offset(time.Second),
			},
		},
	}
	fix.RunTest(c, func(manager *lease.Manager, _ *testclock.Clock) {
		checker1, err := manager.Checker(lease1.Namespace, lease1.ModelUUID)
		c.Assert(err, jc.ErrorIsNil)
		checker2, err := manager.Checker(lease2.Namespace, lease2.ModelUUID)
		c.Assert(err, jc.ErrorIsNil)

		t1 := checker1.Token(lease1.Lease, "sgt-howie")
		c.Assert(t1.Check(), gc.Equals, nil)

		t2 := checker2.Token(lease2.Lease, "sgt-howie")
		err = t2.Check()
		c.Assert(errors.Cause(err), gc.Equals, corelease.ErrNotHeld)
	})
}

func (s *CrossSuite) TestCheckAcrossNamespaces(c *gc.C) {
	s.testChecks(c,
		key("ns1", "model", "summerisle"),
		key("ns2", "model", "summerisle"),
	)
}

func (s *CrossSuite) TestCheckAcrossModels(c *gc.C) {
	s.testChecks(c,
		key("ns", "model1", "summerisle"),
		key("ns", "model2", "summerisle"),
	)
}

func (s *CrossSuite) TestDifferentNamespaceValidation(c *gc.C) {
	clock := testclock.NewClock(defaultClockStart)
	store := NewStore(nil, nil, clock)
	manager, err := lease.NewManager(lease.ManagerConfig{
		Clock: clock,
		Store: store,
		Secretary: func(namespace string) (lease.Secretary, error) {
			switch namespace {
			case "ns1":
				return Secretary{}, nil
			case "ns2":
				return OtherSecretary{}, nil
			default:
				return nil, errors.Errorf("bad namespace!")
			}
		},
		MaxSleep:             defaultMaxSleep,
		Logger:               loggo.GetLogger("lease_test"),
		PrometheusRegisterer: noopRegisterer{},
	})
	c.Assert(err, jc.ErrorIsNil)
	defer func() {
		manager.Kill()
		err := manager.Wait()
		c.Check(err, jc.ErrorIsNil)
	}()
	defer store.Wait(c)

	_, err = manager.Claimer("something-else", "model")
	c.Assert(err, gc.ErrorMatches, "bad namespace!")

	c1, err := manager.Claimer("ns1", "model")
	c.Assert(err, jc.ErrorIsNil)
	err = c1.Claim("INVALID", "sgt-howie", time.Minute)
	c.Assert(err, gc.ErrorMatches, `cannot claim lease "INVALID": name not valid`)

	c2, err := manager.Claimer("ns2", "model")
	c.Assert(err, jc.ErrorIsNil)
	err = c2.Claim("INVALID", "sgt-howie", time.Minute)
	c.Assert(err, gc.ErrorMatches, `cannot claim lease "INVALID": lease name not valid`)
}

type OtherSecretary struct{}

// CheckLease is part of the lease.Secretary interface.
func (OtherSecretary) CheckLease(_ corelease.Key) error {
	return errors.NotValidf("lease name")
}

// CheckHolder is part of the lease.Secretary interface.
func (OtherSecretary) CheckHolder(_ string) error {
	return errors.NotValidf("holder name")
}

// CheckDuration is part of the lease.Secretary interface.
func (OtherSecretary) CheckDuration(duration time.Duration) error {
	if duration != time.Hour {
		return errors.NotValidf("time")
	}
	return nil
}