11 "github.com/bytom/bytom/account"
12 "github.com/bytom/bytom/consensus"
13 dbm "github.com/bytom/bytom/database/leveldb"
14 "github.com/bytom/bytom/protocol/bc"
15 "github.com/bytom/bytom/protocol/bc/types"
16 "github.com/bytom/bytom/testutil"
19 func TestGetAccountUtxos(t *testing.T) {
20 testDB := dbm.NewDB("testdb", "leveldb", "temp")
21 defer os.RemoveAll("temp")
24 dbUtxos map[string]*account.UTXO
25 unconfirmedUtxos []*account.UTXO
29 wantUtxos []*account.UTXO
32 dbUtxos: map[string]*account.UTXO{},
35 isSmartContract: false,
36 wantUtxos: []*account.UTXO{},
39 dbUtxos: map[string]*account.UTXO{
40 string(account.StandardUTXOKey(bc.Hash{V0: 1})): &account.UTXO{
41 OutputID: bc.Hash{V0: 1},
43 string(account.StandardUTXOKey(bc.Hash{V0: 2})): &account.UTXO{
44 OutputID: bc.Hash{V0: 2},
46 string(account.StandardUTXOKey(bc.Hash{V0: 3})): &account.UTXO{
47 OutputID: bc.Hash{V0: 3},
49 string(account.ContractUTXOKey(bc.Hash{V0: 4})): &account.UTXO{
50 OutputID: bc.Hash{V0: 4},
53 unconfirmedUtxos: []*account.UTXO{},
55 isSmartContract: false,
56 wantUtxos: []*account.UTXO{
57 &account.UTXO{OutputID: bc.Hash{V0: 1}},
58 &account.UTXO{OutputID: bc.Hash{V0: 2}},
59 &account.UTXO{OutputID: bc.Hash{V0: 3}},
63 dbUtxos: map[string]*account.UTXO{
64 string(account.StandardUTXOKey(bc.Hash{V0: 1})): &account.UTXO{
65 OutputID: bc.Hash{V0: 1},
67 string(account.StandardUTXOKey(bc.Hash{V0: 2})): &account.UTXO{
68 OutputID: bc.Hash{V0: 2},
70 string(account.StandardUTXOKey(bc.Hash{V0: 3})): &account.UTXO{
71 OutputID: bc.Hash{V0: 3},
73 string(account.ContractUTXOKey(bc.Hash{V0: 4})): &account.UTXO{
74 OutputID: bc.Hash{V0: 4},
77 unconfirmedUtxos: []*account.UTXO{
79 OutputID: bc.Hash{V0: 5},
80 ControlProgram: []byte("smart contract"),
85 isSmartContract: true,
86 wantUtxos: []*account.UTXO{
87 &account.UTXO{OutputID: bc.Hash{V0: 4}},
91 dbUtxos: map[string]*account.UTXO{
92 string(account.StandardUTXOKey(bc.Hash{V0: 1})): &account.UTXO{
93 OutputID: bc.Hash{V0: 1},
95 string(account.StandardUTXOKey(bc.Hash{V0: 1, V1: 2})): &account.UTXO{
96 OutputID: bc.Hash{V0: 1, V1: 2},
98 string(account.StandardUTXOKey(bc.Hash{V0: 2})): &account.UTXO{
99 OutputID: bc.Hash{V0: 2},
101 string(account.StandardUTXOKey(bc.Hash{V0: 2, V1: 2})): &account.UTXO{
102 OutputID: bc.Hash{V0: 2, V1: 2},
105 unconfirmedUtxos: []*account.UTXO{
107 OutputID: bc.Hash{V0: 6},
108 ControlProgram: []byte{0x51},
111 id: "0000000000000002",
113 isSmartContract: false,
114 wantUtxos: []*account.UTXO{
115 &account.UTXO{OutputID: bc.Hash{V0: 2}},
116 &account.UTXO{OutputID: bc.Hash{V0: 2, V1: 2}},
120 dbUtxos: map[string]*account.UTXO{
121 string(account.StandardUTXOKey(bc.Hash{V0: 3})): &account.UTXO{
122 OutputID: bc.Hash{V0: 3},
124 string(account.ContractUTXOKey(bc.Hash{V0: 4})): &account.UTXO{
125 OutputID: bc.Hash{V0: 4},
128 unconfirmedUtxos: []*account.UTXO{
130 OutputID: bc.Hash{V0: 5},
131 ControlProgram: []byte("smart contract"),
134 OutputID: bc.Hash{V0: 6},
135 ControlProgram: []byte{0x51},
140 isSmartContract: true,
141 wantUtxos: []*account.UTXO{
143 OutputID: bc.Hash{V0: 5},
144 ControlProgram: []byte("smart contract"),
147 OutputID: bc.Hash{V0: 4},
152 dbUtxos: map[string]*account.UTXO{
153 string(account.StandardUTXOKey(bc.Hash{V0: 3})): &account.UTXO{
154 OutputID: bc.Hash{V0: 3},
156 string(account.ContractUTXOKey(bc.Hash{V0: 4})): &account.UTXO{
157 OutputID: bc.Hash{V0: 4},
160 unconfirmedUtxos: []*account.UTXO{
162 OutputID: bc.Hash{V0: 5},
163 ControlProgram: []byte("smart contract"),
166 OutputID: bc.Hash{V0: 6},
167 ControlProgram: []byte{0x51},
172 isSmartContract: false,
173 wantUtxos: []*account.UTXO{
175 OutputID: bc.Hash{V0: 6},
176 ControlProgram: []byte{0x51},
179 OutputID: bc.Hash{V0: 3},
185 w := &Wallet{DB: testDB}
186 for i, c := range cases {
187 for k, u := range c.dbUtxos {
188 data, err := json.Marshal(u)
192 testDB.Set([]byte(k), data)
195 w.AccountMgr = account.NewManager(testDB, nil)
196 w.AccountMgr.AddUnconfirmedUtxo(c.unconfirmedUtxos)
197 gotUtxos := w.GetAccountUtxos("", c.id, c.unconfirmed, c.isSmartContract)
198 if !testutil.DeepEqual(gotUtxos, c.wantUtxos) {
199 t.Errorf("case %d: got %v want %v", i, gotUtxos, c.wantUtxos)
202 for k := range c.dbUtxos {
203 testDB.Delete([]byte(k))
208 //because can not pass by btm2.0 branch
209 func TestFilterAccountUtxo(t *testing.T) {
210 testDB := dbm.NewDB("testdb", "leveldb", "temp")
211 defer os.RemoveAll("temp")
214 dbPrograms map[string]*account.CtrlProgram
215 input []*account.UTXO
216 wantUtxos []*account.UTXO
219 dbPrograms: map[string]*account.CtrlProgram{},
220 input: []*account.UTXO{},
221 wantUtxos: []*account.UTXO{},
224 dbPrograms: map[string]*account.CtrlProgram{
225 "436f6e74726163743a2a37a64a4e15a772ab43bf3f5956d0d1f353946496788e7f40d0ff1796286a6f": &account.CtrlProgram{
226 AccountID: "testAccount",
227 Address: "testAddress",
232 input: []*account.UTXO{
234 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
235 AssetID: bc.AssetID{V0: 1},
239 ControlProgram: []byte{0x91},
240 AssetID: bc.AssetID{V0: 1},
244 wantUtxos: []*account.UTXO{
246 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
247 AssetID: bc.AssetID{V0: 1},
249 AccountID: "testAccount",
250 Address: "testAddress",
251 ControlProgramIndex: 53,
257 dbPrograms: map[string]*account.CtrlProgram{},
258 input: []*account.UTXO{
260 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
261 AssetID: bc.AssetID{V0: 1},
265 ControlProgram: []byte{0x91},
266 AssetID: bc.AssetID{V0: 1},
270 wantUtxos: []*account.UTXO{},
273 dbPrograms: map[string]*account.CtrlProgram{
274 "436f6e74726163743a2a37a64a4e15a772ab43bf3f5956d0d1f353946496788e7f40d0ff1796286a6f": &account.CtrlProgram{
275 AccountID: "testAccount",
276 Address: "testAddress",
280 "436f6e74726163743adb4d86262c12ba70d50b3ca3ae102d5682436243bd1e8c79569603f75675036a": &account.CtrlProgram{
281 AccountID: "testAccount2",
282 Address: "testAddress2",
287 input: []*account.UTXO{
289 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
290 AssetID: bc.AssetID{V0: 1},
294 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
295 AssetID: bc.AssetID{V0: 1},
299 ControlProgram: []byte{0x00, 0x14, 0xc6, 0xbf, 0x22, 0x19, 0x64, 0x2a, 0xc5, 0x9e, 0x5b, 0xe4, 0xeb, 0xdf, 0x5b, 0x22, 0x49, 0x56, 0xa7, 0x98, 0xa4, 0xdf},
300 AssetID: bc.AssetID{V0: 1},
304 wantUtxos: []*account.UTXO{
306 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
307 AssetID: bc.AssetID{V0: 1},
309 AccountID: "testAccount",
310 Address: "testAddress",
311 ControlProgramIndex: 53,
315 ControlProgram: []byte{0x00, 0x14, 0x62, 0x50, 0x18, 0xb6, 0x85, 0x77, 0xba, 0x9b, 0x26, 0x19, 0xc8, 0x1d, 0x2e, 0x96, 0xba, 0x22, 0xbe, 0x77, 0x77, 0xd7},
316 AssetID: bc.AssetID{V0: 1},
318 AccountID: "testAccount",
319 Address: "testAddress",
320 ControlProgramIndex: 53,
324 ControlProgram: []byte{0x00, 0x14, 0xc6, 0xbf, 0x22, 0x19, 0x64, 0x2a, 0xc5, 0x9e, 0x5b, 0xe4, 0xeb, 0xdf, 0x5b, 0x22, 0x49, 0x56, 0xa7, 0x98, 0xa4, 0xdf},
325 AssetID: bc.AssetID{V0: 1},
327 AccountID: "testAccount2",
328 Address: "testAddress2",
329 ControlProgramIndex: 72,
336 w := &Wallet{DB: testDB}
337 for i, c := range cases {
338 for s, p := range c.dbPrograms {
339 data, err := json.Marshal(p)
343 key, err := hex.DecodeString(s)
347 testDB.Set(key, data)
349 gotUtxos := w.filterAccountUtxo(c.input)
350 sort.Slice(gotUtxos[:], func(i, j int) bool {
351 return gotUtxos[i].Amount < gotUtxos[j].Amount
354 if !testutil.DeepEqual(gotUtxos, c.wantUtxos) {
355 t.Errorf("case %d: got %v want %v", i, gotUtxos, c.wantUtxos)
357 for s := range c.dbPrograms {
358 key, err := hex.DecodeString(s)
367 func TestTxInToUtxos(t *testing.T) {
371 wantUtxos []*account.UTXO
374 tx: types.NewTx(types.TxData{
375 Inputs: []*types.TxInput{
376 types.NewCoinbaseInput([]byte{0x51}),
378 Outputs: []*types.TxOutput{
379 types.NewOriginalTxOutput(*consensus.BTMAssetID, 41250000000, []byte{0x51}, nil),
383 wantUtxos: []*account.UTXO{},
386 tx: types.NewTx(types.TxData{
387 Inputs: []*types.TxInput{
388 types.NewIssuanceInput([]byte{}, 4125, []byte{0x51}, [][]byte{}, []byte{}),
390 Outputs: []*types.TxOutput{
391 types.NewOriginalTxOutput(*consensus.BTMAssetID, 4125, []byte{0x51}, nil),
395 wantUtxos: []*account.UTXO{},
398 tx: types.NewTx(types.TxData{
399 Inputs: []*types.TxInput{
400 types.NewSpendInput([][]byte{}, bc.Hash{V0: 1}, bc.AssetID{V0: 1}, 1, 1, []byte{0x51}, nil),
401 types.NewSpendInput([][]byte{}, bc.Hash{V0: 2}, bc.AssetID{V0: 1}, 3, 2, []byte{0x52}, nil),
402 types.NewSpendInput([][]byte{}, bc.Hash{V0: 3}, *consensus.BTMAssetID, 5, 3, []byte{0x53}, nil),
403 types.NewSpendInput([][]byte{}, bc.Hash{V0: 4}, *consensus.BTMAssetID, 7, 4, []byte{0x54}, nil),
405 Outputs: []*types.TxOutput{
406 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 4, []byte{0x51}, nil),
407 types.NewOriginalTxOutput(*consensus.BTMAssetID, 12, []byte{0x53}, nil),
411 wantUtxos: []*account.UTXO{
413 OutputID: testutil.MustDecodeHash("d7317174b78e6efdd060f19031b9b4d8dfdc8218b8fe7d86324f8c35b9cc572c"),
414 AssetID: bc.AssetID{V0: 1},
416 ControlProgram: []byte{0x51},
417 SourceID: bc.Hash{V0: 1},
421 OutputID: testutil.MustDecodeHash("2170b5a9ed124f4c2f691292dc44cc9a5b834c286a39738aa159c646cce14d95"),
422 AssetID: bc.AssetID{V0: 1},
424 ControlProgram: []byte{0x52},
425 SourceID: bc.Hash{V0: 2},
429 OutputID: testutil.MustDecodeHash("e07a0c076fcee0faccf6fc329875c3273120faaee87d273ff1cea5c64b2fb1e3"),
430 AssetID: *consensus.BTMAssetID,
432 ControlProgram: []byte{0x53},
433 SourceID: bc.Hash{V0: 3},
437 OutputID: testutil.MustDecodeHash("30fe2e3180356a847f152ce2dabab99ebe8df97ebbf100881d9591527f9fd738"),
438 AssetID: *consensus.BTMAssetID,
440 ControlProgram: []byte{0x54},
441 SourceID: bc.Hash{V0: 4},
447 tx: types.NewTx(types.TxData{
448 Inputs: []*types.TxInput{
449 types.NewSpendInput([][]byte{}, bc.Hash{V0: 3}, *consensus.BTMAssetID, 5, 3, []byte{0x53}, nil),
450 types.NewSpendInput([][]byte{}, bc.Hash{V0: 4}, *consensus.BTMAssetID, 7, 4, []byte{0x54}, nil),
452 Outputs: []*types.TxOutput{
453 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 4, []byte{0x51}, nil),
454 types.NewOriginalTxOutput(*consensus.BTMAssetID, 12, []byte{0x53}, nil),
458 wantUtxos: []*account.UTXO{
460 OutputID: testutil.MustDecodeHash("e07a0c076fcee0faccf6fc329875c3273120faaee87d273ff1cea5c64b2fb1e3"),
461 AssetID: *consensus.BTMAssetID,
463 ControlProgram: []byte{0x53},
464 SourceID: bc.Hash{V0: 3},
468 OutputID: testutil.MustDecodeHash("30fe2e3180356a847f152ce2dabab99ebe8df97ebbf100881d9591527f9fd738"),
469 AssetID: *consensus.BTMAssetID,
471 ControlProgram: []byte{0x54},
472 SourceID: bc.Hash{V0: 4},
479 for i, c := range cases {
480 if gotUtxos := txInToUtxos(c.tx); !testutil.DeepEqual(gotUtxos, c.wantUtxos) {
481 for k, v := range gotUtxos {
482 data, _ := json.Marshal(v)
483 fmt.Println(k, string(data))
485 for k, v := range c.wantUtxos {
486 data, _ := json.Marshal(v)
487 fmt.Println(k, string(data))
490 t.Errorf("case %d: got %v want %v", i, gotUtxos, c.wantUtxos)
495 func TestTxOutToUtxos(t *testing.T) {
500 wantUtxos []*account.UTXO
503 tx: types.NewTx(types.TxData{
504 Inputs: []*types.TxInput{
505 types.NewCoinbaseInput([]byte{0x51}),
507 Outputs: []*types.TxOutput{
508 types.NewOriginalTxOutput(*consensus.BTMAssetID, 41250000000, []byte{0x51}, nil),
513 wantUtxos: []*account.UTXO{
515 OutputID: testutil.MustDecodeHash("5ad31f023737c301190026c6e97da10715d5455d9bb32ace3104454faefd2bb6"),
516 AssetID: *consensus.BTMAssetID,
518 ControlProgram: []byte{0x51},
519 SourceID: bc.NewHash([32]byte{0xb4, 0x7e, 0x94, 0x31, 0x88, 0xfe, 0xd3, 0xe9, 0xac, 0x99, 0x7c, 0xfc, 0x99, 0x6d, 0xd7, 0x4d, 0x04, 0x10, 0x77, 0xcb, 0x1c, 0xf8, 0x95, 0x14, 0x00, 0xe3, 0x42, 0x00, 0x8d, 0x05, 0xec, 0xdc}),
526 tx: types.NewTx(types.TxData{
527 Inputs: []*types.TxInput{
528 types.NewSpendInput([][]byte{}, bc.Hash{V0: 1}, bc.AssetID{V0: 1}, 5, 1, []byte{0x51}, nil),
529 types.NewSpendInput([][]byte{}, bc.Hash{V0: 2}, *consensus.BTMAssetID, 7, 1, []byte{0x51}, nil),
531 Outputs: []*types.TxOutput{
532 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 2, []byte{0x51}, nil),
533 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 3, []byte{0x52}, nil),
534 types.NewOriginalTxOutput(*consensus.BTMAssetID, 2, []byte{0x53}, nil),
535 types.NewOriginalTxOutput(*consensus.BTMAssetID, 5, []byte{0x54}, nil),
540 wantUtxos: []*account.UTXO{
542 OutputID: testutil.MustDecodeHash("55de32fa7e96499be625063bc4c7f66bbe5d7b60f233d3a3d7265a9501c77e96"),
543 AssetID: bc.AssetID{V0: 1},
545 ControlProgram: []byte{0x51},
546 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
550 OutputID: testutil.MustDecodeHash("f256f40aa36df9cf954fdf82d5835815adfd21579289063525c64cff59bc6d96"),
551 AssetID: bc.AssetID{V0: 1},
553 ControlProgram: []byte{0x52},
554 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
558 OutputID: testutil.MustDecodeHash("60da633ce48a8bf995a23c040c1b52d543e5818abd7fc0b0faa355cf54acbcca"),
559 AssetID: *consensus.BTMAssetID,
561 ControlProgram: []byte{0x53},
562 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
566 OutputID: testutil.MustDecodeHash("550c883fd09fdccd4a5671c698d0874de9d713fa0b194f0ddffeae8ae79b57fc"),
567 AssetID: *consensus.BTMAssetID,
569 ControlProgram: []byte{0x54},
570 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
576 tx: types.NewTx(types.TxData{
577 Inputs: []*types.TxInput{
578 types.NewSpendInput([][]byte{}, bc.Hash{V0: 1}, bc.AssetID{V0: 1}, 5, 1, []byte{0x51}, nil),
579 types.NewSpendInput([][]byte{}, bc.Hash{V0: 2}, *consensus.BTMAssetID, 7, 1, []byte{0x51}, nil),
581 Outputs: []*types.TxOutput{
582 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 2, []byte{0x51}, nil),
583 types.NewOriginalTxOutput(bc.AssetID{V0: 1}, 3, []byte{0x52}, nil),
584 types.NewOriginalTxOutput(*consensus.BTMAssetID, 2, []byte{0x53}, nil),
585 types.NewOriginalTxOutput(*consensus.BTMAssetID, 5, []byte{0x54}, nil),
590 wantUtxos: []*account.UTXO{
592 OutputID: testutil.MustDecodeHash("55de32fa7e96499be625063bc4c7f66bbe5d7b60f233d3a3d7265a9501c77e96"),
593 AssetID: bc.AssetID{V0: 1},
595 ControlProgram: []byte{0x51},
596 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
600 OutputID: testutil.MustDecodeHash("f256f40aa36df9cf954fdf82d5835815adfd21579289063525c64cff59bc6d96"),
601 AssetID: bc.AssetID{V0: 1},
603 ControlProgram: []byte{0x52},
604 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
608 OutputID: testutil.MustDecodeHash("60da633ce48a8bf995a23c040c1b52d543e5818abd7fc0b0faa355cf54acbcca"),
609 AssetID: *consensus.BTMAssetID,
611 ControlProgram: []byte{0x53},
612 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
616 OutputID: testutil.MustDecodeHash("550c883fd09fdccd4a5671c698d0874de9d713fa0b194f0ddffeae8ae79b57fc"),
617 AssetID: *consensus.BTMAssetID,
619 ControlProgram: []byte{0x54},
620 SourceID: testutil.MustDecodeHash("c0e16f80a168dad26194f9ba5a4244d9d52e08e4636aa50aa58a5b8d65e969d4"),
627 for i, c := range cases {
628 if gotUtxos := txOutToUtxos(c.tx, c.blockHeight); !testutil.DeepEqual(gotUtxos, c.wantUtxos) {
629 for k, v := range gotUtxos {
630 data, _ := json.Marshal(v)
631 fmt.Println("got:", k, string(data))
633 for k, v := range c.wantUtxos {
634 data, _ := json.Marshal(v)
635 fmt.Println("want:", k, string(data))
638 t.Errorf("case %d: got %v want %v", i, gotUtxos, c.wantUtxos)