Properly mock publickey requests

whiteout-io · Aug 5, 2015 · 8636ba2 · 8636ba2
1 parent aa24881
commit 8636ba2
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/test/integration/email-dao-test.js b/test/integration/email-dao-test.js
@@ -249,6 +249,7 @@ describe('Email DAO integration tests', function() {
             function onCleaned() {
                 userStorage = accountService._accountStore;
                 auth = accountService._auth;
+                emailDao = accountService._emailDao;
 
                 auth.setCredentials({
                     emailAddress: testAccount.user,
@@ -257,19 +258,17 @@ describe('Email DAO integration tests', function() {
                     imap: {} // a preconfigured smtpclient with mocked tcp sockets
                 });
 
+                // stub rest request to key server
+                sinon.stub(emailDao._keychain._publicKeyDao, 'get').returns(resolves(mockKeyPair.publicKey));
+                sinon.stub(emailDao._keychain._publicKeyDao, 'getByUserId').returns(resolves(mockKeyPair.publicKey));
+
                 auth.init().then(function() {
                     accountService.init({
                         emailAddress: testAccount.user
                     }).then(function() {
-                        emailDao = accountService._emailDao;
-
                         // retrieve the pgpbuilder from the emaildao and initialize the pgpmailer with the existing pgpbuilder
                         pgpMailer = new PgpMailer({}, emailDao._pgpbuilder);
 
-                        // stub rest request to key server
-                        sinon.stub(emailDao._keychain._publicKeyDao, 'get').returns(resolves(mockKeyPair.publicKey));
-                        sinon.stub(emailDao._keychain._publicKeyDao, 'getByUserId').returns(resolves(mockKeyPair.publicKey));
-
                         emailDao.unlock({
                             passphrase: testAccount.pass,
                             keypair: mockKeyPair