pkg/front_end/testcases/nnbd/null_shorting_explicit_extension.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.

class Class {
  Class? _field;
}

extension Extension on Class {
  Class? get field => _field;
  void set field(Class? value) {
    _field = value;
  }

  Class? method() => field;

  Class? operator [](Class? key) => field;
  void operator []=(Class? key, Class? value) {
    field = value;
  }

  Class? operator +(int value) => field;
}

main() {
  propertyAccess(null);
  indexAccess(null);
  operatorAccess(null);
  ifNull(null);
}

void propertyAccess(Class? c) {
  Extension(c)?.field;
  Extension(c)?.field = new Class();
  c = Extension(c)?.field = new Class();
  Extension(c)?.method();

  Extension(c)?.field.field;
  Extension(c)?.field.field = new Class();
  c = Extension(c)?.field.field = new Class();
  Extension(c)?.field.method();
  Extension(c)?.field = new Class().field;
  c = Extension(c)?.field = new Class().field;
  Extension(c)?.field = new Class().field = new Class();
  c = Extension(c)?.field = new Class().field = new Class();
  Extension(c)?.field = new Class().method();
  c = Extension(c)?.field = new Class().method();
  Extension(c)?.method().field;
  Extension(c)?.method().field = new Class();
  Extension(c)?.method().method();

  Extension(c)?.field.field.field;
  Extension(c)?.field.field.field = new Class();
  c = Extension(c)?.field.field.field = new Class();
  Extension(c)?.field.field.method();
  Extension(c)?.field = new Class().field.field;
  c = Extension(c)?.field = new Class().field.field;
  Extension(c)?.field = new Class().field.field = new Class();
  c = Extension(c)?.field = new Class().field.field = new Class();
  Extension(c)?.field = new Class().field.method();
  c = Extension(c)?.field = new Class().field.method();
  Extension(c)?.method().field.field;
  Extension(c)?.method().field.field = new Class();
  Extension(c)?.method().field.method();

  Extension(c)?.field.field = new Class().field;
  c = Extension(c)?.field.field = new Class().field;
  Extension(c)?.field.field = new Class().field = new Class();
  c = Extension(c)?.field.field = new Class().field = new Class();
  Extension(c)?.field.field = new Class().method();
  c = Extension(c)?.field.field = new Class().method();
  Extension(c)?.field = new Class().field = new Class().field;
  c = Extension(c)?.field = new Class().field = new Class().field;
  Extension(c)?.field = new Class().field = new Class().field = new Class();
  c = Extension(c)?.field = new Class().field = new Class().field = new Class();
  Extension(c)?.field = new Class().field = new Class().method();
  c = Extension(c)?.field = new Class().field = new Class().method();
  Extension(c)?.method().field = new Class().field;
  c = Extension(c)?.method().field = new Class().field;
  Extension(c)?.method().field = new Class().field = new Class();
  c = Extension(c)?.method().field = new Class().field = new Class();
  Extension(c)?.method().field = new Class().method();
  c = Extension(c)?.method().field = new Class().method();

  Extension(c)?.field.method().field;
  Extension(c)?.field.method().field = new Class();
  c = Extension(c)?.field.method().field = new Class();
  Extension(c)?.field.method().method();
  Extension(c)?.field = new Class().method().field;
  c = Extension(c)?.field = new Class().method().field;
  Extension(c)?.field = new Class().method().field = new Class();
  c = Extension(c)?.field = new Class().method().field = new Class();
  Extension(c)?.field = new Class().method().method();
  c = Extension(c)?.field = new Class().method().method();
  Extension(c)?.method().method().field;
  Extension(c)?.method().method().field = new Class();
  Extension(c)?.method().method().method();
}

void indexAccess(Class? c) {
  // TODO(johnniwinther): Handle null aware index access.
  //Extension(c)?.[c];
  //Extension(c)?.[c] = new Class();
  //Extension(c)?.[c].method();
  Extension(c)?.field[c];
  Extension(c)?.field[c] = new Class();
  c = Extension(c)?.field[c] = new Class();
  Extension(c)?.field[c].method();
  Extension(c)?.field[c] += 0;
  c = Extension(c)?.field[c] += 0;
  // TODO(johnniwinther): ++ should probably not be null-shorted, awaiting spec
  // update.
  Extension(c)?.field[c]++;
  c = Extension(c)?.field[c]++;
  ++Extension(c)?.field[c];
  c = ++Extension(c)?.field[c];
}

void operatorAccess(Class? c) {
  // TODO(johnniwinther): + should _not_ be null-shortened.
  Extension(c)?.field + 0;
  Extension(c)?.field += 0;
  c = Extension(c)?.field += 0;
  Extension(c)?.field.field += 0;
  c = Extension(c)?.field.field += 0;
  // TODO(johnniwinther): ++ should probably not be null-shorted, awaiting spec
  // update.
  Extension(c)?.field++;
  c = Extension(c)?.field++;
  ++Extension(c)?.field;
  c = ++Extension(c)?.field;
}

void ifNull(Class? c) {
  Extension(c)?.field ??= c;
  c = Extension(c)?.field ??= c;
  Extension(c)?.field[c] ??= c;
}