ethers.js/tests/test-secret-storage.js

'use strict'
var Wallet = require('../index.js');

var fs = require('fs');

module.exports = function(test) {

    var crowdsale = [
        {
            address: '0x2e326fA404Fc3661de4F4361776ed9bBABDC26E3',
            data: fs.readFileSync('./test-wallets/wallet-test-encseed-foo.json').toString(),
            password: 'foo',
            privateKey: '0xcf367fc32bf789b3339c6664af4a12263e9db0e0eb70f247da1d1165e150c487',
            type: 'crowdsale'
        },
        {
            address: '0x0b88d4b324ec24C8c078551e6e5075547157E5b6',
            data: fs.readFileSync('./test-wallets/wallet-test-encseed-no-password.json').toString(),
            password: '',
            privateKey: '0xd4375d2a931db84ea8825b69a3128913597744d9236cacec675cc18e1bda4446',
            type: 'crowdsale'
        }
    ];

    var geth = [
        {
            address: '0x88a5C2d9919e46F883EB62F7b8Dd9d0CC45bc290',
            data: fs.readFileSync('./test-wallets/wallet-test-geth-foo.json').toString(),
            password: 'foo',
            privateKey: '0xf03e581353c794928373fb0893bc731aefc4c4e234e643f3a46998b03cd4d7c5',
            type: 'version3'
        },
        {
            address: '0x4A9cf99357F5789251a8D7FaD5b86D0F31EEB938',
            data: fs.readFileSync('./test-wallets/wallet-test-geth-no-password.json').toString(),
            password: '',
            privateKey: '0xa016182717223d01f776149ec0b4a217d0e9930cad263f205427c6d3cd5560e7',
            type: 'version3'
        },
    ];

    test.expect((crowdsale.length * 4) + (geth.length * 4) + 2);


    // Test crowdsale private key decryption
    crowdsale.forEach(function(testcase) {

        // Check wallet type detection
        test.ok(Wallet.isCrowdsaleWallet(testcase.data), 'wrong wallet type detected');
        test.ok(!Wallet.isValidWallet(testcase.data), 'wrong wallet type detected');

        var wallet = Wallet.decryptCrowdsale(testcase.data, testcase.password);

        test.equal(wallet.privateKey, testcase.privateKey, 'wrong private key');
        test.equal(wallet.address, testcase.address, 'wrong address');
    });

    // Private keys are asynchronous, so we do this to trigger the done
    // only after all testcases have returned
    var expecting = geth.length + 1;
    function checkAsync() {
        expecting--;
        if (expecting === 0) { test.done(); }
    }

    geth.forEach(function(testcase) {
        // Check wallet type detection
        test.ok(Wallet.isValidWallet(testcase.data), 'wrong wallet type detected');
        test.ok(!Wallet.isCrowdsaleWallet(testcase.data), 'wrong wallet type detected');

        // Test private key decryption
        var password = new Buffer(testcase.password, 'utf8');
        Wallet.decrypt(testcase.data, password, function(error, wallet, progress) {
            if (error) {
                console.log(error);
                test.ok(false, 'callback returned error - ' + error.message);
                checkAsync();

            } else if (wallet) {
                test.equals(wallet.privateKey, testcase.privateKey, 'wrong private key')
                test.equals(wallet.address, testcase.address, 'wrong address');
                checkAsync();
            }
        });
    });

    var privateKey = new Buffer(32);
    privateKey.fill(0x42);

    var password = new Buffer("foo", 'utf8');

    (new Wallet(privateKey)).encrypt(password, {
        scrypt: { N: (1 << 10), r: 4, p: 2 },
        iv:   '0xdeadbeef1deadbeef2deadbeef301234',
        salt: '0xabcd1abcd2abcd3abcd4abcd5abcd6ef',
        uuid: '0x01234567890123456789012345678901',
    }, function(error, json, progress) {
        if (error) {
            test.ok(false, 'callback returned error - ' + error.message);
            checkAsync();

        } else if (json) {
            var jsonWallet = fs.readFileSync('./test-wallets/wallet-test-life.json').toString();
            test.equal(json, jsonWallet, 'failed to encrypt wallet');
            Wallet.decrypt(json, password, function(error, wallet, progress) {
                if (error) {
                    test.ok(false, 'callback returned error - ' + error.message);
                    checkAsync();
                } else if (wallet) {
                    test.equal(wallet.privateKey, '0x' + privateKey.toString('hex'), 'decryption failed');
                    checkAsync();
                }
            });
        }
    });
}

module.exports.testSelf = module.exports;