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

     1  package accountingtest
     2  
     3  import (
     4  	"github.com/TrueCloudLab/frostfs-api-go/v2/accounting"
     5  	accountingtest "github.com/TrueCloudLab/frostfs-api-go/v2/refs/test"
     6  	sessiontest "github.com/TrueCloudLab/frostfs-api-go/v2/session/test"
     7  )
     8  
     9  func GenerateBalanceRequest(empty bool) *accounting.BalanceRequest {
    10  	m := new(accounting.BalanceRequest)
    11  
    12  	if !empty {
    13  		m.SetBody(GenerateBalanceRequestBody(false))
    14  	}
    15  
    16  	m.SetMetaHeader(sessiontest.GenerateRequestMetaHeader(empty))
    17  	m.SetVerificationHeader(sessiontest.GenerateRequestVerificationHeader(empty))
    18  
    19  	return m
    20  }
    21  
    22  func GenerateBalanceRequestBody(empty bool) *accounting.BalanceRequestBody {
    23  	m := new(accounting.BalanceRequestBody)
    24  
    25  	if !empty {
    26  		m.SetOwnerID(accountingtest.GenerateOwnerID(false))
    27  	}
    28  
    29  	return m
    30  }
    31  
    32  func GenerateBalanceResponse(empty bool) *accounting.BalanceResponse {
    33  	m := new(accounting.BalanceResponse)
    34  
    35  	if !empty {
    36  		m.SetBody(GenerateBalanceResponseBody(false))
    37  	}
    38  
    39  	m.SetMetaHeader(sessiontest.GenerateResponseMetaHeader(empty))
    40  	m.SetVerificationHeader(sessiontest.GenerateResponseVerificationHeader(empty))
    41  
    42  	return m
    43  }
    44  
    45  func GenerateBalanceResponseBody(empty bool) *accounting.BalanceResponseBody {
    46  	m := new(accounting.BalanceResponseBody)
    47  
    48  	if !empty {
    49  		m.SetBalance(GenerateDecimal(false))
    50  	}
    51  
    52  	return m
    53  }
    54  
    55  func GenerateDecimal(empty bool) *accounting.Decimal {
    56  	m := new(accounting.Decimal)
    57  
    58  	if !empty {
    59  		m.SetValue(1)
    60  		m.SetPrecision(2)
    61  	}
    62  
    63  	return m
    64  }