mail/test/unit/write-ctrl-test.js

define(function(require) {
    'use strict';

    var expect = chai.expect,
        angular = require('angular'),
        mocks = require('angularMocks'),
        WriteCtrl = require('js/controller/write'),
        EmailDAO = require('js/dao/email-dao'),
        OutboxBO = require('js/bo/outbox'),
        KeychainDAO = require('js/dao/keychain-dao'),
        appController = require('js/app-controller');

    describe('Write controller unit test', function() {
        var ctrl, scope,
            origEmailDao, origOutbox, origKeychain,
            emailDaoMock, keychainMock, outboxMock, emailAddress, realname;

        beforeEach(function() {
            // the app controller is a singleton, we need to remember the
            // outbox and email dao to restore it after the tests
            origEmailDao = appController._emailDao;
            origOutbox = appController._outboxBo;
            origKeychain = appController._keychain;

            outboxMock = sinon.createStubInstance(OutboxBO);
            appController._outboxBo = outboxMock;

            emailDaoMock = sinon.createStubInstance(EmailDAO);
            appController._emailDao = emailDaoMock;

            emailAddress = 'fred@foo.com';
            realname = 'Fred Foo';
            emailDaoMock._account = {
                emailAddress: emailAddress,
                realname: realname
            };

            keychainMock = sinon.createStubInstance(KeychainDAO);
            appController._keychain = keychainMock;

            angular.module('writetest', []);
            mocks.module('writetest');
            mocks.inject(function($rootScope, $controller) {
                scope = $rootScope.$new();
                scope.state = {};
                ctrl = $controller(WriteCtrl, {
                    $scope: scope
                });
            });
        });

        afterEach(function() {
            // restore the app controller
            appController._emailDao = origEmailDao;
            appController._outboxBo = origOutbox;
            appController._keychain = origKeychain;
        });

        describe('scope variables', function() {
            it('should be set correctly', function() {
                expect(scope.state.writer).to.exist;
                expect(scope.state.lightbox).to.be.undefined;
                expect(scope.state.writer.write).to.exist;
                expect(scope.state.writer.close).to.exist;
                expect(scope.verify).to.exist;
                expect(scope.checkSendStatus).to.exist;
                expect(scope.updatePreview).to.exist;
                expect(scope.sendToOutbox).to.exist;
                expect(scope.tagStyle).to.exist;
                expect(scope.lookupAddressBook).to.exist;
            });
        });

        describe('close', function() {
            it('should close the writer', function() {
                scope.state.lightbox = 'write';

                scope.state.writer.close();

                expect(scope.state.lightbox).to.be.undefined;
            });
        });

        describe('write', function() {
            it('should prepare write view', function() {
                var verifyMock = sinon.stub(scope, 'verify');

                scope.state.writer.write();

                expect(scope.writerTitle).to.equal('New email');
                expect(scope.to).to.deep.equal([]);
                expect(scope.subject).to.equal('');
                expect(scope.body).to.equal('');
                expect(scope.ciphertextPreview).to.equal(undefined);
                expect(verifyMock.calledOnce).to.be.true;

                scope.verify.restore();
            });

            it('should prefill write view for response', function() {
                var verifyMock = sinon.stub(scope, 'verify'),
                    address = 'pity@dafool',
                    subject = 'Ermahgerd!',
                    body = 'so much body!',
                    re = {
                        id: 'abc',
                        from: [{
                            address: address
                        }],
                        subject: subject,
                        sentDate: new Date(),
                        body: body,
                        references: ['ghi', 'def']
                    };

                scope.sendBtnSecure = true;

                scope.state.writer.write(re);

                expect(scope.writerTitle).to.equal('Reply');
                expect(scope.to).to.deep.equal([{
                    address: address,
                }]);
                expect(scope.subject).to.equal('Re: ' + subject);
                expect(scope.body).to.contain(body);
                expect(scope.references).to.deep.equal(['ghi', 'def', 'abc']);
                expect(scope.ciphertextPreview).to.not.be.empty;
                expect(verifyMock.called).to.be.true;

                scope.verify.restore();
            });

            it('should prefill write view for forward', function() {
                var verifyMock = sinon.stub(scope, 'verify'),
                    address = 'pity@dafool',
                    subject = 'Ermahgerd!',
                    body = 'so much body!',
                    re = {
                        from: [{
                            address: address
                        }],
                        to: [{
                            address: address
                        }],
                        subject: subject,
                        sentDate: new Date(),
                        body: body,
                        attachments: [{}]
                    };

                scope.sendBtnSecure = false;

                scope.state.writer.write(re, null, true);

                expect(scope.writerTitle).to.equal('Forward');
                expect(scope.to).to.deep.equal([]);
                expect(scope.subject).to.equal('Fwd: ' + subject);
                expect(scope.body).to.contain(body);
                expect(scope.ciphertextPreview).to.be.undefined;
                expect(verifyMock.called).to.be.true;
                expect(scope.attachments).to.not.equal(re.attachments); // not the same reference
                expect(scope.attachments).to.deep.equal(re.attachments); // but the same content

                scope.verify.restore();
            });

        });

        describe('verify', function() {
            var checkSendStatusMock;

            beforeEach(function() {
                checkSendStatusMock = sinon.stub(scope, 'checkSendStatus');
            });

            afterEach(function() {
                scope.checkSendStatus.restore();
            });

            it('should do nothing if recipient is not provided', function() {
                scope.verify(undefined);
            });

            it('should not work for invalid email addresses', function() {
                var recipient = {
                    address: ''
                };

                scope.verify(recipient);

                expect(recipient.key).to.be.undefined;
                expect(recipient.secure).to.be.undefined;
                expect(scope.checkSendStatus.callCount).to.equal(2);
                expect(keychainMock.getReceiverPublicKey.called).to.be.false;
            });

            it('should not work for error in keychain', function(done) {
                var recipient = {
                    address: 'asds@example.com'
                };

                keychainMock.refreshKeyForUserId.withArgs(recipient.address).yields({
                    errMsg: '404 not found yadda yadda'
                });

                scope.onError = function() {
                    expect(recipient.key).to.be.undefined;
                    expect(recipient.secure).to.be.false;
                    expect(scope.checkSendStatus.callCount).to.equal(1);
                    expect(keychainMock.refreshKeyForUserId.calledOnce).to.be.true;
                    done();
                };

                scope.verify(recipient);
            });

            it('should work for main userId', function(done) {
                var recipient = {
                    address: 'asdf@example.com'
                };

                keychainMock.refreshKeyForUserId.withArgs(recipient.address).yields(null, {
                    userId: 'asdf@example.com'
                });

                scope.$digest = function() {
                    expect(recipient.key).to.deep.equal({
                        userId: 'asdf@example.com'
                    });
                    expect(recipient.secure).to.be.true;
                    expect(scope.checkSendStatus.callCount).to.equal(2);
                    expect(keychainMock.refreshKeyForUserId.calledOnce).to.be.true;
                    done();
                };

                scope.verify(recipient);
            });

            it('should work for secondary userId', function(done) {
                var recipient = {
                    address: 'asdf@example.com'
                };
                var key = {
                    userId: 'qwer@example.com',
                    userIds: [{
                        emailAddress: 'asdf@example.com'
                    }]
                };

                keychainMock.refreshKeyForUserId.withArgs(recipient.address).yields(null, key);

                scope.$digest = function() {
                    expect(recipient.key).to.deep.equal(key);
                    expect(recipient.secure).to.be.true;
                    expect(scope.checkSendStatus.callCount).to.equal(2);
                    expect(keychainMock.refreshKeyForUserId.calledOnce).to.be.true;
                    done();
                };

                scope.verify(recipient);
            });
        });

        describe('checkSendStatus', function() {
            beforeEach(function() {
                scope.state.writer.write();
            });

            afterEach(function() {});

            it('should not be able to send with no recipients', function() {
                scope.checkSendStatus();

                expect(scope.okToSend).to.be.false;
                expect(scope.sendBtnText).to.be.undefined;
                expect(scope.sendBtnSecure).to.be.undefined;
            });

            it('should be able to send plaintext', function() {
                scope.to = [{
                    address: 'asdf@asdf.de'
                }];
                scope.checkSendStatus();

                expect(scope.okToSend).to.be.true;
                expect(scope.sendBtnText).to.equal('Send');
                expect(scope.sendBtnSecure).to.be.false;
            });

            it('should send plaintext if one receiver is not secure', function() {
                scope.to = [{
                    address: 'asdf@asdf.de',
                    secure: true
                }, {
                    address: 'asdf@asdfg.de'
                }];
                scope.checkSendStatus();

                expect(scope.okToSend).to.be.true;
                expect(scope.sendBtnText).to.equal('Send');
                expect(scope.sendBtnSecure).to.be.false;
            });

            it('should be able to send securely to multiple recipients', function() {
                scope.to = [{
                    address: 'asdf@asdf.de',
                    secure: true
                }, {
                    address: 'asdf@asdfg.de',
                    secure: true
                }];
                scope.checkSendStatus();

                expect(scope.okToSend).to.be.true;
                expect(scope.sendBtnText).to.equal('Send securely');
                expect(scope.sendBtnSecure).to.be.true;
            });
        });

        describe('send to outbox', function() {
            it('should work', function() {
                scope.to = [{
                    address: 'pity@dafool'
                }];
                scope.cc = [];
                scope.bcc = [];
                scope.subject = 'Ermahgerd!';
                scope.body = 'wow. much body! very text!';
                scope.attachments = [];
                scope.state.nav = {
                    currentFolder: 'currentFolder'
                };

                scope.replyTo = {};

                outboxMock.put.withArgs(sinon.match(function(mail) {
                    expect(mail.from).to.deep.equal([{
                        address: emailAddress,
                        name: realname
                    }]);
                    expect(mail.to).to.deep.equal(scope.to);
                    expect(mail.cc).to.deep.equal(scope.cc);
                    expect(mail.bcc).to.deep.equal(scope.bcc);
                    expect(mail.body).to.contain(scope.body);
                    expect(mail.subject).to.equal(scope.subject);
                    expect(mail.attachments).to.be.empty;
                    expect(mail.sentDate).to.exist;


                    return true;
                })).yields();
                emailDaoMock.setFlags.yields();

                scope.onError = function(err) {
                    expect(err).to.not.exist;
                };

                scope.sendToOutbox();

                expect(outboxMock.put.calledOnce).to.be.true;
                expect(emailDaoMock.setFlags.calledOnce).to.be.true;
                expect(scope.state.lightbox).to.be.undefined;
                expect(scope.replyTo.answered).to.be.true;
            });
        });

        describe('lookupAddressBook', function() {
            it('should work', function(done) {
                keychainMock.listLocalPublicKeys.yields(null, [{
                    userId: 'test@asdf.com',
                    publicKey: 'KEY'
                }]);

                var result = scope.lookupAddressBook('test');

                result.then(function(response) {
                    expect(response).to.deep.equal([{
                        address: 'test@asdf.com'
                    }]);
                    done();
                });
                scope.$digest();
            });

            it('should work with cache', function(done) {
                scope.addressBookCache = [{
                    address: 'test@asdf.com'
                }, {
                    address: 'tes@asdf.com'
                }];

                var result = scope.lookupAddressBook('test');

                result.then(function(response) {
                    expect(response).to.deep.equal([{
                        address: 'test@asdf.com'
                    }]);
                    done();
                });
                scope.$digest();
            });
        });

    });
});