From ffbc23684c5e7a86ea759273795e28459ddc8ef5 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Wed, 26 Sep 2018 09:04:04 -0700
Subject: [PATCH] Improve budget-program tests

---
 web3.js/test/budget-program.test.js | 35 ++++++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 3 deletions(-)

diff --git a/web3.js/test/budget-program.test.js b/web3.js/test/budget-program.test.js
index 1ee94e1988..e917b393e2 100644
--- a/web3.js/test/budget-program.test.js
+++ b/web3.js/test/budget-program.test.js
@@ -16,7 +16,8 @@ test('pay', () => {
     123,
   );
   console.log('Pay:', transaction);
-  // TODO: Validate transaction contents
+  expect(transaction.keys).toHaveLength(2);
+  // TODO: Validate transaction contents more
 
   transaction = BudgetProgram.pay(
     from.publicKey,
@@ -26,7 +27,8 @@ test('pay', () => {
     BudgetProgram.signatureCondition(from.publicKey),
   );
   console.log('After:', transaction);
-  // TODO: Validate transaction contents
+  expect(transaction.keys).toHaveLength(3);
+  // TODO: Validate transaction contents more
 
   transaction = BudgetProgram.pay(
     from.publicKey,
@@ -37,6 +39,33 @@ test('pay', () => {
     BudgetProgram.timestampCondition(from.publicKey, new Date()),
   );
   console.log('Or:', transaction);
-  // TODO: Validate transaction contents
+  expect(transaction.keys).toHaveLength(3);
+  // TODO: Validate transaction contents more
+});
+
+test('apply', () => {
+  const from = new Account();
+  const program = new Account();
+  const to = new Account();
+  let transaction;
+
+  transaction = BudgetProgram.applyTimestamp(
+    from.publicKey,
+    program.publicKey,
+    to.publicKey,
+    new Date(),
+  );
+  console.log('applyTimestamp:', transaction);
+  expect(transaction.keys).toHaveLength(3);
+  // TODO: Validate transaction contents more
+
+  transaction = BudgetProgram.applySignature(
+    from.publicKey,
+    program.publicKey,
+    to.publicKey,
+  );
+  console.log('applySignature:', transaction);
+  expect(transaction.keys).toHaveLength(3);
+  // TODO: Validate transaction contents more
 });