test: testing java method snippet arguments (related to flix/flix#8255)

AUTHORS.md

@@ -10,4 +10,5 @@ By adding your name to this document, you agree to release all your contribution
 - [Magnus Holm Rasmussen](https://github.com/mR4smussen)
 - [Holger Dal Mogensen](https://github.com/sockmaster27)
 - [Jakob Schneider Villumsen](https://github.com/jaschdoc)
+- [Alexander Dybdahl Troelsen](https://github.com/LoZander)
 

test/src/completions.test.ts

@@ -16,15 +16,53 @@
 
 import * as assert from 'assert'
 import * as vscode from 'vscode'
-import { getTestDocUri, init, open, typeText, addFile } from './util'
+import { getTestDocUri, init, open, typeText, addFile, deleteFile, copyFile } from './util'
 
 suite('Completions', () => {
   const docUri = getTestDocUri('src/Temp.flix')
+  let tempDocUri: vscode.Uri | null = null
 
   suiteSetup(async () => {
     await init('completions')
   })
 
+  teardown(async () => {
+    if (tempDocUri !== null) {
+      await deleteFile(tempDocUri)
+    }
+  })
+
+  /**
+   * Determines if there is a completion suggestion at the `cursor` in the file `src/fileName`.
+   *
+   * @param fileName name of flix file we're testing. Must include the `.flix` extension.
+   * @param cursor the position of the cursor.
+   * @param predicate the predicate by which we determine if a completion is valid.
+   * @returns `true` if there is a completion suggestion at `cursor` which satisfies `predicate`.
+   */
+  async function validCompletionExists(
+    fileName: string,
+    cursor: vscode.Position,
+    predicate: (item: vscode.CompletionItem) => boolean,
+  ): Promise<boolean> {
+    // Setup file
+    const latentUri = getTestDocUri(`latent/${fileName}`)
+    const srcUri = getTestDocUri(`src/${fileName}`)
+
+    tempDocUri = srcUri
+    await copyFile(latentUri, srcUri)
+
+    await open(srcUri)
+
+    const r = await vscode.commands.executeCommand<vscode.CompletionList>(
+      'vscode.executeCompletionItemProvider',
+      srcUri,
+      cursor,
+    )
+
+    return r.items.some(predicate)
+  }
+
   test('Should propose completing mod', async () => {
     await addFile(docUri, '')
     await open(docUri)
@@ -42,4 +80,36 @@ suite('Completions', () => {
       true,
     )
   })
+
+  suite('Java Methods', () => {
+    test('Should complete Java static method', async () => {
+      const cursor = new vscode.Position(5 - 1, 23 - 1)
+      const pred = item => item.label === 'floor(arg0: double): double \\ IO'
+      assert.strictEqual(await validCompletionExists('JavaMath.flix', cursor, pred), true)
+    })
+
+    test('Should complete Java method with snippet placeholder arg for static method', async () => {
+      const cursor = new vscode.Position(5 - 1, 23 - 1)
+      const pred = item => (item.insertText as vscode.SnippetString).value === 'floor(${1:arg0})'
+      assert.strictEqual(await validCompletionExists('JavaMath.flix', cursor, pred), true)
+    })
+
+    test('Should complete Java method with snippet placeholder args for static method', async () => {
+      const cursor = new vscode.Position(9 - 1, 23 - 1)
+      const pred = item => (item.insertText as vscode.SnippetString).value === 'max(${1:arg0}, ${2:arg1})'
+      assert.strictEqual(await validCompletionExists('JavaMath.flix', cursor, pred), true)
+    })
+
+    test('Should complete Java non-static method', async () => {
+      const cursor = new vscode.Position(8 - 1, 18 - 1)
+      const pred = item => item.label === 'append(arg0: String): StringBuilder \\ IO'
+      assert.strictEqual(await validCompletionExists('JavaStringBuilder.flix', cursor, pred), true)
+    })
+
+    test('Should complete Java method with snipper placeholder arg for non-static method', async () => {
+      const cursor = new vscode.Position(8 - 1, 18 - 1)
+      const pred = item => (item.insertText as vscode.SnippetString).value === 'append(${1:arg0})'
+      assert.strictEqual(await validCompletionExists('JavaStringBuilder.flix', cursor, pred), true)
+    })
+  })
 })

test/testWorkspaces/completions/latent/JavaMath.flix

@@ -0,0 +1,11 @@
+mod JavaMath {
+    import java.lang.Math;
+
+    pub def floor(x: Float64): Float64 \ IO =
+        let r = Math.f
+        r
+
+    pub def max(x: Float64): Float64 \ IO =
+        let r = Math.m
+        r
+}

test/testWorkspaces/completions/latent/JavaStringBuilder.flix

@@ -0,0 +1,12 @@
+mod JavaStringBuilder {
+    import java.lang.StringBuilder;
+
+    pub def create(): StringBuilder \ IO =
+        new StringBuilder()
+
+    pub def append(s: String, builder: StringBuilder): StringBuilder \ IO =
+        builder.a
+
+    pub def toString(builder: StringBuilder): String \ IO =
+        builder.toString()        
+}