460 lines
9.5 KiB
JavaScript
460 lines
9.5 KiB
JavaScript
// @flow
|
|
import bs58 from 'bs58';
|
|
import nacl from 'tweetnacl';
|
|
|
|
import {Account} from '../src/account';
|
|
import {PublicKey} from '../src/publickey';
|
|
import {Transaction} from '../src/transaction';
|
|
import {StakeProgram} from '../src/stake-program';
|
|
import {SystemProgram} from '../src/system-program';
|
|
import {Message} from '../src/message';
|
|
|
|
test('signPartial', () => {
|
|
const account1 = new Account();
|
|
const account2 = new Account();
|
|
const recentBlockhash = account1.publicKey.toBase58(); // Fake recentBlockhash
|
|
const transfer = SystemProgram.transfer({
|
|
fromPubkey: account1.publicKey,
|
|
toPubkey: account2.publicKey,
|
|
lamports: 123,
|
|
});
|
|
|
|
const transaction = new Transaction({recentBlockhash}).add(transfer);
|
|
transaction.sign(account1, account2);
|
|
|
|
const partialTransaction = new Transaction({recentBlockhash}).add(transfer);
|
|
partialTransaction.signPartial(account1, account2.publicKey);
|
|
expect(partialTransaction.signatures[1].signature).toBeNull();
|
|
partialTransaction.addSigner(account2);
|
|
|
|
expect(partialTransaction).toEqual(transaction);
|
|
});
|
|
|
|
test('transfer signatures', () => {
|
|
const account1 = new Account();
|
|
const account2 = new Account();
|
|
const recentBlockhash = account1.publicKey.toBase58(); // Fake recentBlockhash
|
|
const transfer1 = SystemProgram.transfer({
|
|
fromPubkey: account1.publicKey,
|
|
toPubkey: account2.publicKey,
|
|
lamports: 123,
|
|
});
|
|
const transfer2 = SystemProgram.transfer({
|
|
fromPubkey: account2.publicKey,
|
|
toPubkey: account1.publicKey,
|
|
lamports: 123,
|
|
});
|
|
|
|
const orgTransaction = new Transaction({recentBlockhash}).add(
|
|
transfer1,
|
|
transfer2,
|
|
);
|
|
orgTransaction.sign(account1, account2);
|
|
|
|
const newTransaction = new Transaction({
|
|
recentBlockhash: orgTransaction.recentBlockhash,
|
|
signatures: orgTransaction.signatures,
|
|
}).add(transfer1, transfer2);
|
|
|
|
expect(newTransaction).toEqual(orgTransaction);
|
|
});
|
|
|
|
test('dedup signatures', () => {
|
|
const account1 = new Account();
|
|
const account2 = new Account();
|
|
const recentBlockhash = account1.publicKey.toBase58(); // Fake recentBlockhash
|
|
const transfer1 = SystemProgram.transfer({
|
|
fromPubkey: account1.publicKey,
|
|
toPubkey: account2.publicKey,
|
|
lamports: 123,
|
|
});
|
|
const transfer2 = SystemProgram.transfer({
|
|
fromPubkey: account1.publicKey,
|
|
toPubkey: account2.publicKey,
|
|
lamports: 123,
|
|
});
|
|
|
|
const orgTransaction = new Transaction({recentBlockhash}).add(
|
|
transfer1,
|
|
transfer2,
|
|
);
|
|
orgTransaction.sign(account1);
|
|
});
|
|
|
|
test('use nonce', () => {
|
|
const account1 = new Account();
|
|
const account2 = new Account();
|
|
const nonceAccount = new Account();
|
|
const nonce = account2.publicKey.toBase58(); // Fake Nonce hash
|
|
|
|
const nonceInfo = {
|
|
nonce,
|
|
nonceInstruction: SystemProgram.nonceAdvance({
|
|
noncePubkey: nonceAccount.publicKey,
|
|
authorizedPubkey: account1.publicKey,
|
|
}),
|
|
};
|
|
|
|
const transferTransaction = new Transaction({nonceInfo}).add(
|
|
SystemProgram.transfer({
|
|
fromPubkey: account1.publicKey,
|
|
toPubkey: account2.publicKey,
|
|
lamports: 123,
|
|
}),
|
|
);
|
|
transferTransaction.sign(account1);
|
|
|
|
let expectedData = Buffer.alloc(4);
|
|
expectedData.writeInt32LE(4, 0);
|
|
|
|
expect(transferTransaction.instructions).toHaveLength(2);
|
|
expect(transferTransaction.instructions[0].programId).toEqual(
|
|
SystemProgram.programId,
|
|
);
|
|
expect(transferTransaction.instructions[0].data).toEqual(expectedData);
|
|
expect(transferTransaction.recentBlockhash).toEqual(nonce);
|
|
|
|
const stakeAccount = new Account();
|
|
const voteAccount = new Account();
|
|
const stakeTransaction = new Transaction({nonceInfo}).add(
|
|
StakeProgram.delegate({
|
|
stakePubkey: stakeAccount.publicKey,
|
|
authorizedPubkey: account1.publicKey,
|
|
votePubkey: voteAccount.publicKey,
|
|
}),
|
|
);
|
|
stakeTransaction.sign(account1);
|
|
|
|
expect(stakeTransaction.instructions).toHaveLength(2);
|
|
expect(stakeTransaction.instructions[0].programId).toEqual(
|
|
SystemProgram.programId,
|
|
);
|
|
expect(stakeTransaction.instructions[0].data).toEqual(expectedData);
|
|
expect(stakeTransaction.recentBlockhash).toEqual(nonce);
|
|
});
|
|
|
|
test('parse wire format and serialize', () => {
|
|
const keypair = nacl.sign.keyPair.fromSeed(
|
|
Uint8Array.from(Array(32).fill(8)),
|
|
);
|
|
const sender = new Account(Buffer.from(keypair.secretKey)); // Arbitrary known account
|
|
const recentBlockhash = 'EETubP5AKHgjPAhzPAFcb8BAY1hMH639CWCFTqi3hq1k'; // Arbitrary known recentBlockhash
|
|
const recipient = new PublicKey(
|
|
'J3dxNj7nDRRqRRXuEMynDG57DkZK4jYRuv3Garmb1i99',
|
|
); // Arbitrary known public key
|
|
const transfer = SystemProgram.transfer({
|
|
fromPubkey: sender.publicKey,
|
|
toPubkey: recipient,
|
|
lamports: 49,
|
|
});
|
|
const expectedTransaction = new Transaction({recentBlockhash}).add(transfer);
|
|
expectedTransaction.sign(sender);
|
|
|
|
const wireTransaction = Buffer.from(
|
|
'AVuErQHaXv0SG0/PchunfxHKt8wMRfMZzqV0tkC5qO6owYxWU2v871AoWywGoFQr4z+q/7mE8lIufNl/kxj+nQ0BAAEDE5j2LG0aRXxRumpLXz29L2n8qTIWIY3ImX5Ba9F9k8r9Q5/Mtmcn8onFxt47xKj+XdXXd3C8j/FcPu7csUrz/AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAxJrndgN4IFTxep3s6kO0ROug7bEsbx0xxuDkqEvwUusBAgIAAQwCAAAAMQAAAAAAAAA=',
|
|
'base64',
|
|
);
|
|
const tx = Transaction.from(wireTransaction);
|
|
|
|
expect(tx).toEqual(expectedTransaction);
|
|
expect(wireTransaction).toEqual(expectedTransaction.serialize());
|
|
});
|
|
|
|
test('populate transaction', () => {
|
|
const recentBlockhash = new PublicKey(1).toString();
|
|
const message = {
|
|
accountKeys: [
|
|
new PublicKey(1).toString(),
|
|
new PublicKey(2).toString(),
|
|
new PublicKey(3).toString(),
|
|
new PublicKey(4).toString(),
|
|
new PublicKey(5).toString(),
|
|
],
|
|
header: {
|
|
numReadonlySignedAccounts: 0,
|
|
numReadonlyUnsignedAccounts: 3,
|
|
numRequiredSignatures: 2,
|
|
},
|
|
instructions: [
|
|
{
|
|
accounts: [1, 2, 3],
|
|
data: bs58.encode(Buffer.alloc(5).fill(9)),
|
|
programIdIndex: 4,
|
|
},
|
|
],
|
|
recentBlockhash,
|
|
};
|
|
|
|
const signatures = [
|
|
bs58.encode(Buffer.alloc(64).fill(1)),
|
|
bs58.encode(Buffer.alloc(64).fill(2)),
|
|
];
|
|
|
|
const transaction = Transaction.populate(new Message(message), signatures);
|
|
expect(transaction.instructions.length).toEqual(1);
|
|
expect(transaction.signatures.length).toEqual(2);
|
|
expect(transaction.recentBlockhash).toEqual(recentBlockhash);
|
|
});
|
|
|
|
test('serialize unsigned transaction', () => {
|
|
const keypair = nacl.sign.keyPair.fromSeed(
|
|
Uint8Array.from(Array(32).fill(8)),
|
|
);
|
|
const sender = new Account(Buffer.from(keypair.secretKey)); // Arbitrary known account
|
|
const recentBlockhash = 'EETubP5AKHgjPAhzPAFcb8BAY1hMH639CWCFTqi3hq1k'; // Arbitrary known recentBlockhash
|
|
const recipient = new PublicKey(
|
|
'J3dxNj7nDRRqRRXuEMynDG57DkZK4jYRuv3Garmb1i99',
|
|
); // Arbitrary known public key
|
|
const transfer = SystemProgram.transfer({
|
|
fromPubkey: sender.publicKey,
|
|
toPubkey: recipient,
|
|
lamports: 49,
|
|
});
|
|
const expectedTransaction = new Transaction({recentBlockhash}).add(transfer);
|
|
|
|
const wireTransactionArray = [
|
|
1,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
1,
|
|
0,
|
|
1,
|
|
3,
|
|
19,
|
|
152,
|
|
246,
|
|
44,
|
|
109,
|
|
26,
|
|
69,
|
|
124,
|
|
81,
|
|
186,
|
|
106,
|
|
75,
|
|
95,
|
|
61,
|
|
189,
|
|
47,
|
|
105,
|
|
252,
|
|
169,
|
|
50,
|
|
22,
|
|
33,
|
|
141,
|
|
200,
|
|
153,
|
|
126,
|
|
65,
|
|
107,
|
|
209,
|
|
125,
|
|
147,
|
|
202,
|
|
253,
|
|
67,
|
|
159,
|
|
204,
|
|
182,
|
|
103,
|
|
39,
|
|
242,
|
|
137,
|
|
197,
|
|
198,
|
|
222,
|
|
59,
|
|
196,
|
|
168,
|
|
254,
|
|
93,
|
|
213,
|
|
215,
|
|
119,
|
|
112,
|
|
188,
|
|
143,
|
|
241,
|
|
92,
|
|
62,
|
|
238,
|
|
220,
|
|
177,
|
|
74,
|
|
243,
|
|
252,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
196,
|
|
154,
|
|
231,
|
|
118,
|
|
3,
|
|
120,
|
|
32,
|
|
84,
|
|
241,
|
|
122,
|
|
157,
|
|
236,
|
|
234,
|
|
67,
|
|
180,
|
|
68,
|
|
235,
|
|
160,
|
|
237,
|
|
177,
|
|
44,
|
|
111,
|
|
29,
|
|
49,
|
|
198,
|
|
224,
|
|
228,
|
|
168,
|
|
75,
|
|
240,
|
|
82,
|
|
235,
|
|
1,
|
|
2,
|
|
2,
|
|
0,
|
|
1,
|
|
12,
|
|
2,
|
|
0,
|
|
0,
|
|
0,
|
|
49,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
0,
|
|
];
|
|
|
|
const wireTransaction = Buffer.from(wireTransactionArray);
|
|
expect(wireTransaction).toEqual(expectedTransaction.serialize());
|
|
expect(Transaction.from(wireTransaction)).toEqual(expectedTransaction);
|
|
expect(Transaction.from(wireTransactionArray)).toEqual(expectedTransaction);
|
|
expect(Transaction.from(Uint8Array.from(wireTransactionArray))).toEqual(
|
|
expectedTransaction,
|
|
);
|
|
});
|
|
|
|
test('get sign data for transaction', () => {
|
|
const from_keypair = nacl.sign.keyPair.fromSeed(
|
|
Uint8Array.from(Array(32).fill(1)),
|
|
);
|
|
const from = new Account(Buffer.from(from_keypair.secretKey));
|
|
const to = new PublicKey(2);
|
|
const recentBlockhash = new PublicKey(3).toBuffer();
|
|
var tx = SystemProgram.transfer({
|
|
fromPubkey: from.publicKey,
|
|
toPubkey: to,
|
|
lamports: 42,
|
|
});
|
|
tx.recentBlockhash = bs58.encode(recentBlockhash);
|
|
const tx_bytes = tx.serializeMessage();
|
|
const signature = nacl.sign.detached(tx_bytes, from.secretKey);
|
|
tx.addSignature(from.publicKey, signature);
|
|
expect(tx.verifySignatures()).toBe(true);
|
|
});
|