@friggframework/devtools 2.0.0-next.0 → 2.0.0-next.2

package/test/mock-integration.js

@@ -1,65 +1,65 @@
-const { Auther, Credential, Entity, IntegrationFactory, createObjectId } = require('@friggframework/core');
+const {
+    Auther,
+    Credential,
+    Entity,
+    IntegrationFactory,
+    createObjectId,
+} = require('@friggframework/core');
 
-
-async function createMockIntegration(IntegrationClassDef, userId = null, config = {},) {
-    const integrationFactory = new IntegrationFactory([IntegrationClassDef]);
+async function createMockIntegration(
+    IntegrationClass,
+    userId = null,
+    config = { type: IntegrationClass.Definition.name }
+) {
+    const integrationFactory = new IntegrationFactory([IntegrationClass]);
     userId = userId || createObjectId();
 
     const insertOptions = {
         new: true,
         upsert: true,
         setDefaultsOnInsert: true,
-    }
-    const user = {user: userId}
-
-    const credential = await Credential.findOneAndUpdate(
-        user,
-        { $set: user },
-        insertOptions
-    );
-    const entity1 = await Entity.findOneAndUpdate(
-        user,
-        {
-            $set: {
-                credential: credential.id,
-                user: userId,
-                name: 'Test user',
-                externalId: '1234567890123456',
-            },
-        },
-        insertOptions
-    );
-    const entity2 = await Entity.findOneAndUpdate(
-        user,
-        {
-            $set: {
-                credential: credential.id,
-                user: userId,
-            },
-        },
-        insertOptions
-    );
+    };
+    const user = { user: userId };
 
-    const entities = [entity1, entity2]
-
-    const integration =
-        await integrationFactory.createIntegration(
-            entities,
-            userId,
-            config,
+    const entities = [];
+    for (const moduleName in IntegrationClass.modules) {
+        const ModuleDef = IntegrationClass.Definition.modules[moduleName];
+        const module = await Auther.getInstance({
+            definition: ModuleDef,
+            userId: userId,
+        });
+        const credential = await module.CredentialModel.findOneAndUpdate(
+            user,
+            { $set: user },
+            insertOptions
+        );
+        entities.push(
+            (
+                await module.EntityModel.findOneAndUpdate(
+                    user,
+                    {
+                        $set: {
+                            credential,
+                            user: userId,
+                            name: `Test ${moduleName}`,
+                            externalId: `1234567890123456_${moduleName}`,
+                        },
+                    },
+                    insertOptions
+                )
+            ).id
         );
+    }
 
-    integration.id = integration.record._id
+    const integration = await integrationFactory.createIntegration(
+        entities,
+        userId,
+        config
+    );
 
-    for (const i in entities){
-        if (Object.entries(IntegrationClassDef.modules).length <= i) break
-        const [moduleName, ModuleDef] = Object.entries(IntegrationClassDef.modules)[i];
-        const module = await Auther.getInstance({definition: ModuleDef, userId: userId})
-        module.entity = entities[i];
-        integration[moduleName] = module;
-    }
+    integration.id = integration.record._id;
 
-    return integration
+    return integration;
 }
 
 function createMockApiObject(jest, api = {}, mockMethodMap) {
@@ -67,17 +67,22 @@ function createMockApiObject(jest, api = {}, mockMethodMap) {
     // and values which are the mock response (or implementation)
     const clone = (data) => JSON.parse(JSON.stringify(data));
 
-    for (const [methodName, mockDataOrImplementation] of Object.entries(mockMethodMap)) {
+    for (const [methodName, mockDataOrImplementation] of Object.entries(
+        mockMethodMap
+    )) {
         if (mockDataOrImplementation instanceof Function) {
             api[methodName] = jest.fn(mockDataOrImplementation);
-        }
-        else if (api[methodName]?.constructor?.name === "AsyncFunction") {
-            api[methodName] = jest.fn().mockResolvedValue(clone(mockDataOrImplementation));
+        } else if (api[methodName]?.constructor?.name === 'AsyncFunction') {
+            api[methodName] = jest
+                .fn()
+                .mockResolvedValue(clone(mockDataOrImplementation));
         } else {
-            api[methodName] = jest.fn().mockReturnValue(clone(mockDataOrImplementation));
+            api[methodName] = jest
+                .fn()
+                .mockReturnValue(clone(mockDataOrImplementation));
         }
     }
     return api;
 }
 
-module.exports = {createMockIntegration, createMockApiObject};
+module.exports = { createMockIntegration, createMockApiObject };

package/frigg-cli/environmentVariables.test.js

@@ -1,86 +0,0 @@
-const { handleEnvVariables } = require('./environmentVariables');
-const { logInfo } = require('./logger');
-const inquirer = require('inquirer');
-const fs = require('fs');
-const dotenv = require('dotenv');
-const { resolve } = require('node:path');
-const { parse } = require('@babel/parser');
-const traverse = require('@babel/traverse');
-
-jest.mock('inquirer');
-jest.mock('fs');
-jest.mock('dotenv');
-jest.mock('./logger');
-jest.mock('@babel/parser');
-jest.mock('@babel/traverse');
-
-describe('handleEnvVariables', () => {
-    const backendPath = '/mock/backend/path';
-    const modulePath = '/mock/module/path';
-
-    beforeEach(() => {
-        jest.clearAllMocks();
-        fs.readFileSync.mockReturnValue(`
-            const Definition = {
-                env: {
-                    client_id: process.env.GOOGLE_CALENDAR_CLIENT_ID,
-                    client_secret: process.env.GOOGLE_CALENDAR_CLIENT_SECRET,
-                    redirect_uri: \`\${process.env.REDIRECT_URI}/google-calendar\`,
-                    scope: process.env.GOOGLE_CALENDAR_SCOPE,
-                }
-            };
-        `);
-        parse.mockReturnValue({});
-        traverse.default.mockImplementation((ast, visitor) => {
-            visitor.ObjectProperty({
-                node: {
-                    key: { name: 'env' },
-                    value: {
-                        properties: [
-                            { key: { name: 'client_id' }, value: { type: 'MemberExpression', object: { name: 'process' }, property: { name: 'GOOGLE_CALENDAR_CLIENT_ID' } } },
-                            { key: { name: 'client_secret' }, value: { type: 'MemberExpression', object: { name: 'process' }, property: { name: 'GOOGLE_CALENDAR_CLIENT_SECRET' } } },
-                            { key: { name: 'redirect_uri' }, value: { type: 'MemberExpression', object: { name: 'process' }, property: { name: 'REDIRECT_URI' } } },
-                            { key: { name: 'scope' }, value: { type: 'MemberExpression', object: { name: 'process' }, property: { name: 'GOOGLE_CALENDAR_SCOPE' } } },
-                        ]
-                    }
-                }
-            });
-        });
-    });
-
-    xit('should identify and handle missing environment variables', async () => {
-        const localEnvPath = resolve(backendPath, '../.env');
-        const localDevConfigPath = resolve(backendPath, '../src/configs/dev.json');
-
-        fs.existsSync.mockImplementation((path) => path === localEnvPath || path === localDevConfigPath);
-        dotenv.parse.mockReturnValue({});
-        fs.readFileSync.mockImplementation((path) => {
-            if (path === resolve(modulePath, 'index.js')) return 'mock module content';
-            if (path === localEnvPath) return '';
-            if (path === localDevConfigPath) return '{}';
-            return '';
-        });
-
-        inquirer.prompt.mockResolvedValueOnce({ addEnvVars: true })
-            .mockResolvedValueOnce({ value: 'client_id_value' })
-            .mockResolvedValueOnce({ value: 'client_secret_value' })
-            .mockResolvedValueOnce({ value: 'redirect_uri_value' })
-            .mockResolvedValueOnce({ value: 'scope_value' });
-
-        await handleEnvVariables(backendPath, modulePath);
-
-        expect(logInfo).toHaveBeenCalledWith('Searching for missing environment variables...');
-        expect(logInfo).toHaveBeenCalledWith('Missing environment variables: GOOGLE_CALENDAR_CLIENT_ID, GOOGLE_CALENDAR_CLIENT_SECRET, REDIRECT_URI, GOOGLE_CALENDAR_SCOPE');
-        expect(inquirer.prompt).toHaveBeenCalledTimes(5);
-        expect(fs.appendFileSync).toHaveBeenCalledWith(localEnvPath, '\nGOOGLE_CALENDAR_CLIENT_ID=client_id_value\nGOOGLE_CALENDAR_CLIENT_SECRET=client_secret_value\nREDIRECT_URI=redirect_uri_value\nGOOGLE_CALENDAR_SCOPE=scope_value');
-        expect(fs.writeFileSync).toHaveBeenCalledWith(
-            localDevConfigPath,
-            JSON.stringify({
-                GOOGLE_CALENDAR_CLIENT_ID: 'client_id_value',
-                GOOGLE_CALENDAR_CLIENT_SECRET: 'client_secret_value',
-                REDIRECT_URI: 'redirect_uri_value',
-                GOOGLE_CALENDAR_SCOPE: 'scope_value'
-            }, null, 2)
-        );
-    });
-});