import {expect, test} from 'bun:test';
import {exampleVideos} from '@remotion/example-videos';
import {parseMedia} from '../parse-media';
import {nodeReader} from '../readers/from-node';

test('Parse only header of WebM', async () => {
	const {internalStats, container} = await parseMedia({
		src: exampleVideos.stretchedVp8,
		fields: {
			size: true,
			internalStats: true,
			container: true,
		},
		acknowledgeRemotionLicense: true,
		reader: nodeReader,
	});

	expect(internalStats).toEqual({
		finalCursorOffset: 0,
		skippedBytes: 13195359,
	});
	expect(container).toEqual('webm');
});

test('Parse WebM partially', async () => {
	const {internalStats, numberOfAudioChannels, sampleRate} = await parseMedia({
		src: exampleVideos.stretchedVp8,
		fields: {
			tracks: true,
			internalStats: true,
			sampleRate: true,
			numberOfAudioChannels: true,
		},
		acknowledgeRemotionLicense: true,
		reader: nodeReader,
	});

	expect(internalStats).toEqual({
		finalCursorOffset: 4540,
		skippedBytes: 13190819,
	});
	expect(numberOfAudioChannels).toBe(2);
	expect(sampleRate).toBe(44100);
});

test('Parse WebM fully', async () => {
	const {internalStats} = await parseMedia({
		src: exampleVideos.stretchedVp8,
		fields: {
			tracks: true,
			internalStats: true,
		},
		acknowledgeRemotionLicense: true,
		onVideoTrack: () => () => undefined,
		reader: nodeReader,
	});

	expect(internalStats).toEqual({
		finalCursorOffset: 13195359,
		skippedBytes: 0,
	});
});