shaka-player/test/msf/msf_classes_unit.js

filterDescribe('shaka.msf.Reader', isMSFSupported, () => {
  /** @type {!shaka.msf.Reader} */
  let reader;

  // Helper: create a readable stream from Uint8Array chunks
  const createTestStream = (chunks) => {
    let index = 0;
    return new ReadableStream({
      pull: (controller) => {
        if (index < chunks.length) {
          controller.enqueue(chunks[index++]);
        } else {
          controller.close();
        }
      },
    });
  };

  it('should initialize with empty buffer', () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const buffer = new Uint8Array([]);
    const stream = createTestStream([]);
    reader = new shaka.msf.Reader(buffer, stream);
    expect(reader.getByteLength()).toBe(0);
    expect(reader.getBuffer().length).toBe(0);
  });

  it('should read bytes correctly', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([1, 2, 3, 4, 5])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    const bytes = await reader.read(3);
    expect(bytes).toEqual(new Uint8Array([1, 2, 3]));

    const remaining = await reader.readAll();
    expect(remaining).toEqual(new Uint8Array([4, 5]));
  });

  it('should read u8 and u8Bool correctly', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([0x01, 0x00])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    expect(await reader.u8()).toBe(1);
    expect(await reader.u8Bool()).toBe(false);
  });

  it('should read string correctly', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([0x02, 72, 105])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    const str = await reader.string(10);
    expect(str).toBe('Hi');
  });

  it('should throw if string exceeds maxLength', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([0x02, 65, 66])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    // Use expectAsync instead of try/catch/fail
    await expectAsync(reader.string(1)).toBeRejectedWith(
        jasmine.objectContaining({
          message: 'string length 2 exceeds max length 1',
        }));
  });

  it('should read tuple correctly', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([
      new Uint8Array([0x02, 0x01, 65, 0x01, 66]),
    ]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    const tuple = await reader.tuple();
    expect(tuple).toEqual(['A', 'B']);
  });

  it('done() should reflect buffer and stream state', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([1, 2])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    expect(await reader.done()).toBe(false);
    await reader.readAll();
    expect(await reader.done()).toBe(true);
  });

  it('release() and close() should not throw', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const stream = createTestStream([new Uint8Array([1])]);
    reader = new shaka.msf.Reader(new Uint8Array([]), stream);

    expect(() => reader.release()).not.toThrow();

    await expectAsync(reader.close()).toBeResolved();
  });

  it('should read keyValuePairs correctly', async () => {
    if (!isReadableStreamSupported()) {
      pending('ReadableStream is not supported by the platform.');
    }
    const bytes = new Uint8Array([0x02, 0x02, 0x03, 0x01, 0x01, 65]);
    reader = new shaka.msf.Reader(
        new Uint8Array([]), createTestStream([bytes]));

    const pairs = await reader.keyValuePairs();
    expect(pairs.length).toBe(2);
    expect(pairs[0]).toEqual({type: BigInt(2), value: BigInt(3)});
    expect(pairs[1].type).toBe(BigInt(1));
    expect(shaka.util.StringUtils.fromUTF8(
        /** @type {!ArrayBufferView} */ (pairs[1].value),
    )).toBe('A');
  });
});

filterDescribe('shaka.msf.Writer', isMSFSupported, () => {
  /** @type {!shaka.msf.Writer} */
  let writer;

  /** @type {!Array<!Uint8Array>} */
  let writtenChunks;

  // Helper: writable stream storing chunks in array
  const createTestWritable = () => {
    writtenChunks = [];
    return new WritableStream({
      write: (chunk) => {
        writtenChunks.push(/** @type {!Uint8Array} */ (chunk));
      },
    });
  };

  beforeEach(() => {
    const writable = createTestWritable();
    writer = new shaka.msf.Writer(writable);
  });

  it('should write a single Uint8Array chunk', async () => {
    if (!isWritableStreamSupported()) {
      pending('WritableStream is not supported by the platform.');
    }
    const data = new Uint8Array([10, 20]);
    await writer.write(data);
    expect(writtenChunks.length).toBe(1);
    expect(writtenChunks[0]).toEqual(data);
  });

  it('should write multiple chunks sequentially', async () => {
    if (!isWritableStreamSupported()) {
      pending('WritableStream is not supported by the platform.');
    }
    const data1 = new Uint8Array([1]);
    const data2 = new Uint8Array([2]);
    await writer.write(data1);
    await writer.write(data2);
    expect(writtenChunks).toEqual([data1, data2]);
  });
});

filterDescribe('shaka.msf.Receiver', isMSFSupported, () => {
  let reader;
  let receiver;

  function createReader(values) {
    const buffer = shaka.util.BufferUtils.toUint8(values);
    let offset = 0;
    return {
      u53: () => {
        if (offset >= buffer.length) {
          throw new Error('unexpected end of stream');
        }
        return buffer[offset++];
      },
      u53WithSize: () => {
        if (offset >= buffer.length) {
          throw new Error('unexpected end of stream');
        }
        const value = buffer[offset++];
        return {value, bytesRead: 1};
      },
      u62WithSize: () => {
        if (offset >= buffer.length) {
          throw new Error('unexpected end of stream');
        }
        const value = BigInt(buffer[offset++]);
        return {value, bytesRead: 1};
      },
      read: (length) => {
        const result = buffer.subarray(offset, offset + length);
        offset += length;
        return result;
      },
      getByteLength: () => offset,
    };
  }

  it('should decode a server setup with no parameters', async () => {
    const SERVER_SETUP = shaka.msf.Utils.MessageTypeId.SERVER_SETUP;

    // type = SERVER, length = 1, version = 1, param count = 0
    const readerValues = [
      SERVER_SETUP, // type
      0x00, 0x01,       // message length = 1 byte
      0x01,             // version
      0x00,              // param count = 0
    ];

    reader = createReader(readerValues);
    receiver =
        new shaka.msf.Receiver(reader, shaka.msf.Utils.Version.DRAFT_14);

    const result = await receiver.server();
    expect(result.version).toBe(1);
    expect(result.params).toBeUndefined();
  });

  it('should decode server setup with numeric parameter', async () => {
    const SERVER_SETUP = shaka.msf.Utils.MessageTypeId.SERVER_SETUP;

    // type = SERVER, length = 3 bytes, version = 1
    // param count = 1, param type = 2 (even), param value = 42
    const readerValues = [
      SERVER_SETUP, // type
      0x00, 0x03,       // message length
      0x01,             // version
      0x01,             // param count
      0x02,             // param type
      0x2A,              // param value
    ];

    reader = createReader(readerValues);
    receiver =
        new shaka.msf.Receiver(reader, shaka.msf.Utils.Version.DRAFT_14);

    const result = await receiver.server();
    expect(result.version).toBe(1);
    expect(result.params.length).toBe(1);
    expect(result.params[0].type).toBe(BigInt(2));
    expect(result.params[0].value).toBe(BigInt(42));
  });

  it('should throw error if server type is invalid', async () => {
    const readerValues = [
      0x00, 0x00, 0x01, 0x01, // invalid type
    ];

    reader = createReader(readerValues);
    receiver =
        new shaka.msf.Receiver(reader, shaka.msf.Utils.Version.DRAFT_14);

    await expectAsync(receiver.server()).toBeRejectedWith(
        jasmine.objectContaining({
          message: jasmine.stringMatching(/Server SETUP type must be/),
        }),
    );
  });
});

filterDescribe('shaka.msf.Sender', isMSFSupported, () => {
  let sender;
  /** @type {!Array<!Uint8Array>} */
  let writtenChunks;

  function createMockWritableStream() {
    writtenChunks = [];
    return new WritableStream({
      write: (chunk) => {
        writtenChunks.push(shaka.util.BufferUtils.toUint8(chunk));
      },
    });
  }

  it('should send client setup message', async () => {
    if (!isWritableStreamSupported()) {
      pending('WritableStream is not supported by the platform.');
    }
    const writable = createMockWritableStream();
    const writer = new shaka.msf.Writer(writable);
    sender = new shaka.msf.Sender(writer, shaka.msf.Utils.Version.DRAFT_14);

    const clientSetup = {
      versions: [1],
      params: [
        {
          type: BigInt(2),
          value: BigInt(42),
        },
        {
          type: BigInt(3),
          value: new Uint8Array([0xde, 0xad]),
        },
      ],
    };

    await sender.client(clientSetup);

    expect(writtenChunks.length).toBeGreaterThan(0);
    expect(writtenChunks.some((chunk) => chunk.length > 0)).toBe(true);
  });
});