diff --git a/test/double-reveal-test.js b/test/double-reveal-test.js deleted file mode 100644 index 466d130d20..0000000000 --- a/test/double-reveal-test.js +++ /dev/null @@ -1,103 +0,0 @@ -/* eslint-env mocha */ -/* eslint prefer-arrow-callback: "off" */ - -'use strict'; - -const assert = require('bsert'); -const Network = require('../lib/protocol/network'); -const FullNode = require('../lib/node/fullnode'); -const Address = require('../lib/primitives/address'); -const rules = require('../lib/covenants/rules'); - -const network = Network.get('regtest'); - -const node = new FullNode({ - memory: true, - network: 'regtest', - plugins: [require('../lib/wallet/plugin')] -}); - -const {wdb} = node.require('walletdb'); -const name = rules.grindName(10, 20, network); - -let wallet, alice, bob, aliceMiner, bobMiner; - -async function mineBlocks(n, addr) { - addr = addr ? addr : new Address().toString('regtest'); - for (let i = 0; i < n; i++) { - const block = await node.miner.mineBlock(null, addr); - await node.chain.add(block); - } -} - -describe('One wallet, two accounts, one name', function() { - before(async () => { - await node.open(); - - wallet = await wdb.create(); - - alice = await wallet.createAccount({name: 'alice'}); - bob = await wallet.createAccount({name: 'bob'}); - - aliceMiner = await alice.receiveAddress(); - bobMiner = await bob.receiveAddress(); - }); - - after(async () => { - await node.close(); - }); - - it('should fund both accounts', async () => { - await mineBlocks(10, aliceMiner); - await mineBlocks(10, bobMiner); - - // Wallet rescan is an effective way to ensure that - // wallet and chain are synced before proceeding. - await wdb.rescan(0); - - const aliceBal = await wallet.getBalance('alice'); - const bobBal = await wallet.getBalance('bob'); - assert(aliceBal.confirmed === 2000 * 10 * 1e6); - assert(bobBal.confirmed === 2000 * 10 * 1e6); - }); - - it('should open an auction and proceed to REVEAL phase', async () => { - await wallet.sendOpen(name, false, {account: 'alice'}); - await mineBlocks(network.names.treeInterval + 2); - let ns = await node.chain.db.getNameStateByName(name); - assert(ns.isBidding(node.chain.height, network)); - - await wdb.rescan(0); - - await wallet.sendBid(name, 100000, 200000, {account: 'alice'}); - await wallet.sendBid(name, 50000, 200000, {account: 'bob'}); - await mineBlocks(network.names.biddingPeriod); - ns = await node.chain.db.getNameStateByName(name); - assert(ns.isReveal(node.chain.height, network)); - - await wdb.rescan(0); - - const walletBids = await wallet.getBidsByName(name); - assert.strictEqual(walletBids.length, 2); - - for (const bid of walletBids) - assert(bid.own); - }); - - it('should send REVEAL from one account at a time', async () => { - const tx1 = await wallet.sendReveal(name, {account: 'alice'}); - assert(tx1); - - const tx2 = await wallet.sendReveal(name, {account: 'bob'}); - assert(tx2); - - // Reset for next test - await wallet.abandon(tx1.hash()); - await wallet.abandon(tx2.hash()); - }); - - it('should send REVEAL from all accounts', async () => { - const tx = await wallet.sendRevealAll(); - assert(tx); - }); -}); diff --git a/test/wallet-accounts-reveal-test.js b/test/wallet-accounts-reveal-test.js new file mode 100644 index 0000000000..4d15a58349 --- /dev/null +++ b/test/wallet-accounts-reveal-test.js @@ -0,0 +1,196 @@ +/* eslint-env mocha */ +/* eslint prefer-arrow-callback: "off" */ + +'use strict'; + +const assert = require('bsert'); +const Network = require('../lib/protocol/network'); +const FullNode = require('../lib/node/fullnode'); +const Address = require('../lib/primitives/address'); +const rules = require('../lib/covenants/rules'); +const {WalletClient} = require('hs-client'); + +const network = Network.get('regtest'); + +const node = new FullNode({ + memory: true, + network: 'regtest', + plugins: [require('../lib/wallet/plugin')] +}); + +// Prevent mempool from sending duplicate TXs back to the walletDB and txdb. +// This will prevent a race condition when we need to remove spent (but +// unconfirmed) outputs from the wallet so they can be reused in other tests. +node.mempool.emit = () => {}; + +const wclient = new WalletClient({ + port: network.walletPort +}); + +const {wdb} = node.require('walletdb'); + +const name = rules.grindName(5, 1, network); +let wallet, alice, bob, aliceMiner, bobMiner; + +async function mineBlocks(n, addr) { + addr = addr ? addr : new Address().toString('regtest'); + for (let i = 0; i < n; i++) { + const block = await node.miner.mineBlock(null, addr); + await node.chain.add(block); + } +} + +describe('One wallet, two accounts, one name', function() { + before(async () => { + await node.open(); + await wclient.open(); + + wallet = await wdb.create(); + + // We'll use an account number for alice and a string for bob + // to ensure that both types work as options. + alice = await wallet.getAccount(0); + bob = await wallet.createAccount({name: 'bob'}); + + aliceMiner = await alice.receiveAddress(); + bobMiner = await bob.receiveAddress(); + }); + + after(async () => { + await wclient.close(); + await node.close(); + }); + + it('should fund both accounts', async () => { + await mineBlocks(2, aliceMiner); + await mineBlocks(2, bobMiner); + + // Wallet rescan is an effective way to ensure that + // wallet and chain are synced before proceeding. + await wdb.rescan(0); + + const aliceBal = await wallet.getBalance(0); + const bobBal = await wallet.getBalance('bob'); + assert(aliceBal.confirmed === 2000 * 2 * 1e6); + assert(bobBal.confirmed === 2000 * 2 * 1e6); + }); + + it('should open an auction and proceed to REVEAL phase', async () => { + await wallet.sendOpen(name, false, {account: 0}); + await mineBlocks(network.names.treeInterval + 2); + let ns = await node.chain.db.getNameStateByName(name); + assert(ns.isBidding(node.chain.height, network)); + + await wdb.rescan(0); + + await wallet.sendBid(name, 100000, 200000, {account: 0}); + await wallet.sendBid(name, 50000, 200000, {account: 'bob'}); + await mineBlocks(network.names.biddingPeriod); + ns = await node.chain.db.getNameStateByName(name); + assert(ns.isReveal(node.chain.height, network)); + + await wdb.rescan(0); + + const walletBids = await wallet.getBidsByName(name); + assert.strictEqual(walletBids.length, 2); + + for (const bid of walletBids) + assert(bid.own); + + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from one account at a time -- LIBRARY', async () => { + const tx1 = await wallet.sendReveal(name, {account: 0}); + assert(tx1); + + const tx2 = await wallet.sendReveal(name, {account: 'bob'}); + assert(tx2); + + // Reset for next test + await wallet.abandon(tx1.hash()); + await wallet.abandon(tx2.hash()); + + assert.strictEqual(node.mempool.map.size, 2); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from all accounts -- LIBRARY', async () => { + const tx = await wallet.sendRevealAll(); + assert(tx); + + // Reset for next test + await wallet.abandon(tx.hash()); + + assert.strictEqual(node.mempool.map.size, 1); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from one account at a time -- HTTP', async () => { + const tx1 = await wclient.post(`/wallet/${wallet.id}/reveal`, { + name: name, + account: 'default' + }); + assert(tx1); + + const tx2 = await wclient.post(`/wallet/${wallet.id}/reveal`, { + name: name, + account: 'bob' + }); + assert(tx2); + + // Reset for next test + await wallet.abandon(Buffer.from(tx1.hash, 'hex')); + await wallet.abandon(Buffer.from(tx2.hash, 'hex')); + + assert.strictEqual(node.mempool.map.size, 2); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from all accounts -- HTTP', async () => { + const tx = await wclient.post(`/wallet/${wallet.id}/reveal`, { + name: name + }); + assert(tx); + + // Reset for next test + await wallet.abandon(Buffer.from(tx.hash, 'hex')); + + assert.strictEqual(node.mempool.map.size, 1); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from one account at a time -- RPC', async () => { + await wclient.execute('selectwallet', [wallet.id]); + + const tx1 = await wclient.execute('sendreveal', [name, 'default']); + assert(tx1); + + const tx2 = await wclient.execute('sendreveal', [name, 'bob']); + assert(tx2); + + // Reset for next test + await wallet.abandon(Buffer.from(tx1.hash, 'hex')); + await wallet.abandon(Buffer.from(tx2.hash, 'hex')); + + assert.strictEqual(node.mempool.map.size, 2); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); + + it('should send REVEAL from all accounts -- RPC', async () => { + const tx = await wclient.execute('sendreveal', [name]); + assert(tx); + + // Reset for next test + await wallet.abandon(Buffer.from(tx.hash, 'hex')); + + assert.strictEqual(node.mempool.map.size, 1); + await node.mempool.reset(); + assert.strictEqual(node.mempool.map.size, 0); + }); +});