ai/src/prompt/data-content.test.ts

AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript

import { dataContentSchema } from './data-content';
import { describe, it, expect } from 'vitest';

describe('dataContentSchema', () => {
  it('should validate a Buffer', () => {
    const buffer = Buffer.from('Hello, world!');
    const result = dataContentSchema.parse(buffer);
    expect(result).toEqual(buffer);
  });

  it('should reject a non-matching object', () => {
    const nonMatchingObject = { foo: 'bar' };
    expect(() => dataContentSchema.parse(nonMatchingObject)).toThrow();
  });
});