[analyzer] Add search tests for null-aware elements

Part of #56989

Change-Id: I124cd1c411a8f193ae4ac59e4f6121cdedd54138
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/412260
Reviewed-by: Brian Wilkerson <[email protected]>
Commit-Queue: Chloe Stefantsova <[email protected]>

Author: chloestefantsova

pkg/analysis_server/test/integration/search/find_element_references_test.dart

@@ -59,6 +59,111 @@ foo(String str) {}
     expect(result.path.first.name, 'f');
   }
 
+  Future<void> test_findReferences_inNullAware_listElement() async {
+    var text = r'''
+List<int> f(int? foo) {
+  return [?foo /* target */];
+}
+''';
+
+    pathname = sourcePath('foo.dart');
+    writeFile(pathname, text);
+    await standardAnalysisSetup();
+    await analysisFinished;
+
+    var results = (await _findElementReferences(text))!;
+    expect(results, hasLength(1));
+    var result = results.first;
+    expect(result.location.file, pathname);
+    expect(result.isPotential, isFalse);
+    expect(result.kind.name, SearchResultKind.READ.name);
+    expect(result.path.first.name, 'f');
+  }
+
+  Future<void> test_findReferences_inNullAware_mapKey() async {
+    var text = r'''
+Map<int, String> f(int? foo) {
+  return {?foo /* target */: "value"};
+}
+''';
+
+    pathname = sourcePath('foo.dart');
+    writeFile(pathname, text);
+    await standardAnalysisSetup();
+    await analysisFinished;
+
+    var results = (await _findElementReferences(text))!;
+    expect(results, hasLength(1));
+    var result = results.first;
+    expect(result.location.file, pathname);
+    expect(result.isPotential, isFalse);
+    expect(result.kind.name, SearchResultKind.READ.name);
+    expect(result.path.first.name, 'f');
+  }
+
+  Future<void> test_findReferences_inNullAware_mapKeyValue() async {
+    var text = r'''
+Map<int, String> f(int? key, String? value) {
+  return {?key: ?value /* target */};
+}
+''';
+
+    pathname = sourcePath('foo.dart');
+    writeFile(pathname, text);
+    await standardAnalysisSetup();
+    await analysisFinished;
+
+    var results = (await _findElementReferences(text))!;
+    expect(results, hasLength(1));
+    var result = results.first;
+    expect(result.location.file, pathname);
+    expect(result.isPotential, isFalse);
+    expect(result.kind.name, SearchResultKind.READ.name);
+    expect(result.path.first.name, 'f');
+  }
+
+  Future<void> test_findReferences_inNullAware_mapValue() async {
+    var text = r'''
+Map<String, int> f(int? foo) {
+  return {"key": ?foo /* target */};
+}
+''';
+
+    pathname = sourcePath('foo.dart');
+    writeFile(pathname, text);
+    await standardAnalysisSetup();
+    await analysisFinished;
+
+    var results = (await _findElementReferences(text))!;
+    expect(results, hasLength(1));
+    var result = results.first;
+    expect(result.location.file, pathname);
+    expect(result.isPotential, isFalse);
+    expect(result.kind.name, SearchResultKind.READ.name);
+    expect(result.path.first.name, 'f');
+  }
+
+  Future<void> test_findReferences_inNullAware_setElement() async {
+    var text = r'''
+Set<int> f(int? foo) {
+  return {?foo /* target */};
+}
+''';
+
+    pathname = sourcePath('foo.dart');
+    writeFile(pathname, text);
+    await standardAnalysisSetup();
+    await analysisFinished;
+
+    var results = (await _findElementReferences(text))!;
+    expect(results, hasLength(1));
+    var result = results.first;
+    expect(result.location.file, pathname);
+    expect(result.isPotential, isFalse);
+    expect(result.kind.name, SearchResultKind.READ.name);
+    expect(result.path.first.name, 'f');
+  }
+
   Future<List<SearchResult>?> _findElementReferences(String text) async {
     var offset = text.indexOf(' /* target */') - 1;
     var result = await sendSearchFindElementReferences(pathname, offset, false);

pkg/analysis_server/test/search/element_references_test.dart

@@ -921,6 +921,32 @@ void f() {
     assertHasResult(SearchResultKind.READ, 'vvv();');
   }
 
+  Future<void> test_localVariable_inNullAwareElements() async {
+    await setPriorityFiles2([testFile]);
+    addTestFile('''
+void f(bool t) {
+  int? vvv = t ? 0 : null;
+  List<int> list = [?vvv];
+  Set<int> set = {?vvv};
+  Map<int, String> mapKey = {?vvv: "value"};
+  Map<String, int> mapValue = {"key": ?vvv, /* mapValue */};
+  Map<int, int> mapKeyValue1 = {?vvv: vvv, /* mapKeyValue1 */}; // Using promotion.
+  Map<int, int> mapKeyValue2 = {?vvv: ?vvv, /* mapKeyValue2 */}; // Ignoring promotion.
+}
+''');
+    await findElementReferences(search: 'vvv = t ? 0 : null', false);
+    expect(searchElement!.kind, ElementKind.LOCAL_VARIABLE);
+    expect(results, hasLength(8));
+    assertHasResult(SearchResultKind.READ, 'vvv];');
+    assertHasResult(SearchResultKind.READ, 'vvv};');
+    assertHasResult(SearchResultKind.READ, 'vvv: "value"');
+    assertHasResult(SearchResultKind.READ, 'vvv, /* mapValue */};');
+    assertHasResult(SearchResultKind.READ, 'vvv: vvv');
+    assertHasResult(SearchResultKind.READ, 'vvv, /* mapKeyValue1 */};');
+    assertHasResult(SearchResultKind.READ, 'vvv: ?vvv');
+    assertHasResult(SearchResultKind.READ, 'vvv, /* mapKeyValue2 */};');
+  }
+
   Future<void> test_mixin() async {
     addTestFile('''
 mixin A {}