Update objectToDiagnosticsNode to stop failing.

packages/flutter/lib/src/widgets/widget_inspector.dart

@@ -1764,17 +1764,20 @@ mixin WidgetInspectorService {
     // TODO(polina-c): start always assuming Diagnosticable, when DevTools stops sending DiagnosticsNode to
     // APIs that invoke this method.
     // https://github.com/flutter/devtools/issues/3951
-    final Object? theObject = toObject(diagnosticsOrDiagnosticableId);
-    if (theObject is DiagnosticsNode) {
-      return theObject;
-    }
-    if (theObject is Diagnosticable) {
-      return theObject.toDiagnosticsNode();
+    final Object? object = toObject(diagnosticsOrDiagnosticableId);
+    return objectToDiagnosticsNode(object);
+  }
+
+  /// If posiible, returns [DiagnosticsNode] for the object.
+  @visibleForTesting
+  static DiagnosticsNode? objectToDiagnosticsNode(Object? object) {
+    if (object is DiagnosticsNode) {
+      return object;
     }
-    if (theObject == null) {
-      return null;
+    if (object is Diagnosticable) {
+      return object.toDiagnosticsNode();
     }
-    throw StateError('Unexpected object type ${theObject.runtimeType}.');
+    return null;
   }
 
   List<Object> _getChildrenSummaryTree(String? diagnosticsOrDiagnosticableId, String groupName) {

packages/flutter/test/widgets/widget_inspector_test.dart

@@ -261,6 +261,10 @@ class _TestWidgetInspectorService extends TestWidgetInspectorService {
       }
     });
 
+    test ('objectToDiagnosticsNode returns null for non-diagnosticable', () {
+      expect(WidgetInspectorService.objectToDiagnosticsNode(Alignment.bottomCenter), isNull);
+    });
+
     testWidgets('WidgetInspector smoke test', (WidgetTester tester) async {
       // This is a smoke test to verify that adding the inspector doesn't crash.
       await tester.pumpWidget(