ExternalVendorCode/corda
1
0
Fork 0
mirror of https://github.com/corda/corda.git synced 2025-01-29 15:43:55 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Make TransactionBuilder's input references deep copied on copy.

Browse Source
This commit is contained in:
Xavier Leprêtre 2020-04-25 22:50:09 +04:00
parent 6581815ec2
commit 9396ee9551
2 changed files with 106 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
core-tests/src/test/kotlin/net/corda/coretests/transactions/TransactionBuilderTest.kt
View File

@ -29,6 +29,7 @@ import org.junit.Before
import org.junit.Rule
import org.junit.Test
import java.security.PublicKey
import java.time.Instant
class TransactionBuilderTest {
@Rule
@ -150,4 +151,103 @@ class TransactionBuilderTest {
override val signerKeys: List<PublicKey> get() = parties.map { it.owningKey }
}, DummyContract.PROGRAM_ID, signerKeys = parties.map { it.owningKey })
@Test(timeout=300_000)
fun `list accessors are mutable copies`() {
val inputState1 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val inputStateRef1 = StateRef(SecureHash.randomSHA256(), 0)
val referenceState1 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val referenceStateRef1 = StateRef(SecureHash.randomSHA256(), 1)
val timeWindow = TimeWindow.untilOnly(Instant.now())
val builder = TransactionBuilder(notary)
.addInputState(StateAndRef(inputState1, inputStateRef1))
.addAttachment(SecureHash.allOnesHash)
.addOutputState(TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary))
.addCommand(DummyCommandData, notary.owningKey)
.addReferenceState(StateAndRef(referenceState1, referenceStateRef1).referenced())
val inputState2 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val inputStateRef2 = StateRef(SecureHash.randomSHA256(), 0)
val referenceState2 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val referenceStateRef2 = StateRef(SecureHash.randomSHA256(), 1)
// List accessors are mutable.
(builder.inputStates() as ArrayList).add(inputStateRef2)
(builder.attachments() as ArrayList).add(SecureHash.zeroHash)
(builder.outputStates() as ArrayList).add(TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary))
(builder.commands() as ArrayList).add(Command(DummyCommandData, notary.owningKey))
(builder.referenceStates() as ArrayList).add(referenceStateRef2)
// List accessors are copies.
assertThat(builder.inputStates()).hasSize(1)
assertThat(builder.attachments()).hasSize(1)
assertThat(builder.outputStates()).hasSize(1)
assertThat(builder.commands()).hasSize(1)
assertThat(builder.referenceStates()).hasSize(1)
}
@Test(timeout=300_000)
fun `copy makes copy except lockId`() {
val inputState = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val inputStateRef = StateRef(SecureHash.randomSHA256(), 0)
val referenceState = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val referenceStateRef = StateRef(SecureHash.randomSHA256(), 1)
val timeWindow = TimeWindow.untilOnly(Instant.now())
val builder = TransactionBuilder(notary)
.addInputState(StateAndRef(inputState, inputStateRef))
.addAttachment(SecureHash.allOnesHash)
.addOutputState(TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary))
.addCommand(DummyCommandData, notary.owningKey)
.setTimeWindow(timeWindow)
.setPrivacySalt(PrivacySalt())
.addReferenceState(StateAndRef(referenceState, referenceStateRef).referenced())
val copy = builder.copy()
assertThat(builder.notary).isEqualTo(copy.notary)
assertThat(builder.lockId).isNotEqualTo(copy.lockId)
assertThat(builder.inputStates()).isEqualTo(copy.inputStates())
assertThat(builder.attachments()).isEqualTo(copy.attachments())
assertThat(builder.outputStates()).isEqualTo(copy.outputStates())
assertThat(builder.commands()).isEqualTo(copy.commands())
// assertThat(builder.timeWindow()).isEqualTo(copy.timeWindow())
// assertThat(builder.privacySalt()).isEqualTo(copy.privacySalt())
assertThat(builder.referenceStates()).isEqualTo(copy.referenceStates())
}
@Test(timeout=300_000)
fun `copy makes deep copy of lists`() {
val inputState1 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val inputStateRef1 = StateRef(SecureHash.randomSHA256(), 0)
val referenceState1 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val referenceStateRef1 = StateRef(SecureHash.randomSHA256(), 1)
val builder = TransactionBuilder(notary)
.addInputState(StateAndRef(inputState1, inputStateRef1))
.addAttachment(SecureHash.allOnesHash)
.addOutputState(TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary))
.addCommand(DummyCommandData, notary.owningKey)
.addReferenceState(StateAndRef(referenceState1, referenceStateRef1).referenced())
val inputState2 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val inputStateRef2 = StateRef(SecureHash.randomSHA256(), 0)
val referenceState2 = TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary)
val referenceStateRef2 = StateRef(SecureHash.randomSHA256(), 1)
val copy = builder.copy()
.addInputState(StateAndRef(inputState2, inputStateRef2))
.addAttachment(SecureHash.zeroHash)
.addOutputState(TransactionState(DummyState(), DummyContract.PROGRAM_ID, notary))
.addCommand(DummyCommandData, notary.owningKey)
.addReferenceState(StateAndRef(referenceState2, referenceStateRef2).referenced())
// Lists on the copy are longer
assertThat(copy.inputStates()).hasSize(2)
assertThat(copy.attachments()).hasSize(2)
assertThat(copy.outputStates()).hasSize(2)
assertThat(copy.commands()).hasSize(2)
assertThat(copy.referenceStates()).hasSize(2)
// Lists on the original are unchanged
assertThat(builder.inputStates()).hasSize(1)
assertThat(builder.attachments()).hasSize(1)
assertThat(builder.outputStates()).hasSize(1)
assertThat(builder.commands()).hasSize(1)
assertThat(builder.referenceStates()).hasSize(1)
}
}

12
core/src/main/kotlin/net/corda/core/transactions/TransactionBuilder.kt
View File

@ -100,7 +100,7 @@ open class TransactionBuilder(
commands = ArrayList(commands),
window = window,
privacySalt = privacySalt,
references = references,
references = ArrayList(references),
serviceHub = serviceHub
)
t.inputsWithTransactionState.addAll(this.inputsWithTransactionState)
@ -813,19 +813,19 @@ open class TransactionBuilder(
this.privacySalt = privacySalt
}
/** Returns an immutable list of input [StateRef]s. */
/** Returns a mutable copy of the list of input [StateRef]s. */
fun inputStates(): List<StateRef> = ArrayList(inputs)
/** Returns an immutable list of reference input [StateRef]s. */
/** Returns a mutable copy of the list of reference input [StateRef]s. */
fun referenceStates(): List<StateRef> = ArrayList(references)
/** Returns an immutable list of attachment hashes. */
/** Returns a mutable copy of the list of attachment hashes. */
fun attachments(): List<AttachmentId> = ArrayList(attachments)
/** Returns an immutable list of output [TransactionState]s. */
/** Returns a mutable copy of the list of output [TransactionState]s. */
fun outputStates(): List<TransactionState<*>> = ArrayList(outputs)
/** Returns an immutable list of [Command]s. */
/** Returns a mutable copy of the list of [Command]s. */
fun commands(): List<Command<*>> = ArrayList(commands)
/**