pkg/analysis_server/test/src/edit/nnbd_migration/offset_mapper_test.dart

// Copyright (c) 2019, 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/src/edit/nnbd_migration/offset_mapper.dart';
import 'package:analysis_server/src/protocol_server.dart';
import 'package:test/test.dart';
import 'package:test_reflective_loader/test_reflective_loader.dart';

import '../../../analysis_abstract.dart';

void main() {
  defineReflectiveSuite(() {
    defineReflectiveTests(OffsetMapperTest);
  });
}

@reflectiveTest
class OffsetMapperTest extends AbstractAnalysisTest {
  void test_identity() {
    var mapper = OffsetMapper.identity;
    expect(mapper.map(0), 0);
    expect(mapper.map(20), 20);
    expect(mapper.map(0xFFFFFF), 0xFFFFFF);
  }

  void test_insertMapper() {
    var mapper = OffsetMapper.forInsertion(10, 5);
    expect(mapper.map(0), 0);
    expect(mapper.map(9), 9);
    expect(mapper.map(10), 15);
    expect(mapper.map(11), 16);
    expect(mapper.map(20), 25);
  }

  void test_multipleEdits() {
    var mapper = OffsetMapper.forEdits([
      SourceEdit(13, 0, '?'),
      SourceEdit(21, 0, '!'),
      SourceEdit(32, 0, '?'),
    ]);
    expect(mapper.map(0), 0);
    expect(mapper.map(13), 13);
    expect(mapper.map(14), 15);
    expect(mapper.map(21), 22);
    expect(mapper.map(22), 24);
    expect(mapper.map(32), 34);
    expect(mapper.map(33), 36);
    expect(mapper.map(55), 58);
  }

  void test_rebase_insertMapper() {
    var mapper = OffsetMapper.rebase(
        OffsetMapper.forInsertion(5, 5), OffsetMapper.forInsertion(10, 5));
    expect(mapper.map(0), 0);
    expect(mapper.map(4), 4);
    expect(mapper.map(5), 10);
    expect(mapper.map(6), 11);
    expect(mapper.map(9), 14);
    expect(mapper.map(10), 20);
    expect(mapper.map(11), 21);
    expect(mapper.map(12), 22);
  }

  void test_sequence_insertMappers() {
    var mapper = OffsetMapper.sequence(
        OffsetMapper.forInsertion(30, 10), OffsetMapper.forInsertion(10, 10));
    expect(mapper.map(0), 0);
    expect(mapper.map(9), 9);
    expect(mapper.map(10), 20);
    expect(mapper.map(11), 21);
    expect(mapper.map(29), 39);
    expect(mapper.map(30), 50);
    expect(mapper.map(31), 51);
  }

  void test_sequence_insertMappers_firstBeforeSecond() {
    var mapper = OffsetMapper.sequence(
        OffsetMapper.forInsertion(10, 10), OffsetMapper.forInsertion(30, 10));
    expect(mapper.map(0), 0);
    expect(mapper.map(9), 9);
    expect(mapper.map(10), 20);
    expect(mapper.map(11), 21);
    expect(mapper.map(19), 29);
    expect(mapper.map(20), 40);
    expect(mapper.map(21), 41);
  }

  void test_sequence_insertMappers_overlapping() {
    // by inserting into the middle of a previous insertion, we just effectively
    // make the first insertion longer.
    var mapper = OffsetMapper.sequence(
        OffsetMapper.forInsertion(10, 10), OffsetMapper.forInsertion(15, 5));
    expect(mapper.map(0), 0);
    expect(mapper.map(9), 9);
    expect(mapper.map(10), 25);
    expect(mapper.map(11), 26);
    expect(mapper.map(20), 35);
  }

  void test_singleEdit() {
    var mapper = OffsetMapper.forEdits([
      SourceEdit(13, 0, '?'),
    ]);
    expect(mapper.map(0), 0);
    expect(mapper.map(13), 13);
    expect(mapper.map(14), 15);
  }
}