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

     1  package reputationtest
     2  
     3  import (
     4  	refstest "github.com/TrueCloudLab/frostfs-api-go/v2/refs/test"
     5  	"github.com/TrueCloudLab/frostfs-api-go/v2/reputation"
     6  	sessiontest "github.com/TrueCloudLab/frostfs-api-go/v2/session/test"
     7  )
     8  
     9  func GeneratePeerID(empty bool) *reputation.PeerID {
    10  	m := new(reputation.PeerID)
    11  
    12  	if !empty {
    13  		m.SetPublicKey([]byte{1, 2, 3})
    14  	}
    15  
    16  	return m
    17  }
    18  
    19  func GenerateTrust(empty bool) *reputation.Trust {
    20  	m := new(reputation.Trust)
    21  
    22  	if !empty {
    23  		m.SetValue(1)
    24  		m.SetPeer(GeneratePeerID(false))
    25  	}
    26  
    27  	return m
    28  }
    29  
    30  func GeneratePeerToPeerTrust(empty bool) *reputation.PeerToPeerTrust {
    31  	m := new(reputation.PeerToPeerTrust)
    32  
    33  	if !empty {
    34  		m.SetTrustingPeer(GeneratePeerID(false))
    35  		m.SetTrust(GenerateTrust(false))
    36  	}
    37  
    38  	return m
    39  }
    40  
    41  func GenerateGlobalTrustBody(empty bool) *reputation.GlobalTrustBody {
    42  	m := new(reputation.GlobalTrustBody)
    43  
    44  	if !empty {
    45  		m.SetManager(GeneratePeerID(false))
    46  		m.SetTrust(GenerateTrust(false))
    47  	}
    48  
    49  	return m
    50  }
    51  
    52  func GenerateGlobalTrust(empty bool) *reputation.GlobalTrust {
    53  	m := new(reputation.GlobalTrust)
    54  
    55  	if !empty {
    56  		m.SetVersion(refstest.GenerateVersion(false))
    57  		m.SetBody(GenerateGlobalTrustBody(false))
    58  		m.SetSignature(refstest.GenerateSignature(empty))
    59  	}
    60  
    61  	return m
    62  }
    63  
    64  func GenerateTrusts(empty bool) []reputation.Trust {
    65  	var res []reputation.Trust
    66  
    67  	if !empty {
    68  		res = append(res,
    69  			*GenerateTrust(false),
    70  			*GenerateTrust(false),
    71  		)
    72  	}
    73  
    74  	return res
    75  }
    76  
    77  func GenerateAnnounceLocalTrustRequestBody(empty bool) *reputation.AnnounceLocalTrustRequestBody {
    78  	m := new(reputation.AnnounceLocalTrustRequestBody)
    79  
    80  	if !empty {
    81  		m.SetEpoch(13)
    82  		m.SetTrusts(GenerateTrusts(false))
    83  	}
    84  
    85  	return m
    86  }
    87  
    88  func GenerateAnnounceLocalTrustRequest(empty bool) *reputation.AnnounceLocalTrustRequest {
    89  	m := new(reputation.AnnounceLocalTrustRequest)
    90  
    91  	if !empty {
    92  		m.SetBody(GenerateAnnounceLocalTrustRequestBody(false))
    93  		m.SetMetaHeader(sessiontest.GenerateRequestMetaHeader(empty))
    94  		m.SetVerificationHeader(sessiontest.GenerateRequestVerificationHeader(empty))
    95  	}
    96  
    97  	return m
    98  }
    99  
   100  func GenerateAnnounceLocalTrustResponseBody(empty bool) *reputation.AnnounceLocalTrustResponseBody {
   101  	m := new(reputation.AnnounceLocalTrustResponseBody)
   102  
   103  	return m
   104  }
   105  
   106  func GenerateAnnounceLocalTrustResponse(empty bool) *reputation.AnnounceLocalTrustResponse {
   107  	m := new(reputation.AnnounceLocalTrustResponse)
   108  
   109  	if !empty {
   110  		m.SetBody(GenerateAnnounceLocalTrustResponseBody(false))
   111  		m.SetMetaHeader(sessiontest.GenerateResponseMetaHeader(empty))
   112  		m.SetVerificationHeader(sessiontest.GenerateResponseVerificationHeader(empty))
   113  	}
   114  
   115  	return m
   116  }
   117  
   118  func GenerateAnnounceIntermediateResultRequestBody(empty bool) *reputation.AnnounceIntermediateResultRequestBody {
   119  	m := new(reputation.AnnounceIntermediateResultRequestBody)
   120  
   121  	if !empty {
   122  		m.SetEpoch(123)
   123  		m.SetIteration(564)
   124  		m.SetTrust(GeneratePeerToPeerTrust(false))
   125  	}
   126  
   127  	return m
   128  }
   129  
   130  func GenerateAnnounceIntermediateResultRequest(empty bool) *reputation.AnnounceIntermediateResultRequest {
   131  	m := new(reputation.AnnounceIntermediateResultRequest)
   132  
   133  	if !empty {
   134  		m.SetBody(GenerateAnnounceIntermediateResultRequestBody(false))
   135  	}
   136  
   137  	m.SetMetaHeader(sessiontest.GenerateRequestMetaHeader(empty))
   138  	m.SetVerificationHeader(sessiontest.GenerateRequestVerificationHeader(empty))
   139  
   140  	return m
   141  }
   142  
   143  func GenerateAnnounceIntermediateResultResponseBody(empty bool) *reputation.AnnounceIntermediateResultResponseBody {
   144  	m := new(reputation.AnnounceIntermediateResultResponseBody)
   145  
   146  	return m
   147  }
   148  
   149  func GenerateAnnounceIntermediateResultResponse(empty bool) *reputation.AnnounceIntermediateResultResponse {
   150  	m := new(reputation.AnnounceIntermediateResultResponse)
   151  
   152  	if !empty {
   153  		m.SetBody(GenerateAnnounceIntermediateResultResponseBody(false))
   154  	}
   155  
   156  	m.SetMetaHeader(sessiontest.GenerateResponseMetaHeader(empty))
   157  	m.SetVerificationHeader(sessiontest.GenerateResponseVerificationHeader(empty))
   158  
   159  	return m
   160  }