test: add extensions unit tests

test/operations/extensions/createExtension.spec.ts

@@ -0,0 +1,44 @@
+import { describe, expect, it } from 'vitest';
+import { createExtension } from '../../../src/operations/extensions';
+import { options1 } from '../../utils';
+
+describe('operations', () => {
+  describe('extensions', () => {
+    describe('createExtension', () => {
+      const createExtensionFn = createExtension(options1);
+
+      it('should return a function', () => {
+        expect(createExtensionFn).toBeTypeOf('function');
+      });
+
+      it('should return sql statement', () => {
+        const statement = createExtensionFn('hstore');
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual(['CREATE EXTENSION "hstore";']);
+      });
+
+      it('should return sql statement with dropOptions', () => {
+        const statement = createExtensionFn('hstore', {
+          ifNotExists: true,
+          schema: 'my_schema',
+        });
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual([
+          'CREATE EXTENSION IF NOT EXISTS "hstore" SCHEMA "my_schema";',
+        ]);
+      });
+
+      it('should return sql statement with array', () => {
+        const statement = createExtensionFn(['hstore', 'uuid-ossp']);
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual([
+          'CREATE EXTENSION "hstore";',
+          'CREATE EXTENSION "uuid-ossp";',
+        ]);
+      });
+    });
+  });
+});

test/operations/extensions/dropExtension.spec.ts

@@ -0,0 +1,44 @@
+import { describe, expect, it } from 'vitest';
+import { dropExtension } from '../../../src/operations/extensions';
+import { options1 } from '../../utils';
+
+describe('operations', () => {
+  describe('extensions', () => {
+    describe('dropExtension', () => {
+      const dropExtensionFn = dropExtension(options1);
+
+      it('should return a function', () => {
+        expect(dropExtensionFn).toBeTypeOf('function');
+      });
+
+      it('should return sql statement', () => {
+        const statement = dropExtensionFn('hstore');
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual(['DROP EXTENSION "hstore";']);
+      });
+
+      it('should return sql statement with dropOptions', () => {
+        const statement = dropExtensionFn('hstore', {
+          ifExists: true,
+          cascade: true,
+        });
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual([
+          'DROP EXTENSION IF EXISTS "hstore" CASCADE;',
+        ]);
+      });
+
+      it('should return sql statement with array', () => {
+        const statement = dropExtensionFn(['hstore', 'uuid-ossp']);
+
+        expect(statement).toBeTypeOf('object');
+        expect(statement).toStrictEqual([
+          'DROP EXTENSION "hstore";',
+          'DROP EXTENSION "uuid-ossp";',
+        ]);
+      });
+    });
+  });
+});