Split general DocumentMarkerController::AddMarker() method into spelling/grammar versions

third_party/WebKit/Source/core/testing/Internals.cpp

 /*
  * Copyright (C) 2012 Google Inc. All rights reserved.
  * Copyright (C) 2013 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
@@ skipping 903 matching lines @@
   }
 
   WTF::Optional<DocumentMarker::MarkerType> type = MarkerTypeFrom(marker_type);
   if (!type) {
     exception_state.ThrowDOMException(
         kSyntaxError,
         "The marker type provided ('" + marker_type + "') is invalid.");
     return;
   }
 
+  if (type != DocumentMarker::kSpelling && type != DocumentMarker::kGrammar) {
+    exception_state.ThrowDOMException(kSyntaxError,
+                                      "internals.setMarker() currently only "
+                                      "supports spelling and grammar markers; "
+                                      "attempted to add marker of type '" +
+                                          marker_type + "'.");
+    return;
+  }
+
   document->UpdateStyleAndLayoutIgnorePendingStylesheets();
-  document->Markers().AddMarker(range->StartPosition(), range->EndPosition(),
-                                type.value());
+  if (type == DocumentMarker::kSpelling) {
+    document->Markers().AddSpellingMarker(range->StartPosition(),
+                                          range->EndPosition());
+  } else {
+    document->Markers().AddGrammarMarker(range->StartPosition(),
+                                         range->EndPosition());
+  }
 }
 
 unsigned Internals::markerCountForNode(Node* node,
                                        const String& marker_type,
                                        ExceptionState& exception_state) {
   DCHECK(node);
   WTF::Optional<DocumentMarker::MarkerTypes> marker_types =
       MarkerTypesFrom(marker_type);
   if (!marker_types) {
     exception_state.ThrowDOMException(
@@ skipping 2344 matching lines @@
 
 void Internals::crash() {
   CHECK(false) << "Intentional crash";
 }
 
 void Internals::setIsLowEndDevice(bool is_low_end_device) {
   MemoryCoordinator::SetIsLowEndDeviceForTesting(is_low_end_device);
 }
 
 }  // namespace blink