// File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.

import Cloudflare, { toFile } from 'cloudflare';
import { Response } from 'node-fetch';

const client = new Cloudflare({
  apiKey: '144c9defac04969c7bfad8efaa8ea194',
  apiEmail: 'user@example.com',
  baseURL: process.env['TEST_API_BASE_URL'] ?? 'http://127.0.0.1:4010',
});

describe('resource indexes', () => {
  test('create: only required params', async () => {
    const responsePromise = client.vectorize.indexes.create({
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      config: { dimensions: 768, metric: 'cosine' },
      name: 'example-index',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('create: required and optional params', async () => {
    const response = await client.vectorize.indexes.create({
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      config: { dimensions: 768, metric: 'cosine' },
      name: 'example-index',
      description: 'This is my example index.',
    });
  });

  test('list: only required params', async () => {
    const responsePromise = client.vectorize.indexes.list({ account_id: '023e105f4ecef8ad9ca31a8372d0c353' });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('list: required and optional params', async () => {
    const response = await client.vectorize.indexes.list({ account_id: '023e105f4ecef8ad9ca31a8372d0c353' });
  });

  test('delete: only required params', async () => {
    const responsePromise = client.vectorize.indexes.delete('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('delete: required and optional params', async () => {
    const response = await client.vectorize.indexes.delete('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
  });

  test('deleteByIds: only required params', async () => {
    const responsePromise = client.vectorize.indexes.deleteByIds('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('deleteByIds: required and optional params', async () => {
    const response = await client.vectorize.indexes.deleteByIds('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      ids: ['5121db81354a40c6aedc3fe1ace51c59', 'f90eb49c2107486abdfd78c67e853430'],
    });
  });

  test('get: only required params', async () => {
    const responsePromise = client.vectorize.indexes.get('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('get: required and optional params', async () => {
    const response = await client.vectorize.indexes.get('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
  });

  test('getByIds: only required params', async () => {
    const responsePromise = client.vectorize.indexes.getByIds('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('getByIds: required and optional params', async () => {
    const response = await client.vectorize.indexes.getByIds('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      ids: ['5121db81354a40c6aedc3fe1ace51c59', 'f90eb49c2107486abdfd78c67e853430'],
    });
  });

  test('info: only required params', async () => {
    const responsePromise = client.vectorize.indexes.info('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('info: required and optional params', async () => {
    const response = await client.vectorize.indexes.info('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
  });

  // TODO: investigate broken test
  test.skip('insert: only required params', async () => {
    const responsePromise = client.vectorize.indexes.insert('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      body: await toFile(Buffer.from('# my file contents'), 'README.md'),
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  // TODO: investigate broken test
  test.skip('insert: required and optional params', async () => {
    const response = await client.vectorize.indexes.insert('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      body: await toFile(Buffer.from('# my file contents'), 'README.md'),
      'unparsable-behavior': 'error',
    });
  });

  test('listVectors: only required params', async () => {
    const responsePromise = client.vectorize.indexes.listVectors('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('listVectors: required and optional params', async () => {
    const response = await client.vectorize.indexes.listVectors('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      count: 50,
      cursor: 'suUTaDY5PFUiRweVccnzyt9n75suNPbXHPshvCzue5mHjtj7Letjvzlza9eGj099',
    });
  });

  test('query: only required params', async () => {
    const responsePromise = client.vectorize.indexes.query('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      vector: [0.5, 0.5, 0.5],
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  test('query: required and optional params', async () => {
    const response = await client.vectorize.indexes.query('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      vector: [0.5, 0.5, 0.5],
      filter: { has_viewed: { $ne: true }, streaming_platform: 'netflix' },
      returnMetadata: 'none',
      returnValues: true,
      topK: 5,
    });
  });

  // TODO: investigate broken test
  test.skip('upsert: only required params', async () => {
    const responsePromise = client.vectorize.indexes.upsert('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      body: await toFile(Buffer.from('# my file contents'), 'README.md'),
    });
    const rawResponse = await responsePromise.asResponse();
    expect(rawResponse).toBeInstanceOf(Response);
    const response = await responsePromise;
    expect(response).not.toBeInstanceOf(Response);
    const dataAndResponse = await responsePromise.withResponse();
    expect(dataAndResponse.data).toBe(response);
    expect(dataAndResponse.response).toBe(rawResponse);
  });

  // TODO: investigate broken test
  test.skip('upsert: required and optional params', async () => {
    const response = await client.vectorize.indexes.upsert('example-index', {
      account_id: '023e105f4ecef8ad9ca31a8372d0c353',
      body: await toFile(Buffer.from('# my file contents'), 'README.md'),
      'unparsable-behavior': 'error',
    });
  });
});
