More protobufjs migration

ts/test-both/ContactsParser_test.ts

@@ -0,0 +1,126 @@
+// Copyright 2015-2021 Signal Messenger, LLC
+// SPDX-License-Identifier: AGPL-3.0-only
+
+import { assert } from 'chai';
+import { Writer } from 'protobufjs';
+
+import * as Bytes from '../Bytes';
+import { typedArrayToArrayBuffer } from '../Crypto';
+import { SignalService as Proto } from '../protobuf';
+import { ContactBuffer, GroupBuffer } from '../textsecure/ContactsParser';
+
+describe('ContactsParser', () => {
+  function generateAvatar(): Uint8Array {
+    const result = new Uint8Array(255);
+    for (let i = 0; i < result.length; i += 1) {
+      result[i] = i;
+    }
+    return result;
+  }
+
+  describe('ContactBuffer', () => {
+    function getTestBuffer(): ArrayBuffer {
+      const avatarBuffer = generateAvatar();
+
+      const contactInfoBuffer = Proto.ContactDetails.encode({
+        name: 'Zero Cool',
+        number: '+10000000000',
+        uuid: '7198E1BD-1293-452A-A098-F982FF201902',
+        avatar: { contentType: 'image/jpeg', length: avatarBuffer.length },
+      }).finish();
+
+      const writer = new Writer();
+      writer.bytes(contactInfoBuffer);
+      const prefixedContact = writer.finish();
+
+      const chunks: Array<Uint8Array> = [];
+      for (let i = 0; i < 3; i += 1) {
+        chunks.push(prefixedContact);
+        chunks.push(avatarBuffer);
+      }
+
+      return typedArrayToArrayBuffer(Bytes.concatenate(chunks));
+    }
+
+    it('parses an array buffer of contacts', () => {
+      const arrayBuffer = getTestBuffer();
+      const contactBuffer = new ContactBuffer(arrayBuffer);
+      let contact = contactBuffer.next();
+      let count = 0;
+      while (contact !== undefined) {
+        count += 1;
+        assert.strictEqual(contact.name, 'Zero Cool');
+        assert.strictEqual(contact.number, '+10000000000');
+        assert.strictEqual(
+          contact.uuid,
+          '7198e1bd-1293-452a-a098-f982ff201902'
+        );
+        assert.strictEqual(contact.avatar?.contentType, 'image/jpeg');
+        assert.strictEqual(contact.avatar?.length, 255);
+        assert.strictEqual(contact.avatar?.data.byteLength, 255);
+        const avatarBytes = new Uint8Array(
+          contact.avatar?.data || new ArrayBuffer(0)
+        );
+        for (let j = 0; j < 255; j += 1) {
+          assert.strictEqual(avatarBytes[j], j);
+        }
+        contact = contactBuffer.next();
+      }
+      assert.strictEqual(count, 3);
+    });
+  });
+
+  describe('GroupBuffer', () => {
+    function getTestBuffer(): ArrayBuffer {
+      const avatarBuffer = generateAvatar();
+
+      const groupInfoBuffer = Proto.GroupDetails.encode({
+        id: new Uint8Array([1, 3, 3, 7]),
+        name: 'Hackers',
+        membersE164: ['cereal', 'burn', 'phreak', 'joey'],
+        avatar: { contentType: 'image/jpeg', length: avatarBuffer.length },
+      }).finish();
+
+      const writer = new Writer();
+      writer.bytes(groupInfoBuffer);
+      const prefixedGroup = writer.finish();
+
+      const chunks: Array<Uint8Array> = [];
+      for (let i = 0; i < 3; i += 1) {
+        chunks.push(prefixedGroup);
+        chunks.push(avatarBuffer);
+      }
+
+      return typedArrayToArrayBuffer(Bytes.concatenate(chunks));
+    }
+
+    it('parses an array buffer of groups', () => {
+      const arrayBuffer = getTestBuffer();
+      const groupBuffer = new GroupBuffer(arrayBuffer);
+      let group = groupBuffer.next();
+      let count = 0;
+      while (group !== undefined) {
+        count += 1;
+        assert.strictEqual(group.name, 'Hackers');
+        assert.deepEqual(group.id, new Uint8Array([1, 3, 3, 7]));
+        assert.sameMembers(group.membersE164, [
+          'cereal',
+          'burn',
+          'phreak',
+          'joey',
+        ]);
+        assert.strictEqual(group.avatar?.contentType, 'image/jpeg');
+        assert.strictEqual(group.avatar?.length, 255);
+        assert.strictEqual(group.avatar?.data.byteLength, 255);
+        const avatarBytes = new Uint8Array(
+          group.avatar?.data || new ArrayBuffer(0)
+        );
+        for (let j = 0; j < 255; j += 1) {
+          assert.strictEqual(avatarBytes[j], j);
+        }
+        group = groupBuffer.next();
+      }
+      assert.strictEqual(count, 3);
+    });
+  });
+});