Skip to content

Commit 05be1ad

Browse files
committed
add second account mock
1 parent 6ee649f commit 05be1ad

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

multichain.go

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -115,8 +115,9 @@ const (
115115
// These assets are define separately because they are mock assets. These
116116
// assets should only be used for testing.
117117

118-
AMOCK = Asset("AMOCK") // Account-based mock asset
119-
UMOCK = Asset("UMOCK") // UTXO-based mock asset
118+
AMOCK1 = Asset("AMOCK1") // Account-based mock asset
119+
AMOCK2 = Asset("AMOCK2") // Account-based mock asset
120+
UMOCK = Asset("UMOCK") // UTXO-based mock asset
120121
)
121122

122123
// OriginChain returns the chain upon which the asset originates. For example,
@@ -151,8 +152,10 @@ func (asset Asset) OriginChain() Chain {
151152
// These assets are define separately because they are mock assets. These
152153
// assets should only be used for testing.
153154

154-
case AMOCK:
155-
return AccountMocker
155+
case AMOCK1:
156+
return AccountMocker1
157+
case AMOCK2:
158+
return AccountMocker2
156159
case UMOCK:
157160
return UTXOMocker
158161

@@ -171,7 +174,7 @@ func (asset Asset) ChainType() ChainType {
171174

172175
// These assets are define separately because they are mock assets. These
173176
// assets should only be used for testing.
174-
case AMOCK:
177+
case AMOCK1, AMOCK2:
175178
return ChainTypeAccountBased
176179
case UMOCK:
177180
return ChainTypeUTXOBased
@@ -221,8 +224,9 @@ const (
221224
// These chains are define separately because they are mock chains. These
222225
// chains should only be used for testing.
223226

224-
AccountMocker = Chain("AccountMocker")
225-
UTXOMocker = Chain("UTXOMocker")
227+
AccountMocker1 = Chain("AccountMocker1")
228+
AccountMocker2 = Chain("AccountMocker2")
229+
UTXOMocker = Chain("UTXOMocker")
226230
)
227231

228232
// SizeHint returns the number of bytes required to represent the chain in
@@ -254,7 +258,7 @@ func (chain Chain) ChainType() ChainType {
254258

255259
// These chains are define separately because they are mock chains. These
256260
// chains should only be used for testing.
257-
case AccountMocker:
261+
case AccountMocker1, AccountMocker2:
258262
return ChainTypeAccountBased
259263
case UTXOMocker:
260264
return ChainTypeUTXOBased

0 commit comments

Comments
 (0)