Function TestSlotState

pkg/topom/context_test.go:13–119

(x *testing.T)

)

func TestSlotState(x *testing.T) {
	t := openTopom()
	defer t.Close()

	const sid = 1
	const gid1 = 1
	const gid2 = 2
	const server1 = "server1:port"
	const server2 = "server2:port"

	check := func() {
		ctx, err := t.newContext()
		assert.MustNoError(err)
		m, err := ctx.getSlotMapping(sid)
		assert.MustNoError(err)
		assert.Must(m.Id == sid)
		assert.Must(m.GroupId == gid1 && m.Action.TargetId == gid2)

		g1, err := ctx.getGroup(gid1)
		assert.MustNoError(err)
		assert.Must(ctx.getGroupMaster(gid1) == server1)

		g2, err := ctx.getGroup(gid2)
		assert.MustNoError(err)
		assert.Must(ctx.getGroupMaster(gid2) == server2)

		slot := ctx.toSlot(m, nil)

		switch m.Action.State {
		case models.ActionPrepared:
			assert.Must(slot.Locked == true)
		case models.ActionMigrating:
			switch {
			case g1.Promoting.State == models.ActionPrepared:
				assert.Must(slot.Locked == true)
			case g2.Promoting.State == models.ActionPrepared:
				assert.Must(slot.Locked == true)
			default:
				assert.Must(slot.Locked == false)
				assert.Must(slot.BackendAddr == server2)
				assert.Must(slot.MigrateFrom == server1)
			}
		case models.ActionFinished:
			switch {
			case g2.Promoting.State == models.ActionPrepared:
				assert.Must(slot.Locked == true)
			default:
				assert.Must(slot.Locked == false)
				assert.Must(slot.BackendAddr == server2)
				assert.Must(slot.MigrateFrom == "")
			}
		default:
			switch {
			case g1.Promoting.State == models.ActionPrepared:
				assert.Must(slot.Locked == true)
			default:
				assert.Must(slot.Locked == false)
				assert.Must(slot.BackendAddr == server1)