weave-typescript 0.11.14 → 0.11.16

package/dist/weavesql/weavedb/test_script_sql.js

@@ -0,0 +1,605 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deleteScriptBindingQuery = exports.listScriptBindingsQuery = exports.upsertScriptBindingQuery = exports.setCurrentTestScriptVersionQuery = exports.getNextTestScriptVersionNumberQuery = exports.listTestScriptVersionsQuery = exports.getTestScriptVersionQuery = exports.createTestScriptVersionQuery = exports.listWorkflowTestScriptBindingsQuery = exports.detachTestScriptFromWorkflowQuery = exports.attachTestScriptToWorkflowQuery = exports.deleteTestScriptQuery = exports.archiveTestScriptQuery = exports.updateTestScriptQuery = exports.listTestScriptsByWorkflowQuery = exports.listTestScriptsByProjectQuery = exports.getTestScriptQuery = exports.createTestScriptQuery = void 0;
+exports.createTestScript = createTestScript;
+exports.getTestScript = getTestScript;
+exports.listTestScriptsByProject = listTestScriptsByProject;
+exports.listTestScriptsByWorkflow = listTestScriptsByWorkflow;
+exports.updateTestScript = updateTestScript;
+exports.archiveTestScript = archiveTestScript;
+exports.deleteTestScript = deleteTestScript;
+exports.attachTestScriptToWorkflow = attachTestScriptToWorkflow;
+exports.detachTestScriptFromWorkflow = detachTestScriptFromWorkflow;
+exports.listWorkflowTestScriptBindings = listWorkflowTestScriptBindings;
+exports.createTestScriptVersion = createTestScriptVersion;
+exports.getTestScriptVersion = getTestScriptVersion;
+exports.listTestScriptVersions = listTestScriptVersions;
+exports.getNextTestScriptVersionNumber = getNextTestScriptVersionNumber;
+exports.setCurrentTestScriptVersion = setCurrentTestScriptVersion;
+exports.upsertScriptBinding = upsertScriptBinding;
+exports.listScriptBindings = listScriptBindings;
+exports.deleteScriptBinding = deleteScriptBinding;
+exports.createTestScriptQuery = `-- name: CreateTestScript :one
+INSERT INTO weave_v1.test_script (
+    id,
+    project_id,
+    test_case_id,
+    name,
+    description,
+    language,
+    framework,
+    status,
+    metadata
+)
+VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9
+)
+RETURNING id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at`;
+async function createTestScript(client, args) {
+    const result = await client.query({
+        text: exports.createTestScriptQuery,
+        values: [args.id, args.projectId, args.testCaseId, args.name, args.description, args.language, args.framework, args.status, args.metadata],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        projectId: row[1],
+        testCaseId: row[2],
+        name: row[3],
+        description: row[4],
+        language: row[5],
+        framework: row[6],
+        status: row[7],
+        currentVersionId: row[8],
+        metadata: row[9],
+        createdAt: row[10],
+        updatedAt: row[11],
+        archivedAt: row[12]
+    };
+}
+exports.getTestScriptQuery = `-- name: GetTestScript :one
+SELECT id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at
+FROM weave_v1.test_script
+WHERE id = $1
+  AND project_id = $2`;
+async function getTestScript(client, args) {
+    const result = await client.query({
+        text: exports.getTestScriptQuery,
+        values: [args.id, args.projectId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        projectId: row[1],
+        testCaseId: row[2],
+        name: row[3],
+        description: row[4],
+        language: row[5],
+        framework: row[6],
+        status: row[7],
+        currentVersionId: row[8],
+        metadata: row[9],
+        createdAt: row[10],
+        updatedAt: row[11],
+        archivedAt: row[12]
+    };
+}
+exports.listTestScriptsByProjectQuery = `-- name: ListTestScriptsByProject :many
+SELECT id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at
+FROM weave_v1.test_script
+WHERE project_id = $1
+  AND ($2::BOOLEAN OR archived_at IS NULL)
+ORDER BY updated_at DESC
+LIMIT $4
+OFFSET $3`;
+async function listTestScriptsByProject(client, args) {
+    const result = await client.query({
+        text: exports.listTestScriptsByProjectQuery,
+        values: [args.projectId, args.includeArchived, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            projectId: row[1],
+            testCaseId: row[2],
+            name: row[3],
+            description: row[4],
+            language: row[5],
+            framework: row[6],
+            status: row[7],
+            currentVersionId: row[8],
+            metadata: row[9],
+            createdAt: row[10],
+            updatedAt: row[11],
+            archivedAt: row[12]
+        };
+    });
+}
+exports.listTestScriptsByWorkflowQuery = `-- name: ListTestScriptsByWorkflow :many
+SELECT s.id, s.project_id, s.test_case_id, s.name, s.description, s.language, s.framework, s.status, s.current_version_id, s.metadata, s.created_at, s.updated_at, s.archived_at
+FROM weave_v1.test_script s
+         INNER JOIN weave_v1.workflow_test_script b
+                    ON b.test_script_id = s.id
+                        AND b.project_id = s.project_id
+WHERE b.project_id = $1
+  AND b.workflow_id = $2
+  AND ($3::BOOLEAN OR s.archived_at IS NULL)
+ORDER BY s.updated_at DESC
+LIMIT $5
+OFFSET $4`;
+async function listTestScriptsByWorkflow(client, args) {
+    const result = await client.query({
+        text: exports.listTestScriptsByWorkflowQuery,
+        values: [args.projectId, args.workflowId, args.includeArchived, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            projectId: row[1],
+            testCaseId: row[2],
+            name: row[3],
+            description: row[4],
+            language: row[5],
+            framework: row[6],
+            status: row[7],
+            currentVersionId: row[8],
+            metadata: row[9],
+            createdAt: row[10],
+            updatedAt: row[11],
+            archivedAt: row[12]
+        };
+    });
+}
+exports.updateTestScriptQuery = `-- name: UpdateTestScript :one
+UPDATE weave_v1.test_script
+SET test_case_id = $1,
+    name = $2,
+    description = $3,
+    language = $4,
+    framework = $5,
+    status = $6,
+    metadata = $7,
+    updated_at = NOW()
+WHERE id = $8
+  AND project_id = $9
+RETURNING id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at`;
+async function updateTestScript(client, args) {
+    const result = await client.query({
+        text: exports.updateTestScriptQuery,
+        values: [args.testCaseId, args.name, args.description, args.language, args.framework, args.status, args.metadata, args.id, args.projectId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        projectId: row[1],
+        testCaseId: row[2],
+        name: row[3],
+        description: row[4],
+        language: row[5],
+        framework: row[6],
+        status: row[7],
+        currentVersionId: row[8],
+        metadata: row[9],
+        createdAt: row[10],
+        updatedAt: row[11],
+        archivedAt: row[12]
+    };
+}
+exports.archiveTestScriptQuery = `-- name: ArchiveTestScript :one
+UPDATE weave_v1.test_script
+SET status = 'archived',
+    archived_at = NOW(),
+    updated_at = NOW()
+WHERE id = $1
+  AND project_id = $2
+RETURNING id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at`;
+async function archiveTestScript(client, args) {
+    const result = await client.query({
+        text: exports.archiveTestScriptQuery,
+        values: [args.id, args.projectId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        projectId: row[1],
+        testCaseId: row[2],
+        name: row[3],
+        description: row[4],
+        language: row[5],
+        framework: row[6],
+        status: row[7],
+        currentVersionId: row[8],
+        metadata: row[9],
+        createdAt: row[10],
+        updatedAt: row[11],
+        archivedAt: row[12]
+    };
+}
+exports.deleteTestScriptQuery = `-- name: DeleteTestScript :exec
+DELETE
+FROM weave_v1.test_script
+WHERE id = $1
+  AND project_id = $2`;
+async function deleteTestScript(client, args) {
+    await client.query({
+        text: exports.deleteTestScriptQuery,
+        values: [args.id, args.projectId],
+        rowMode: "array"
+    });
+}
+exports.attachTestScriptToWorkflowQuery = `-- name: AttachTestScriptToWorkflow :one
+INSERT INTO weave_v1.workflow_test_script (
+    project_id,
+    workflow_id,
+    test_script_id,
+    attached_by,
+    metadata
+)
+VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5
+)
+ON CONFLICT (workflow_id, test_script_id) DO UPDATE
+SET attached_by = EXCLUDED.attached_by,
+    metadata = EXCLUDED.metadata,
+    attached_at = NOW()
+RETURNING project_id, workflow_id, test_script_id, attached_by, metadata, attached_at`;
+async function attachTestScriptToWorkflow(client, args) {
+    const result = await client.query({
+        text: exports.attachTestScriptToWorkflowQuery,
+        values: [args.projectId, args.workflowId, args.testScriptId, args.attachedBy, args.metadata],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        projectId: row[0],
+        workflowId: row[1],
+        testScriptId: row[2],
+        attachedBy: row[3],
+        metadata: row[4],
+        attachedAt: row[5]
+    };
+}
+exports.detachTestScriptFromWorkflowQuery = `-- name: DetachTestScriptFromWorkflow :exec
+DELETE
+FROM weave_v1.workflow_test_script
+WHERE project_id = $1
+  AND workflow_id = $2
+  AND test_script_id = $3`;
+async function detachTestScriptFromWorkflow(client, args) {
+    await client.query({
+        text: exports.detachTestScriptFromWorkflowQuery,
+        values: [args.projectId, args.workflowId, args.testScriptId],
+        rowMode: "array"
+    });
+}
+exports.listWorkflowTestScriptBindingsQuery = `-- name: ListWorkflowTestScriptBindings :many
+SELECT project_id, workflow_id, test_script_id, attached_by, metadata, attached_at
+FROM weave_v1.workflow_test_script
+WHERE project_id = $1
+  AND workflow_id = $2
+ORDER BY attached_at DESC
+LIMIT $4
+OFFSET $3`;
+async function listWorkflowTestScriptBindings(client, args) {
+    const result = await client.query({
+        text: exports.listWorkflowTestScriptBindingsQuery,
+        values: [args.projectId, args.workflowId, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            projectId: row[0],
+            workflowId: row[1],
+            testScriptId: row[2],
+            attachedBy: row[3],
+            metadata: row[4],
+            attachedAt: row[5]
+        };
+    });
+}
+exports.createTestScriptVersionQuery = `-- name: CreateTestScriptVersion :one
+INSERT INTO weave_v1.test_script_version (
+    id,
+    test_script_id,
+    version_number,
+    source_code,
+    metadata,
+    prompt_version,
+    source_location_type,
+    storage_connection_id,
+    source_uri,
+    external_revision,
+    source_checksum,
+    source_size_bytes
+)
+VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10,
+    $11,
+    $12
+)
+RETURNING id, test_script_id, version_number, source_code, metadata, prompt_version, source_location_type, storage_connection_id, source_uri, external_revision, source_checksum, source_size_bytes, created_at`;
+async function createTestScriptVersion(client, args) {
+    const result = await client.query({
+        text: exports.createTestScriptVersionQuery,
+        values: [args.id, args.testScriptId, args.versionNumber, args.sourceCode, args.metadata, args.promptVersion, args.sourceLocationType, args.storageConnectionId, args.sourceUri, args.externalRevision, args.sourceChecksum, args.sourceSizeBytes],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        testScriptId: row[1],
+        versionNumber: row[2],
+        sourceCode: row[3],
+        metadata: row[4],
+        promptVersion: row[5],
+        sourceLocationType: row[6],
+        storageConnectionId: row[7],
+        sourceUri: row[8],
+        externalRevision: row[9],
+        sourceChecksum: row[10],
+        sourceSizeBytes: row[11],
+        createdAt: row[12]
+    };
+}
+exports.getTestScriptVersionQuery = `-- name: GetTestScriptVersion :one
+SELECT sv.id, sv.test_script_id, sv.version_number, sv.source_code, sv.metadata, sv.prompt_version, sv.source_location_type, sv.storage_connection_id, sv.source_uri, sv.external_revision, sv.source_checksum, sv.source_size_bytes, sv.created_at
+FROM weave_v1.test_script_version sv
+         INNER JOIN weave_v1.test_script s ON s.id = sv.test_script_id
+WHERE sv.id = $1
+  AND sv.test_script_id = $2
+  AND s.project_id = $3`;
+async function getTestScriptVersion(client, args) {
+    const result = await client.query({
+        text: exports.getTestScriptVersionQuery,
+        values: [args.id, args.testScriptId, args.projectId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        testScriptId: row[1],
+        versionNumber: row[2],
+        sourceCode: row[3],
+        metadata: row[4],
+        promptVersion: row[5],
+        sourceLocationType: row[6],
+        storageConnectionId: row[7],
+        sourceUri: row[8],
+        externalRevision: row[9],
+        sourceChecksum: row[10],
+        sourceSizeBytes: row[11],
+        createdAt: row[12]
+    };
+}
+exports.listTestScriptVersionsQuery = `-- name: ListTestScriptVersions :many
+SELECT sv.id, sv.test_script_id, sv.version_number, sv.source_code, sv.metadata, sv.prompt_version, sv.source_location_type, sv.storage_connection_id, sv.source_uri, sv.external_revision, sv.source_checksum, sv.source_size_bytes, sv.created_at
+FROM weave_v1.test_script_version sv
+         INNER JOIN weave_v1.test_script s ON s.id = sv.test_script_id
+WHERE sv.test_script_id = $1
+  AND s.project_id = $2
+ORDER BY sv.version_number DESC
+LIMIT $4
+OFFSET $3`;
+async function listTestScriptVersions(client, args) {
+    const result = await client.query({
+        text: exports.listTestScriptVersionsQuery,
+        values: [args.testScriptId, args.projectId, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            testScriptId: row[1],
+            versionNumber: row[2],
+            sourceCode: row[3],
+            metadata: row[4],
+            promptVersion: row[5],
+            sourceLocationType: row[6],
+            storageConnectionId: row[7],
+            sourceUri: row[8],
+            externalRevision: row[9],
+            sourceChecksum: row[10],
+            sourceSizeBytes: row[11],
+            createdAt: row[12]
+        };
+    });
+}
+exports.getNextTestScriptVersionNumberQuery = `-- name: GetNextTestScriptVersionNumber :one
+SELECT COALESCE(MAX(version_number), 0)::INTEGER + 1 AS version_number
+FROM weave_v1.test_script_version
+WHERE test_script_id = $1`;
+async function getNextTestScriptVersionNumber(client, args) {
+    const result = await client.query({
+        text: exports.getNextTestScriptVersionNumberQuery,
+        values: [args.testScriptId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        versionNumber: row[0]
+    };
+}
+exports.setCurrentTestScriptVersionQuery = `-- name: SetCurrentTestScriptVersion :one
+UPDATE weave_v1.test_script
+SET current_version_id = $1,
+    updated_at = NOW()
+WHERE id = $2
+  AND project_id = $3
+RETURNING id, project_id, test_case_id, name, description, language, framework, status, current_version_id, metadata, created_at, updated_at, archived_at`;
+async function setCurrentTestScriptVersion(client, args) {
+    const result = await client.query({
+        text: exports.setCurrentTestScriptVersionQuery,
+        values: [args.currentVersionId, args.id, args.projectId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        projectId: row[1],
+        testCaseId: row[2],
+        name: row[3],
+        description: row[4],
+        language: row[5],
+        framework: row[6],
+        status: row[7],
+        currentVersionId: row[8],
+        metadata: row[9],
+        createdAt: row[10],
+        updatedAt: row[11],
+        archivedAt: row[12]
+    };
+}
+exports.upsertScriptBindingQuery = `-- name: UpsertScriptBinding :one
+INSERT INTO weave_v1.script_binding (
+    id,
+    test_script_id,
+    binding_key,
+    binding_type,
+    dataset_id,
+    column_name,
+    default_value,
+    is_required,
+    metadata
+)
+VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9
+)
+ON CONFLICT (test_script_id, binding_key) DO UPDATE
+SET binding_type = EXCLUDED.binding_type,
+    dataset_id = EXCLUDED.dataset_id,
+    column_name = EXCLUDED.column_name,
+    default_value = EXCLUDED.default_value,
+    is_required = EXCLUDED.is_required,
+    metadata = EXCLUDED.metadata,
+    updated_at = NOW()
+RETURNING id, test_script_id, binding_key, binding_type, dataset_id, column_name, default_value, is_required, metadata, created_at, updated_at`;
+async function upsertScriptBinding(client, args) {
+    const result = await client.query({
+        text: exports.upsertScriptBindingQuery,
+        values: [args.id, args.testScriptId, args.bindingKey, args.bindingType, args.datasetId, args.columnName, args.defaultValue, args.isRequired, args.metadata],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        testScriptId: row[1],
+        bindingKey: row[2],
+        bindingType: row[3],
+        datasetId: row[4],
+        columnName: row[5],
+        defaultValue: row[6],
+        isRequired: row[7],
+        metadata: row[8],
+        createdAt: row[9],
+        updatedAt: row[10]
+    };
+}
+exports.listScriptBindingsQuery = `-- name: ListScriptBindings :many
+SELECT b.id, b.test_script_id, b.binding_key, b.binding_type, b.dataset_id, b.column_name, b.default_value, b.is_required, b.metadata, b.created_at, b.updated_at
+FROM weave_v1.script_binding b
+         INNER JOIN weave_v1.test_script s ON s.id = b.test_script_id
+WHERE b.test_script_id = $1
+  AND s.project_id = $2
+ORDER BY b.binding_key`;
+async function listScriptBindings(client, args) {
+    const result = await client.query({
+        text: exports.listScriptBindingsQuery,
+        values: [args.testScriptId, args.projectId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            testScriptId: row[1],
+            bindingKey: row[2],
+            bindingType: row[3],
+            datasetId: row[4],
+            columnName: row[5],
+            defaultValue: row[6],
+            isRequired: row[7],
+            metadata: row[8],
+            createdAt: row[9],
+            updatedAt: row[10]
+        };
+    });
+}
+exports.deleteScriptBindingQuery = `-- name: DeleteScriptBinding :exec
+DELETE
+FROM weave_v1.script_binding b
+USING weave_v1.test_script s
+WHERE b.id = $1
+  AND b.test_script_id = $2
+  AND s.id = b.test_script_id
+  AND s.project_id = $3`;
+async function deleteScriptBinding(client, args) {
+    await client.query({
+        text: exports.deleteScriptBindingQuery,
+        values: [args.id, args.testScriptId, args.projectId],
+        rowMode: "array"
+    });
+}