First draft of a feature tree pane (#4782)

src/lib/operations.test.ts

@@ -0,0 +1,122 @@
+import { defaultRustSourceRange } from 'lang/wasm'
+import { filterOperations } from './operations'
+import { Operation } from 'wasm-lib/kcl/bindings/Operation'
+
+function stdlib(name: string): Operation {
+  return {
+    type: 'StdLibCall',
+    name,
+    unlabeledArg: null,
+    labeledArgs: {},
+    sourceRange: defaultRustSourceRange(),
+    isError: false,
+  }
+}
+
+function userCall(name: string): Operation {
+  return {
+    type: 'UserDefinedFunctionCall',
+    name,
+    functionSourceRange: defaultRustSourceRange(),
+    unlabeledArg: null,
+    labeledArgs: {},
+    sourceRange: defaultRustSourceRange(),
+  }
+}
+function userReturn(): Operation {
+  return {
+    type: 'UserDefinedFunctionReturn',
+  }
+}
+
+describe('operations filtering', () => {
+  it('drops stdlib operations inside a user-defined function call', async () => {
+    const operations = [
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std2'),
+      stdlib('std3'),
+      userReturn(),
+      stdlib('std4'),
+      stdlib('std5'),
+    ]
+    const actual = filterOperations(operations)
+    expect(actual).toEqual([
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std4'),
+      stdlib('std5'),
+    ])
+  })
+  it('drops user-defined function calls that contain no stdlib operations', async () => {
+    const operations = [
+      stdlib('std1'),
+      userCall('foo'),
+      userReturn(),
+      stdlib('std2'),
+      userCall('bar'),
+      userReturn(),
+      stdlib('std3'),
+    ]
+    const actual = filterOperations(operations)
+    expect(actual).toEqual([stdlib('std1'), stdlib('std2'), stdlib('std3')])
+  })
+  it('preserves user-defined function calls at the end of the list', async () => {
+    const operations = [stdlib('std1'), userCall('foo')]
+    const actual = filterOperations(operations)
+    expect(actual).toEqual([stdlib('std1'), userCall('foo')])
+  })
+  it('drops all user-defined function return operations', async () => {
+    // The returns allow us to group operations with the call, but we never
+    // display the returns.
+    const operations = [
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std2'),
+      userReturn(),
+      stdlib('std3'),
+      stdlib('std4'),
+      userCall('foo2'),
+      stdlib('std5'),
+      stdlib('std6'),
+      userReturn(),
+      stdlib('std7'),
+    ]
+    const actual = filterOperations(operations)
+    expect(actual).toEqual([
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std3'),
+      stdlib('std4'),
+      userCall('foo2'),
+      stdlib('std7'),
+    ])
+  })
+  it('correctly filters with nested function calls', async () => {
+    const operations = [
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std2'),
+      userReturn(),
+      stdlib('std3'),
+      stdlib('std4'),
+      userCall('foo2'),
+      stdlib('std5'),
+      userCall('foo3-nested'),
+      stdlib('std6'),
+      userReturn(),
+      stdlib('std7'),
+      userReturn(),
+      stdlib('std8'),
+    ]
+    const actual = filterOperations(operations)
+    expect(actual).toEqual([
+      stdlib('std1'),
+      userCall('foo'),
+      stdlib('std3'),
+      stdlib('std4'),
+      userCall('foo2'),
+      stdlib('std8'),
+    ])
+  })
+})