QueryResult to RPG data structure test

Author: janfh

src/views/results/codegen.test.ts

@@ -1,5 +1,6 @@
 import { assert, expect, test } from 'vitest'
-import { columnToRpgDefinition } from './codegen';
+import { columnToRpgDefinition, queryResultToRpgDs } from './codegen';
+import { QueryResult } from '@ibm/mapepire-js';
 
 test('Column to RPG definition', () => {
     let rpgdef;
@@ -41,3 +42,51 @@ test('Column to RPG definition', () => {
     expect(rpgdef).toBe('// type:SOME_UNKNOWN_TYPE precision:0 scale:0');
 });
 
+test('QueryResult to RPG data structure', () => {
+    const queryResult: QueryResult<any> = {
+        metadata: {
+            column_count: 3,
+            columns: [
+                {
+                    display_size: 0,
+                    label: 'id',
+                    name: 'id',
+                    type: 'INTEGER',
+                    precision: 0,
+                    scale: 0
+                },
+                {
+                    display_size: 0,
+                    label: 'name',
+                    name: 'name',
+                    type: 'VARCHAR',
+                    precision: 80,
+                    scale: 0
+                },
+                {
+                    display_size: 0,
+                    label: 'salary',
+                    name: 'salary',
+                    type: 'DECIMAL',
+                    precision: 13,
+                    scale: 2
+                },
+            ]
+        },
+        is_done: true,
+        has_results: true,
+        update_count: 0,
+        data: [],
+        id: '',
+        success: true,
+        sql_rc: 0,
+        sql_state: '',
+        execution_time: 0
+    };
+    const ds = queryResultToRpgDs(queryResult);
+    const lines = ds.split('\n').filter(l => l !== '');
+    expect(lines.length).toBe(5);
+    expect(lines.at(0)).toBe('dcl-ds row_t qualified template;');
+    expect(lines.at(4)).toBe('end-ds;');
+});
+