import React from 'react' import assert from 'assert' import sinon from 'sinon' import { mountWithRouter } from '../../../../../test/lib/render-helpers' import Lock from '../index' describe('Lock', () => { it('replaces history with default route when isUnlocked false', () => { const props = { isUnlocked: false, history: { replace: sinon.spy(), }, } mountWithRouter( ) assert.equal(props.history.replace.getCall(0).args[0], '/') }) it('locks and pushes history with default route when isUnlocked true', (done) => { const props = { isUnlocked: true, lockMetamask: sinon.stub(), history: { push: sinon.spy(), }, } props.lockMetamask.resolves() mountWithRouter( ) assert(props.lockMetamask.calledOnce) setImmediate(() => { assert.equal(props.history.push.getCall(0).args[0], '/') done() }) }) })