frostfs-api-go/session/test/generate.go

package sessiontest

import (
	acltest "git.frostfs.info/TrueCloudLab/aarifullin/v2/acl/test"
	"git.frostfs.info/TrueCloudLab/aarifullin/v2/internal/random"
	refstest "git.frostfs.info/TrueCloudLab/aarifullin/v2/refs/test"
	session "git.frostfs.info/TrueCloudLab/aarifullin/v2/session/grpc"
	statustest "git.frostfs.info/TrueCloudLab/aarifullin/v2/status/test"
)

func GenerateCreateRequestBody(empty bool) *session.CreateRequest_Body {
	m := new(session.CreateRequest_Body)

	if !empty {
		m.SetExpiration(555)
		m.SetOwnerId(refstest.GenerateOwnerID(false))
	}

	return m
}

func GenerateCreateRequest(empty bool) *session.CreateRequest {
	m := new(session.CreateRequest)

	if !empty {
		m.SetBody(GenerateCreateRequestBody(false))
	}

	m.SetMetaHeader(GenerateRequestMetaHeader(empty))
	m.SetVerifyHeader(GenerateRequestVerificationHeader(empty))

	return m
}

func GenerateCreateResponseBody(empty bool) *session.CreateResponse_Body {
	m := new(session.CreateResponse_Body)

	if !empty {
		m.SetId([]byte{1, 2, 3})
		m.SetSessionKey([]byte{4, 5, 6})
	}

	return m
}

func GenerateCreateResponse(empty bool) *session.CreateResponse {
	m := new(session.CreateResponse)

	if !empty {
		m.SetBody(GenerateCreateResponseBody(false))
	}

	m.SetMetaHeader(GenerateResponseMetaHeader(empty))
	m.SetVerifyHeader(GenerateResponseVerificationHeader(empty))

	return m
}

func GenerateResponseVerificationHeader(empty bool) *session.ResponseVerificationHeader {
	return generateResponseVerificationHeader(empty, true)
}

func generateResponseVerificationHeader(empty, withOrigin bool) *session.ResponseVerificationHeader {
	m := new(session.ResponseVerificationHeader)

	if !empty {
		m.SetBodySignature(refstest.GenerateSignature(false))
	}

	m.SetMetaSignature(refstest.GenerateSignature(empty))
	m.SetOriginSignature(refstest.GenerateSignature(empty))

	if withOrigin {
		m.SetOrigin(generateResponseVerificationHeader(empty, false))
	}

	return m
}

func GenerateResponseMetaHeader(empty bool) *session.ResponseMetaHeader {
	return generateResponseMetaHeader(empty, true)
}

func generateResponseMetaHeader(empty, withOrigin bool) *session.ResponseMetaHeader {
	m := new(session.ResponseMetaHeader)

	if !empty {
		m.SetEpoch(13)
		m.SetTtl(100)
	}

	m.SetXHeaders(GenerateXHeaders(empty))
	m.SetVersion(refstest.GenerateVersion(empty))
	m.SetStatus(statustest.Status(empty))

	if withOrigin {
		m.SetOrigin(generateResponseMetaHeader(empty, false))
	}

	return m
}

func GenerateRequestVerificationHeader(empty bool) *session.RequestVerificationHeader {
	return generateRequestVerificationHeader(empty, true)
}

func generateRequestVerificationHeader(empty, withOrigin bool) *session.RequestVerificationHeader {
	m := new(session.RequestVerificationHeader)

	if !empty {
		m.SetBodySignature(refstest.GenerateSignature(false))
	}

	m.SetMetaSignature(refstest.GenerateSignature(empty))
	m.SetOriginSignature(refstest.GenerateSignature(empty))

	if withOrigin {
		m.SetOrigin(generateRequestVerificationHeader(empty, false))
	}

	return m
}

func GenerateRequestMetaHeader(empty bool) *session.RequestMetaHeader {
	return generateRequestMetaHeader(empty, true)
}

func generateRequestMetaHeader(empty, withOrigin bool) *session.RequestMetaHeader {
	m := new(session.RequestMetaHeader)

	if !empty {
		m.SetEpoch(13)
		m.SetTtl(100)
		m.SetNetworkMagic(1337)
	}

	m.SetXHeaders(GenerateXHeaders(empty))
	m.SetVersion(refstest.GenerateVersion(empty))
	m.SetSessionToken(GenerateSessionToken(empty))
	m.SetBearerToken(acltest.GenerateBearerToken(empty))

	if withOrigin {
		m.SetOrigin(generateRequestMetaHeader(empty, false))
	}

	return m
}

func GenerateSessionToken(empty bool) *session.SessionToken {
	m := new(session.SessionToken)

	if !empty {
		m.SetBody(GenerateSessionTokenBody(false))
	}

	m.SetSignature(refstest.GenerateSignature(empty))

	return m
}

func GenerateSessionTokenBody(empty bool) *session.SessionToken_Body {
	m := new(session.SessionToken_Body)

	if !empty {
		m.SetId([]byte{1})
		m.SetSessionKey([]byte{2})
		m.SetOwnerId(refstest.GenerateOwnerID(false))
		m.SetLifetime(GenerateTokenLifetime(false))

		switch random.Uint32(2) {
		case 0:
			m.SetObjectSessionContext(GenerateObjectSessionContext(false))
		case 1:
			m.SetContainerSessionContext(GenerateContainerSessionContext(false))
		}
	}

	return m
}

func GenerateTokenLifetime(empty bool) *session.SessionToken_Body_TokenLifetime {
	m := new(session.SessionToken_Body_TokenLifetime)

	if !empty {
		m.SetExp(1)
		m.SetIat(2)
		m.SetExp(3)
	}

	return m
}

func GenerateObjectSessionContext(empty bool) *session.ObjectSessionContext {
	m := new(session.ObjectSessionContext)

	if !empty {
		m.SetVerb(session.ObjectSessionContext_HEAD)
		m.SetTarget(refstest.GenerateContainerID(false), refstest.GenerateObjectIDs(false))
	}

	return m
}

func GenerateContainerSessionContext(empty bool) *session.ContainerSessionContext {
	m := new(session.ContainerSessionContext)

	if !empty {
		m.SetVerb(session.ContainerSessionContext_DELETE)
		m.SetWildcard(true)
		m.SetContainerId(refstest.GenerateContainerID(false))
	}

	return m
}

func GenerateXHeader(empty bool) *session.XHeader {
	m := new(session.XHeader)

	if !empty {
		m.SetKey("key")
		m.SetValue("val")
	}

	return m
}

func GenerateXHeaders(empty bool) []*session.XHeader {
	var xs []*session.XHeader

	if !empty {
		xs = append(xs,
			GenerateXHeader(false),
			GenerateXHeader(false),
		)
	}

	return xs
}