-
Notifications
You must be signed in to change notification settings - Fork 66
/
contract_bytecode_query_unit_test.go
170 lines (138 loc) · 7.09 KB
/
contract_bytecode_query_unit_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
//go:build all || unit
// +build all unit
package hiero
// SPDX-License-Identifier: Apache-2.0
import (
"bytes"
"testing"
"time"
"github.com/hiero-ledger/hiero-sdk-go/v2/proto/services"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestUnitContractBytecodeQueryValidate(t *testing.T) {
t.Parallel()
client, err := _NewMockClient()
client.SetLedgerID(*NewLedgerIDTestnet())
require.NoError(t, err)
client.SetAutoValidateChecksums(true)
contractID, err := ContractIDFromString("0.0.123-esxsf")
require.NoError(t, err)
contractBytecode := NewContractBytecodeQuery().
SetContractID(contractID)
err = contractBytecode.validateNetworkOnIDs(client)
require.NoError(t, err)
}
func TestUnitContractBytecodeQueryValidateWrong(t *testing.T) {
t.Parallel()
client, err := _NewMockClient()
client.SetLedgerID(*NewLedgerIDTestnet())
require.NoError(t, err)
client.SetAutoValidateChecksums(true)
contractID, err := ContractIDFromString("0.0.123-rmkykd")
require.NoError(t, err)
contractBytecode := NewContractBytecodeQuery().
SetContractID(contractID)
err = contractBytecode.validateNetworkOnIDs(client)
assert.Error(t, err)
if err != nil {
assert.Equal(t, "network mismatch or wrong checksum given, given checksum: rmkykd, correct checksum esxsf, network: testnet", err.Error())
}
}
func TestUnitContractBytecodeQueryMock(t *testing.T) {
t.Parallel()
smartContractBytecode := []byte("608060405234801561001057600080fd5b506040516104d73803806104d78339818101604052602081101561003357600080fd5b810190808051604051939291908464010000000082111561005357600080fd5b90830190602082018581111561006857600080fd5b825164010000000081118282018810171561008257600080fd5b82525081516020918201929091019080838360005b838110156100af578181015183820152602001610097565b50505050905090810190601f1680156100dc5780820380516001836020036101000a031916815260200191505b506040525050600080546001600160a01b0319163317905550805161010890600190602084019061010f565b50506101aa565b828054600181600116156101000203166002900490600052602060002090601f016020900481019282601f1061015057805160ff191683800117855561017d565b8280016001018555821561017d579182015b8281111561017d578251825591602001919060010190610162565b5061018992915061018d565b5090565b6101a791905b808211156101895760008155600101610193565b90565b61031e806101b96000396000f3fe608060405234801561001057600080fd5b50600436106100415760003560e01c8063368b87721461004657806341c0e1b5146100ee578063ce6d41de146100f6575b600080fd5b6100ec6004803603602081101561005c57600080fd5b81019060208101813564010000000081111561007757600080fd5b82018360208201111561008957600080fd5b803590602001918460018302840111640100000000831117156100ab57600080fd5b91908080601f016020809104026020016040519081016040528093929190818152602001838380828437600092019190915250929550610173945050505050565b005b6100ec6101a2565b6100fe6101ba565b6040805160208082528351818301528351919283929083019185019080838360005b83811015610138578181015183820152602001610120565b50505050905090810190601f1680156101655780820380516001836020036101000a031916815260200191505b509250505060405180910390f35b6000546001600160a01b0316331461018a5761019f565b805161019d906001906020840190610250565b505b50565b6000546001600160a01b03163314156101b85733ff5b565b60018054604080516020601f600260001961010087891615020190951694909404938401819004810282018101909252828152606093909290918301828280156102455780601f1061021a57610100808354040283529160200191610245565b820191906000526020600020905b81548152906001019060200180831161022857829003601f168201915b505050505090505b90565b828054600181600116156101000203166002900490600052602060002090601f016020900481019282601f1061029157805160ff19168380011785556102be565b828001600101855582156102be579182015b828111156102be5782518255916020019190600101906102a3565b506102ca9291506102ce565b5090565b61024d91905b808211156102ca57600081556001016102d456fea264697066735822122084964d4c3f6bc912a9d20e14e449721012d625aa3c8a12de41ae5519752fc89064736f6c63430006000033")
responses := [][]interface{}{{
&services.Response{
Response: &services.Response_ContractGetBytecodeResponse{
ContractGetBytecodeResponse: &services.ContractGetBytecodeResponse{
Header: &services.ResponseHeader{NodeTransactionPrecheckCode: services.ResponseCodeEnum_OK, ResponseType: services.ResponseType_COST_ANSWER, Cost: 2},
},
},
},
&services.Response{
Response: &services.Response_ContractGetBytecodeResponse{
ContractGetBytecodeResponse: &services.ContractGetBytecodeResponse{
Header: &services.ResponseHeader{NodeTransactionPrecheckCode: services.ResponseCodeEnum_OK, ResponseType: services.ResponseType_COST_ANSWER, Cost: 2},
},
},
},
&services.Response{
Response: &services.Response_ContractGetBytecodeResponse{
ContractGetBytecodeResponse: &services.ContractGetBytecodeResponse{
Header: &services.ResponseHeader{NodeTransactionPrecheckCode: services.ResponseCodeEnum_OK, ResponseType: services.ResponseType_ANSWER_ONLY, Cost: 2},
Bytecode: smartContractBytecode,
},
},
},
}}
client, server := NewMockClientAndServer(responses)
defer server.Close()
query := NewContractBytecodeQuery().
SetNodeAccountIDs([]AccountID{{Account: 3}}).
SetContractID(ContractID{Contract: 3}).
SetMaxQueryPayment(NewHbar(2))
_, err := query.GetCost(client)
require.NoError(t, err)
bytecode, err := query.Execute(client)
require.NoError(t, err)
require.Equal(t, bytes.Compare(bytecode, smartContractBytecode), 0)
}
func TestUnitContractBytecodeQueryGet(t *testing.T) {
t.Parallel()
contractID := ContractID{Contract: 7}
balance := NewContractBytecodeQuery().
SetContractID(contractID).
SetQueryPayment(NewHbar(2)).
SetMaxQueryPayment(NewHbar(10)).
SetNodeAccountIDs([]AccountID{{Account: 10}, {Account: 11}, {Account: 12}})
balance.GetContractID()
balance.GetNodeAccountIDs()
balance.GetMinBackoff()
balance.GetMaxBackoff()
balance.GetMaxRetryCount()
balance.GetPaymentTransactionID()
balance.GetQueryPayment()
balance.GetMaxQueryPayment()
}
func TestUnitContractBytecodeQuerySetNothing(t *testing.T) {
t.Parallel()
balance := NewContractBytecodeQuery()
balance.GetContractID()
balance.GetNodeAccountIDs()
balance.GetMinBackoff()
balance.GetMaxBackoff()
balance.GetMaxRetryCount()
balance.GetPaymentTransactionID()
balance.GetQueryPayment()
balance.GetMaxQueryPayment()
}
func TestUnitContractBytecodeQueryCoverage(t *testing.T) {
t.Parallel()
checksum := "dmqui"
grpc := time.Second * 3
contract := ContractID{Contract: 3, checksum: &checksum}
nodeAccountID := []AccountID{{Account: 10}}
transactionID := TransactionIDGenerate(AccountID{Account: 324})
client, err := _NewMockClient()
client.SetLedgerID(*NewLedgerIDTestnet())
require.NoError(t, err)
client.SetAutoValidateChecksums(true)
query := NewContractBytecodeQuery().
SetMaxRetry(3).
SetMaxBackoff(time.Second * 30).
SetMinBackoff(time.Second * 10).
SetContractID(contract).
SetNodeAccountIDs(nodeAccountID).
SetPaymentTransactionID(transactionID).
SetMaxQueryPayment(NewHbar(23)).
SetQueryPayment(NewHbar(3)).
SetGrpcDeadline(&grpc)
err = query.validateNetworkOnIDs(client)
require.NoError(t, err)
query.GetNodeAccountIDs()
query.GetMaxBackoff()
query.GetMinBackoff()
query.getName()
query.GetContractID()
}