forked from TrueCloudLab/frostfs-sdk-go
127 lines
1.8 KiB
Go
127 lines
1.8 KiB
Go
package refstest
|
|
|
|
import (
|
|
crand "crypto/rand"
|
|
"crypto/sha256"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs"
|
|
)
|
|
|
|
func GenerateVersion(empty bool) *refs.Version {
|
|
m := new(refs.Version)
|
|
|
|
if !empty {
|
|
m.SetMajor(2)
|
|
m.SetMinor(1)
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateOwnerID(empty bool) *refs.OwnerID {
|
|
m := new(refs.OwnerID)
|
|
|
|
if !empty {
|
|
id := make([]byte, 25)
|
|
_, _ = crand.Read(id)
|
|
|
|
m.SetValue(id)
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateAddress(empty bool) *refs.Address {
|
|
m := new(refs.Address)
|
|
|
|
if !empty {
|
|
m.SetObjectID(GenerateObjectID(false))
|
|
m.SetContainerID(GenerateContainerID(false))
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateObjectID(empty bool) *refs.ObjectID {
|
|
m := new(refs.ObjectID)
|
|
|
|
if !empty {
|
|
id := make([]byte, sha256.Size)
|
|
_, _ = crand.Read(id)
|
|
|
|
m.SetValue(id)
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateObjectIDs(empty bool) []refs.ObjectID {
|
|
var ids []refs.ObjectID
|
|
|
|
if !empty {
|
|
ids = append(ids,
|
|
*GenerateObjectID(false),
|
|
*GenerateObjectID(false),
|
|
)
|
|
}
|
|
|
|
return ids
|
|
}
|
|
|
|
func GenerateContainerID(empty bool) *refs.ContainerID {
|
|
m := new(refs.ContainerID)
|
|
|
|
if !empty {
|
|
id := make([]byte, sha256.Size)
|
|
_, _ = crand.Read(id)
|
|
|
|
m.SetValue(id)
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateContainerIDs(empty bool) []refs.ContainerID {
|
|
var res []refs.ContainerID
|
|
|
|
if !empty {
|
|
res = append(res,
|
|
*GenerateContainerID(false),
|
|
*GenerateContainerID(false),
|
|
)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func GenerateSignature(empty bool) *refs.Signature {
|
|
m := new(refs.Signature)
|
|
|
|
if !empty {
|
|
key := make([]byte, 33)
|
|
_, _ = crand.Read(key)
|
|
|
|
sign := make([]byte, 65)
|
|
_, _ = crand.Read(sign)
|
|
|
|
m.SetScheme(refs.ECDSA_SHA512)
|
|
m.SetKey(key)
|
|
m.SetSign(sign)
|
|
}
|
|
|
|
return m
|
|
}
|
|
|
|
func GenerateChecksum(empty bool) *refs.Checksum {
|
|
m := new(refs.Checksum)
|
|
|
|
if !empty {
|
|
cs := make([]byte, sha256.Size)
|
|
_, _ = crand.Read(cs)
|
|
|
|
m.SetType(refs.SHA256)
|
|
m.SetSum(cs)
|
|
}
|
|
|
|
return m
|
|
}
|