mail/test/unit/oauth-test.js

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

    var OAuth = require('js/util/oauth'),
        RestDAO = require('js/dao/rest-dao'),
        expect = chai.expect;

    describe('OAuth unit tests', function() {
        var oauth, googleApiStub, identityStub, getPlatformInfoStub, removeCachedStub,
            testEmail = 'test@example.com';

        beforeEach(function() {
            googleApiStub = sinon.createStubInstance(RestDAO);
            oauth = new OAuth(googleApiStub);

            window.chrome = window.chrome || {};

            window.chrome.identity = window.chrome.identity || {};
            if (typeof window.chrome.identity.getAuthToken !== 'function') {
                window.chrome.identity.getAuthToken = function() {};
            }
            identityStub = sinon.stub(window.chrome.identity, 'getAuthToken');

            if (typeof window.chrome.identity.removeCachedAuthToken !== 'function') {
                window.chrome.identity.removeCachedAuthToken = function() {};
            }
            removeCachedStub = sinon.stub(window.chrome.identity, 'removeCachedAuthToken');

            window.chrome.runtime = window.chrome.runtime || {};
            if (typeof window.chrome.runtime.getPlatformInfo !== 'function') {
                window.chrome.runtime.getPlatformInfo = function() {};
            }
            getPlatformInfoStub = sinon.stub(window.chrome.runtime, 'getPlatformInfo');
        });

        afterEach(function() {
            identityStub.restore();
            getPlatformInfoStub.restore();
            removeCachedStub.restore();
        });

        describe('isSupported', function() {
            it('should work', function() {
                expect(oauth.isSupported()).to.be.true;
            });
        });

        describe('refreshToken', function() {
            var getOAuthTokenStub;

            beforeEach(function() {
                getOAuthTokenStub = sinon.stub(oauth, 'getOAuthToken');
            });
            afterEach(function() {
                getOAuthTokenStub.restore();
            });

            it('should work', function() {
                removeCachedStub.withArgs({
                    token: 'oldToken'
                }).yields();

                getOAuthTokenStub.withArgs(testEmail).yields();

                oauth.refreshToken({
                    oldToken: 'oldToken',
                    emailAddress: testEmail
                }, function(err) {
                    expect(err).to.not.exist;
                    expect(removeCachedStub.calledOnce).to.be.true;
                    expect(getOAuthTokenStub.calledOnce).to.be.true;
                });
            });

            it('should work without email', function() {
                removeCachedStub.withArgs({
                    token: 'oldToken'
                }).yields();

                getOAuthTokenStub.withArgs(undefined).yields();

                oauth.refreshToken({
                    oldToken: 'oldToken',
                }, function(err) {
                    expect(err).to.not.exist;
                    expect(removeCachedStub.calledOnce).to.be.true;
                    expect(getOAuthTokenStub.calledOnce).to.be.true;
                    expect(getOAuthTokenStub.calledWith(undefined)).to.be.true;
                });
            });

            it('should fail without all options', function() {
                oauth.refreshToken({
                    emailAddress: testEmail
                }, function(err) {
                    expect(err).to.exist;
                    expect(removeCachedStub.called).to.be.false;
                    expect(getOAuthTokenStub.called).to.be.false;
                });
            });
        });

        describe('getOAuthToken', function() {
            it('should work for empty emailAddress', function(done) {
                getPlatformInfoStub.yields({
                    os: 'android'
                });
                identityStub.withArgs({
                    interactive: true
                }).yields('token');

                oauth.getOAuthToken(undefined, function(err, token) {
                    expect(err).to.not.exist;
                    expect(token).to.equal('token');
                    done();
                });
            });

            it('should work on android app', function(done) {
                getPlatformInfoStub.yields({
                    os: 'android'
                });
                identityStub.withArgs({
                    interactive: true,
                    accountHint: testEmail
                }).yields('token');

                oauth.getOAuthToken(testEmail, function(err, token) {
                    expect(err).to.not.exist;
                    expect(token).to.equal('token');
                    done();
                });
            });

            it('should work on desktop chrome', function(done) {
                getPlatformInfoStub.yields({
                    os: 'mac'
                });
                identityStub.withArgs({
                    interactive: true
                }).yields('token');

                oauth.getOAuthToken(testEmail, function(err, token) {
                    expect(err).to.not.exist;
                    expect(token).to.equal('token');
                    done();
                });
            });

            it('should fail', function(done) {
                getPlatformInfoStub.yields({
                    os: 'android'
                });
                identityStub.yields();

                oauth.getOAuthToken(testEmail, function(err, token) {
                    expect(err).to.exist;
                    expect(token).to.not.exist;
                    done();
                });
            });
        });

        describe('queryEmailAddress', function() {
            it('should work', function(done) {
                googleApiStub.get.withArgs({
                    uri: '/oauth2/v3/userinfo?access_token=token'
                }).yields(null, {
                    email: 'asdf@example.com'
                });

                oauth.queryEmailAddress('token', function(err, emailAddress) {
                    expect(err).to.not.exist;
                    expect(emailAddress).to.equal('asdf@example.com');
                    done();
                });
            });

            it('should fail due to invalid token', function(done) {
                oauth.queryEmailAddress('', function(err, emailAddress) {
                    expect(err).to.exist;
                    expect(emailAddress).to.not.exist;
                    done();
                });
            });

            it('should fail due to error in rest api', function(done) {
                googleApiStub.get.withArgs({
                    uri: '/oauth2/v3/userinfo?access_token=token'
                }).yields(new Error());

                oauth.queryEmailAddress('token', function(err, emailAddress) {
                    expect(err).to.exist;
                    expect(emailAddress).to.not.exist;
                    done();
                });
            });
        });

    });
});