test/unit/test-crypto.js - main - githubmirror/amphtml

test/unit/test-crypto.js
194 строки6 KB
Daniel Rozenberg
❄️ Unskip some tests that no longer fail (#39329)
09 авг 2023, 22:33
09 авг 2023, 22:331986291
import {Services} from '#service';
import {installDocService} from '#service/ampdoc-impl';
import {Crypto} from '#service/crypto-impl';
import {installExtensionsService} from '#service/extensions-impl';
import {Platform} from '#service/platform-impl';

import {FakePerformance} from '#testing/fake-dom';

import {installCryptoPolyfill} from '../../extensions/amp-crypto-polyfill/0.1/amp-crypto-polyfill';

describes.realWin('crypto-impl', {}, (env) => {
  let win;
  let crypto;

  beforeEach(() => {
    win = env.win;
  });

  function uint8Array(array) {
    const uint8Array = new Uint8Array(array.length);
    for (let i = 0; i < array.length; i++) {
      uint8Array[i] = array[i];
    }
    return uint8Array;
  }

  function testSuite(description, win, expectedError) {
    describe(description, () => {
      beforeEach(() => {
        crypto = createCrypto(win || env.win);
      });

      it('should hash "abc" in sha384', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return crypto.sha384('abc').then((buffer) => {
          expect(buffer.length).to.equal(48);
          expect(buffer[0]).to.equal(203);
          expect(buffer[1]).to.equal(0);
          expect(buffer[47]).to.equal(167);
        });
      });

      it('should hash [1,2,3] in sha384', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return crypto.sha384(uint8Array([1, 2, 3])).then((buffer) => {
          expect(buffer.length).to.equal(48);
          expect(buffer[0]).to.equal(134);
          expect(buffer[1]).to.equal(34);
          expect(buffer[47]).to.equal(246);
        });
      });

      it('should hash "abc" in sha384Base64', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return expect(crypto.sha384Base64('abc')).to.eventually.equal(
          'ywB1P0WjXou1oD1pmsZQBycsMqsO3tFjGotgWkP_W-2AhgcroefMI1i67KE0yCWn'
        );
      });

      it('should hash "foobar" in sha384Base64', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return expect(crypto.sha384Base64('foobar')).to.eventually.equal(
          'PJww2fZl501RXIQpYNSkUcg6ASX9Pec5LXs3IxrxDHLqWK7fzfiaV2W_kCr5Ps8G'
        );
      });

      it('should hash [1,2,3] in sha384', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return expect(
          crypto.sha384Base64(uint8Array([1, 2, 3]))
        ).to.eventually.equal(
          'hiKdxtL_vqxzgHRBVKpwApHAZDUqDb3H' +
            'e57T8sjh2sTcMlhn053f8dJim3o5PUf2'
        );
      });

      it('should throw when input contains chars out of range [0,255]', () => {
        allowConsoleError(() => {
          expect(() => crypto.sha384('abc今')).to.throw();
          expect(() => crypto.sha384Base64('abc今')).to.throw();
          expect(() => crypto.uniform('abc今')).to.throw();
        });
      });

      it('should hash "abc" to uniform number', () => {
        if (expectedError) {
          expectAsyncConsoleError(expectedError);
        }
        return crypto.uniform('abc').then((result) => {
          expect(result.toFixed(6)).to.equal('0.792976');
        });
      });
    });
  }

  function createCrypto(win) {
    if (!win.document) {
      win.document = env.win.document;
    }
    win.performance = new FakePerformance(win);
    installDocService(win, /* isSingleDoc */ true);
    installExtensionsService(win);
    const extensions = Services.extensionsFor(win);
    env.sandbox
      .stub(extensions, 'preloadExtension')
      .callsFake((extensionId) => {
        expect(extensionId).to.equal('amp-crypto-polyfill');
        installCryptoPolyfill(win);
        return Promise.resolve();
      });

    return new Crypto(win);
  }

  function isModernChrome() {
    const platform = new Platform(window);
    return platform.isChrome() && platform.getMajorVersion() >= 37;
  }

  testSuite('with native crypto API');
  testSuite('with crypto lib', {...win, crypto: null});
  testSuite('with native crypto API rejects', {
    ...win,
    crypto: {
      subtle: {
        digest: () => Promise.reject('Operation not supported'),
      },
    },
  });
  testSuite(
    'with native crypto API throws',
    {
      ...win,
      crypto: {
        subtle: {
          digest: () => {
            throw new Error();
          },
        },
      },
    },
    '[Crypto] SubtleCrypto failed, fallback to closure lib. Error'
  );

  it('native API result should exactly equal to crypto lib result', () => {
    const fakeWin = {...win, crypto: null};
    return Promise.all([
      createCrypto(win).sha384('abc'),
      createCrypto(fakeWin).sha384('abc'),
    ]).then((results) => {
      expect(results[0]).to.jsonEqual(results[1]);
    });
  });

  // Run tests below only on browsers that we're confident about the existence
  // of native Crypto API.
  if (isModernChrome()) {
    it(
      'should not load closure lib when native API is available ' +
        '(string input)',
      () => {
        return new Crypto(win).sha384Base64('abc').then((hash) => {
          expect(hash).to.equal(
            'ywB1P0WjXou1oD1pmsZQBycsMqsO3tFjGotgWkP_W-2AhgcroefMI1i67KE0yCWn'
          );
        });
      }
    );

    it(
      'should not load closure lib when native API is available ' +
        '(Uint8Array input)',
      () => {
        return new Crypto(win)
          .sha384Base64(uint8Array([1, 2, 3]))
          .then((hash) => {
            expect(hash).to.equal(
              'hiKdxtL_vqxzgHRBVKpwApHAZDUqDb3He57T8sjh2sTcMlhn053f8dJim3o5PUf2'
            );
          });
      }
    );
  }
});