import '../../utils/test-setup';
import {
	closeTestingConnections,
	createTestingConnections,
	reloadTestingDatabases,
} from '../../utils/test-utils';
import { DataSource } from '../../../src/data-source/DataSource';
import { Post } from './entity/Post';

describe('github issues > #4719 HStore with empty string values', () => {
	let connections: DataSource[];
	before(
		async () =>
			(connections = await createTestingConnections({
				entities: [__dirname + '/entity/*{.js,.ts}'],
				enabledDrivers: ['postgres'],
			})),
	);
	beforeEach(() => reloadTestingDatabases(connections));
	after(() => closeTestingConnections(connections));

	it('should handle HStore with empty string keys or values', () =>
		Promise.all(
			connections.map(async (connection) => {
				const queryRunner = connection.createQueryRunner();
				const postRepository = connection.getRepository(Post);

				const post = new Post();
				post.hstoreObj = {
					name: 'Alice',
					surname: 'A',
					age: 25,
					blank: '',
					'': 'blank-key',
					'"': '"',
					foo: null,
				};
				const { id } = await postRepository.save(post);

				const loadedPost = await postRepository.findOneOrFail({
					where: {
						id: id,
					},
				});
				loadedPost.hstoreObj.should.be.deep.equal({
					name: 'Alice',
					surname: 'A',
					age: '25',
					blank: '',
					'': 'blank-key',
					'"': '"',
					foo: null,
				});
				await queryRunner.release();
			}),
		));

	it("should not allow 'hstore injection'", () =>
		Promise.all(
			connections.map(async (connection) => {
				const queryRunner = connection.createQueryRunner();
				const postRepository = connection.getRepository(Post);

				const post = new Post();
				post.hstoreObj = { username: `", admin=>"1`, admin: '0' };
				const { id } = await postRepository.save(post);

				const loadedPost = await postRepository.findOneOrFail({
					where: {
						id: id,
					},
				});
				loadedPost.hstoreObj.should.be.deep.equal({
					username: `", admin=>"1`,
					admin: '0',
				});
				await queryRunner.release();
			}),
		));
});