pkg/analysis_server/test/lsp/completion.dart

// Copyright (c) 2021, the Dart project authors. Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

import 'package:analysis_server/lsp_protocol/protocol_generated.dart';
import 'package:test/test.dart';

import 'server_abstract.dart';

mixin CompletionTestMixin on AbstractLspAnalysisServerTest {
  Future<void> verifyCompletions(
    Uri fileUri,
    String content, {
    List<String> expectCompletions,
    String verifyEditsFor,
    String expectedContent,
    String expectedContentIfInserting,
    bool verifyInsertReplaceRanges = false,
  }) async {
    // If verifyInsertReplaceRanges is true, we need both expected contents.
    assert(verifyInsertReplaceRanges == false ||
        (expectedContent != null && expectedContentIfInserting != null));

    if (!initialized) {
      var textDocCapabilities =
          withCompletionItemSnippetSupport(emptyTextDocumentClientCapabilities);

      if (verifyInsertReplaceRanges) {
        textDocCapabilities =
            withCompletionItemInsertReplaceSupport(textDocCapabilities);
      }
      await initialize(textDocumentCapabilities: textDocCapabilities);
    }

    await openFile(fileUri, withoutMarkers(content));
    final res = await getCompletion(fileUri, positionFromMarker(content));
    await closeFile(fileUri);

    for (final expectedCompletion in expectCompletions) {
      expect(
        res.any((c) => c.label == expectedCompletion),
        isTrue,
        reason:
            '"$expectedCompletion" was not in ${res.map((c) => '"${c.label}"')}',
      );
    }

    // Check the edits apply correctly.
    if (verifyEditsFor != null) {
      final item = res.singleWhere((c) => c.label == verifyEditsFor);
      final insertFormat = item.insertTextFormat;

      if (verifyInsertReplaceRanges &&
          expectedContent != expectedContentIfInserting) {
        // Replacing.
        final replaced = applyTextEdits(
          withoutMarkers(content),
          [textEditForReplace(item.textEdit)],
        );
        expect(withCaret(replaced, insertFormat), equals(expectedContent));

        // Inserting.
        final inserted = applyTextEdits(
          withoutMarkers(content),
          [textEditForInsert(item.textEdit)],
        );
        expect(withCaret(inserted, insertFormat),
            equals(expectedContentIfInserting));
      } else {
        final updated = applyTextEdits(
          withoutMarkers(content),
          [toTextEdit(item.textEdit)],
        );
        expect(withCaret(updated, insertFormat), equals(expectedContent));
      }
    }
  }

  /// Replaces the LSP snippet placeholder '${0:}' with '^' for easier verifying
  /// of the cursor position in completions.
  String withCaret(String contents, InsertTextFormat format) =>
      format == InsertTextFormat.Snippet
          ? contents.replaceFirst(r'${0:}', '^')
          : contents;
}