Server utils config test

packages/server/test/utils/config.test.js

@@ -0,0 +1,49 @@
+const assert = require('assert')
+const path = require('path')
+
+// Mock dotenv to control the environment variables for testing
+require('dotenv').config = ({ path, override }) => {
+    process.env.LOG_PATH = '/custom/log/path'
+    process.env.LOG_LEVEL = 'debug'
+}
+
+// Mocked path join to control paths for testing
+const originalPathJoin = path.join
+path.join = (...args) => {
+    if (args.includes('logs')) {
+        return '/default/log/path'
+    }
+    return originalPathJoin(...args)
+}
+
+// Import the configuration module
+const config = require('../../dist/utils/config').default
+
+// Restore path.join after the test
+path.join = originalPathJoin
+
+// Test cases
+function testLoggingConfig() {
+    const loggingConfig = config.logging
+
+    // Check if LOG_PATH environment variable is considered
+    assert.strictEqual(loggingConfig.dir, '/custom/log/path', 'Logging directory should match the LOG_PATH environment variable')
+
+    // Check if LOG_LEVEL environment variable is considered
+    assert.strictEqual(loggingConfig.server.level, 'debug', 'Server log level should match the LOG_LEVEL environment variable')
+    assert.strictEqual(loggingConfig.express.level, 'debug', 'Express log level should match the LOG_LEVEL environment variable')
+
+    // Check the default filenames
+    assert.strictEqual(loggingConfig.server.filename, 'server.log', 'Server log filename should be server.log')
+    assert.strictEqual(loggingConfig.server.errorFilename, 'server-error.log', 'Server error log filename should be server-error.log')
+    assert.strictEqual(
+        loggingConfig.express.filename,
+        'server-requests.log.jsonl',
+        'Express log filename should be server-requests.log.jsonl'
+    )
+
+    console.log('All config tests passed')
+}
+
+// Run the tests
+testLoggingConfig()

packages/server/test/utils/hub.test.js

@@ -0,0 +1,140 @@
+const assert = require('assert')
+const { parsePrompt } = require('../../dist/utils/hub')
+
+// Test: Parse system message prompts correctly
+const testSystemMessagePrompt = () => {
+    const prompt = JSON.stringify({
+        kwargs: {
+            messages: [
+                {
+                    id: 'SystemMessagePromptTemplate1',
+                    kwargs: {
+                        prompt: {
+                            kwargs: {
+                                template: 'System message template'
+                            }
+                        }
+                    }
+                }
+            ]
+        }
+    })
+
+    const result = parsePrompt(prompt)
+    const expected = [
+        {
+            type: 'systemMessagePrompt',
+            typeDisplay: 'System Message',
+            template: 'System message template'
+        }
+    ]
+    assert.deepStrictEqual(result, expected)
+    console.log('Test passed: System message prompt')
+}
+
+// Test: Parse human message prompts correctly
+const testHumanMessagePrompt = () => {
+    const prompt = JSON.stringify({
+        kwargs: {
+            messages: [
+                {
+                    id: 'HumanMessagePromptTemplate1',
+                    kwargs: {
+                        prompt: {
+                            kwargs: {
+                                template: 'Human message template'
+                            }
+                        }
+                    }
+                }
+            ]
+        }
+    })
+
+    const result = parsePrompt(prompt)
+    const expected = [
+        {
+            type: 'humanMessagePrompt',
+            typeDisplay: 'Human Message',
+            template: 'Human message template'
+        }
+    ]
+    assert.deepStrictEqual(result, expected)
+    console.log('Test passed: Human message prompt')
+}
+
+// Test: Parse AI message prompts correctly
+const testAIMessagePrompt = () => {
+    const prompt = JSON.stringify({
+        kwargs: {
+            messages: [
+                {
+                    id: 'AIMessagePromptTemplate1',
+                    kwargs: {
+                        prompt: {
+                            kwargs: {
+                                template: 'AI message template'
+                            }
+                        }
+                    }
+                }
+            ]
+        }
+    })
+
+    const result = parsePrompt(prompt)
+    const expected = [
+        {
+            type: 'aiMessagePrompt',
+            typeDisplay: 'AI Message',
+            template: 'AI message template'
+        }
+    ]
+    assert.deepStrictEqual(result, expected)
+    console.log('Test passed: AI message prompt')
+}
+
+// Test: Parse template prompts correctly
+const testTemplatePrompt = () => {
+    const prompt = JSON.stringify({
+        kwargs: {
+            template: 'General template'
+        }
+    })
+
+    const result = parsePrompt(prompt)
+    const expected = [
+        {
+            type: 'template',
+            typeDisplay: 'Prompt',
+            template: 'General template'
+        }
+    ]
+    assert.deepStrictEqual(result, expected)
+    console.log('Test passed: Template prompt')
+}
+
+// Test: Handle empty messages array
+const testEmptyMessages = () => {
+    const prompt = JSON.stringify({
+        kwargs: {
+            messages: []
+        }
+    })
+
+    const result = parsePrompt(prompt)
+    const expected = []
+    assert.deepStrictEqual(result, expected)
+    console.log('Test passed: Empty messages array')
+}
+
+// Run all tests
+const runTests = () => {
+    testSystemMessagePrompt()
+    testHumanMessagePrompt()
+    testAIMessagePrompt()
+    testTemplatePrompt()
+    testEmptyMessages()
+}
+
+runTests()

packages/server/tsconfig.json

@@ -14,5 +14,11 @@
         "strictPropertyInitialization": false,
         "declaration": true
     },
-    "include": ["src"]
+    "include": [
+        "src/**/*.ts"
+    ],
+    "exclude": [
+        "node_modules",
+        "**/*.test.ts"
+    ]
 }