/**
 * Copyright 2024 Google LLC
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import * as assert from 'assert';
import { describe, it } from 'node:test';
import { enumFormatter } from '../../src/formats/enum.js';
import { Message } from '../../src/message.js';
import { MessageData } from '../../src/model.js';

describe('enumFormat', () => {
  const messageTests = [
    {
      desc: 'parses simple enum value',
      message: {
        role: 'model',
        content: [{ text: 'VALUE1' }],
      },
      want: 'VALUE1',
    },
    {
      desc: 'trims whitespace',
      message: {
        role: 'model',
        content: [{ text: '  VALUE2\n' }],
      },
      want: 'VALUE2',
    },
  ];

  for (const rt of messageTests) {
    it(rt.desc, () => {
      const parser = enumFormatter.handler();
      assert.strictEqual(
        parser.parseMessage(new Message(rt.message as MessageData)),
        rt.want
      );
    });
  }

  const errorTests = [
    {
      desc: 'throws error for number schema type',
      schema: { type: 'number' },
      wantError: /Must supply a 'string' or 'enum' schema type/,
    },
    {
      desc: 'throws error for array schema type',
      schema: { type: 'array' },
      wantError: /Must supply a 'string' or 'enum' schema type/,
    },
  ];

  for (const et of errorTests) {
    it(et.desc, () => {
      assert.throws(() => {
        enumFormatter.handler(et.schema);
      }, et.wantError);
    });
  }
});