github.com/TrueCloudLab/frostfs-api-go/v2@v2.0.0-20230228134343-196241c4e79a/refs/test/generate.go (about)

     1  package refstest
     2  
     3  import (
     4  	"math/rand"
     5  
     6  	"github.com/TrueCloudLab/frostfs-api-go/v2/refs"
     7  )
     8  
     9  func GenerateVersion(empty bool) *refs.Version {
    10  	m := new(refs.Version)
    11  
    12  	if !empty {
    13  		m.SetMajor(2)
    14  		m.SetMinor(1)
    15  	}
    16  
    17  	return m
    18  }
    19  
    20  func GenerateOwnerID(empty bool) *refs.OwnerID {
    21  	m := new(refs.OwnerID)
    22  
    23  	if !empty {
    24  		m.SetValue([]byte{1, 2, 3})
    25  	}
    26  
    27  	return m
    28  }
    29  
    30  func GenerateAddress(empty bool) *refs.Address {
    31  	m := new(refs.Address)
    32  
    33  	if !empty {
    34  		m.SetObjectID(GenerateObjectID(false))
    35  		m.SetContainerID(GenerateContainerID(false))
    36  	}
    37  
    38  	return m
    39  }
    40  
    41  func GenerateObjectID(empty bool) *refs.ObjectID {
    42  	m := new(refs.ObjectID)
    43  
    44  	if !empty {
    45  		m.SetValue([]byte{1, 2, 3})
    46  	}
    47  
    48  	return m
    49  }
    50  
    51  func GenerateObjectIDs(empty bool) []refs.ObjectID {
    52  	var ids []refs.ObjectID
    53  
    54  	if !empty {
    55  		ids = append(ids,
    56  			*GenerateObjectID(false),
    57  			*GenerateObjectID(false),
    58  		)
    59  	}
    60  
    61  	return ids
    62  }
    63  
    64  func GenerateContainerID(empty bool) *refs.ContainerID {
    65  	m := new(refs.ContainerID)
    66  
    67  	if !empty {
    68  		m.SetValue([]byte{1, 2, 3})
    69  	}
    70  
    71  	return m
    72  }
    73  
    74  func GenerateContainerIDs(empty bool) []refs.ContainerID {
    75  	var res []refs.ContainerID
    76  
    77  	if !empty {
    78  		res = append(res,
    79  			*GenerateContainerID(false),
    80  			*GenerateContainerID(false),
    81  		)
    82  	}
    83  
    84  	return res
    85  }
    86  
    87  func GenerateSignature(empty bool) *refs.Signature {
    88  	m := new(refs.Signature)
    89  
    90  	if !empty {
    91  		m.SetKey([]byte{1})
    92  		m.SetSign([]byte{2})
    93  		m.SetScheme(refs.SignatureScheme(rand.Int31() % 3))
    94  	}
    95  
    96  	return m
    97  }
    98  
    99  func GenerateChecksum(empty bool) *refs.Checksum {
   100  	m := new(refs.Checksum)
   101  
   102  	if !empty {
   103  		m.SetType(1)
   104  		m.SetSum([]byte{1, 2, 3})
   105  	}
   106  
   107  	return m
   108  }
   109  
   110  func GenerateSubnetID(empty bool) *refs.SubnetID {
   111  	m := new(refs.SubnetID)
   112  
   113  	if !empty {
   114  		m.SetValue(666)
   115  	}
   116  
   117  	return m
   118  }