[#276] Merge repo with frostfs-api-go

Signed-off-by: Pavel Pogodaev <p.pogodaev@yadro.com>

api/session/test/generate.go

@@ -0,0 +1,251 @@
+package sessiontest
+
+import (
+	crand "crypto/rand"
+	"math/rand"
+	"time"
+
+	acltest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/acl/test"
+	refstest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs/test"
+	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/session"
+	statustest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/status/test"
+)
+
+func GenerateCreateRequestBody(empty bool) *session.CreateRequestBody {
+	m := new(session.CreateRequestBody)
+
+	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.SetVerificationHeader(GenerateRequestVerificationHeader(empty))
+
+	return m
+}
+
+func GenerateCreateResponseBody(empty bool) *session.CreateResponseBody {
+	m := new(session.CreateResponseBody)
+
+	if !empty {
+		id := make([]byte, 16)
+		_, _ = crand.Read(id)
+
+		m.SetID(id)
+		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.SetVerificationHeader(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.Token {
+	m := new(session.Token)
+
+	if !empty {
+		m.SetBody(GenerateSessionTokenBody(false))
+	}
+
+	m.SetSignature(refstest.GenerateSignature(empty))
+
+	return m
+}
+
+func GenerateSessionTokenBody(empty bool) *session.TokenBody {
+	m := new(session.TokenBody)
+
+	if !empty {
+		id := make([]byte, 16)
+		_, _ = crand.Read(id)
+
+		m.SetID(id)
+		m.SetSessionKey([]byte{2})
+		m.SetOwnerID(refstest.GenerateOwnerID(false))
+		m.SetLifetime(GenerateTokenLifetime(false))
+
+		switch randomInt(2) {
+		case 0:
+			m.SetContext(GenerateObjectSessionContext(false))
+		case 1:
+			m.SetContext(GenerateContainerSessionContext(false))
+		}
+	}
+
+	return m
+}
+
+func GenerateTokenLifetime(empty bool) *session.TokenLifetime {
+	m := new(session.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.ObjectVerbHead)
+		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.ContainerVerbDelete)
+		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
+}
+
+func randomInt(n int) int {
+	return rand.New(rand.NewSource(time.Now().UnixNano())).Intn(n)
+}