[analyzer] Add tests for null-aware elements and color codes

Part of #56989

Change-Id: I90895dde478eb85ce8cc653c32fe32fa45b49dc8
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/397904
Commit-Queue: Chloe Stefantsova <[email protected]>
Reviewed-by: Keerti Parthasarathy <[email protected]>

Author: chloestefantsova

pkg/analysis_server/test/src/computer/color_computer_test.dart

@@ -90,10 +90,10 @@ class ColorComputerTest extends AbstractContextTest {
     // in non-const contexts.
     'const CupertinoDynamicColor.withBrightness(color: CupertinoColors.white, darkColor: CupertinoColors.black)':
         {
-          null: 0xFFFFFFFF,
-          'CupertinoColors.white': 0xFFFFFFFF,
-          'CupertinoColors.black': 0xFF000000,
-        },
+      null: 0xFFFFFFFF,
+      'CupertinoColors.white': 0xFFFFFFFF,
+      'CupertinoColors.black': 0xFF000000,
+    },
   };
 
   late String testPath;
@@ -156,8 +156,7 @@ class ColorComputerTest extends AbstractContextTest {
     expect(
       colors,
       hasLength(expectedColorValues.length),
-      reason:
-          '${expectedColorValues.length} colors should be detected in:\n'
+      reason: '${expectedColorValues.length} colors should be detected in:\n'
           '$dartCode',
     );
 
@@ -181,10 +180,10 @@ class ColorComputerTest extends AbstractContextTest {
       );
 
       void expectComponent(int actual, int expected, String name) => expect(
-        actual,
-        expected,
-        reason: '$name value for $expectedColorCode is not correct',
-      );
+            actual,
+            expected,
+            reason: '$name value for $expectedColorCode is not correct',
+          );
 
       expectComponent(color.color.alpha, expectedAlpha, 'Alpha');
       expectComponent(color.color.red, expectedRed, 'Red');
@@ -195,10 +194,9 @@ class ColorComputerTest extends AbstractContextTest {
 
   void expectNoErrors(ResolvedUnitResult result) {
     // If the test code has errors, generate a suitable failure to help debug.
-    var errors =
-        result.errors
-            .where((error) => error.severity == Severity.error)
-            .toList();
+    var errors = result.errors
+        .where((error) => error.severity == Severity.error)
+        .toList();
     if (errors.isNotEmpty) {
       throw 'Code has errors: $errors\n\n${result.content}';
     }
@@ -260,12 +258,15 @@ class MyTheme {
         instanceMaterialRedAccent = Colors.redAccent;
 }
 ''';
-    await expectColors(testCode, {
-      'MyTheme.staticWhite': 0xFFFFFFFF,
-      'MyTheme.staticMaterialRedAccent': 0xFFFFAA00,
-      'theme.instanceWhite': 0xFFFFFFFF,
-      'theme.instanceMaterialRedAccent': 0xFFFFAA00,
-    }, otherCode: otherCode);
+    await expectColors(
+        testCode,
+        {
+          'MyTheme.staticWhite': 0xFFFFFFFF,
+          'MyTheme.staticMaterialRedAccent': 0xFFFFAA00,
+          'theme.instanceWhite': 0xFFFFFFFF,
+          'theme.instanceMaterialRedAccent': 0xFFFFAA00,
+        },
+        otherCode: otherCode);
   }
 
   Future<void> test_customConst_initializer() async {
@@ -282,10 +283,13 @@ void f() {
 const myThemeWhite = Colors.white;
 const myThemeMaterialRedAccent = Colors.redAccent;
 ''';
-    await expectColors(testCode, {
-      'myThemeWhite': 0xFFFFFFFF,
-      'myThemeMaterialRedAccent': 0xFFFFAA00,
-    }, otherCode: otherCode);
+    await expectColors(
+        testCode,
+        {
+          'myThemeWhite': 0xFFFFFFFF,
+          'myThemeMaterialRedAccent': 0xFFFFAA00,
+        },
+        otherCode: otherCode);
   }
 
   Future<void> test_local_const() async {
@@ -352,6 +356,100 @@ void f() {
     await checkAllColors(testCode);
   }
 
+  Future<void> test_nullAwareElement_inList_const() async {
+    const testCode = '''
+void f() {
+  const colors = [
+    ?[[COLOR]],
+  ];
+}
+''';
+    await checkAllColors(testCode, onlyConst: true);
+  }
+
+  Future<void> test_nullAwareElement_inList_nonConst() async {
+    const testCode = '''
+void f() {
+  final colors = [
+    ?[[COLOR]],
+  ];
+}
+''';
+    await checkAllColors(testCode);
+  }
+
+  Future<void> test_nullAwareElement_inSet_const() async {
+    // In the following test case the 'Color' object is placed inside of a list
+    // literal, since the 'Color' class defines 'operator==', and its objects
+    // can't be elements of constant sets.
+    const testCode = '''
+void f() {
+  const colors = {
+    ?[ [[COLOR]] ],
+  };
+}
+''';
+    await checkAllColors(testCode, onlyConst: true);
+  }
+
+  Future<void> test_nullAwareElement_inSet_nonConst() async {
+    const testCode = '''
+void f() {
+  final colors = {
+    ?[[COLOR]],
+  };
+}
+''';
+    await checkAllColors(testCode);
+  }
+
+  Future<void> test_nullAwareKey_inMap_const() async {
+    // In the following test case the 'Color' object is placed inside of a list
+    // literal, since the 'Color' class defines 'operator==', and its objects
+    // can't be keys of constant maps.
+    const testCode = '''
+void f() {
+  const colors = {
+    ?[ [[COLOR]] ]: "value",
+  };
+}
+''';
+    await checkAllColors(testCode, onlyConst: true);
+  }
+
+  Future<void> test_nullAwareKey_inMap_nonConst() async {
+    const testCode = '''
+void f() {
+  final colors = {
+    ?[[COLOR]]: "value",
+  };
+}
+''';
+    await checkAllColors(testCode);
+  }
+
+  Future<void> test_nullAwareValue_inMap_const() async {
+    const testCode = '''
+void f() {
+  const colors = {
+    "key": ?[[COLOR]],
+  };
+}
+''';
+    await checkAllColors(testCode, onlyConst: true);
+  }
+
+  Future<void> test_nullAwareValue_inMap_nonConst() async {
+    const testCode = '''
+void f() {
+  final colors = {
+    "key": ?[[COLOR]],
+  };
+}
+''';
+    await checkAllColors(testCode);
+  }
+
   Future<void> test_topLevel_const() async {
     const testCode = '''
 const a = [[COLOR]];