import 'reflect-metadata';
import { DataSource } from '../../../src/data-source/DataSource';
import {
	closeTestingConnections,
	createTestingConnections,
	reloadTestingDatabases,
} from '../../utils/test-utils';
import { Document } from '../bulk-save-case2/entity/Document';

describe('benchmark > bulk-save > case-querybuilder', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				__dirname,
				enabledDrivers: ['postgres'],
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('testing bulk save of 10000 objects', () =>
		Promise.all(
			connections.map(async (connection) => {
				const documents: Document[] = [];
				for (let i = 0; i < 10000; i++) {
					const document = new Document();

					document.id = i.toString();
					document.docId = 'label/' + i;
					document.context =
						'Lorem ipsum dolor sit amet, consectetur adipiscing elit. Praesent vel faucibus nunc. Etiam volutpat vel urna in scelerisque. Cras a erat ipsum. ';
					document.label = 'label/' + i;
					document.date = new Date();

					documents.push(document);
				}

				await connection
					.createQueryRunner()
					.query(
						`CREATE TABLE "document" ("id" text NOT NULL, "docId" text NOT NULL, "label" text NOT NULL, "context" text NOT NULL, "date" TIMESTAMP WITH TIME ZONE NOT NULL, CONSTRAINT "PK_e57d3357f83f3cdc0acffc3d777" PRIMARY KEY ("id"))`,
					);
				await connection.manager
					.createQueryBuilder()
					.insert()
					.into('document', ['id', 'docId', 'label', 'context', 'date'])
					.values(documents)
					.execute();
			}),
		));
});