Introduce EventSourceParser

third_party/WebKit/Source/core/page/EventSourceTest.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/page/EventSource.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "bindings/core/v8/ScriptState.h"
 #include "bindings/core/v8/ScriptValue.h"
 #include "bindings/core/v8/SerializedScriptValue.h"
 #include "bindings/core/v8/V8Binding.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/dom/MessagePort.h"
 #include "core/events/Event.h"
 #include "core/events/EventListener.h"
 #include "core/events/MessageEvent.h"
 #include "core/loader/MockThreadableLoader.h"
 #include "core/page/EventSourceInit.h"
 #include "core/testing/DummyPageHolder.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "wtf/text/CharacterNames.h"
 
 #include <string.h>
 #include <v8.h>
 
 namespace blink {
 
 namespace {
 
 String dataAsString(ScriptState* scriptState, MessageEvent* event)
 {
@@ skipping 31 matching lines @@
 };
 
 class EventSourceTest : public ::testing::Test {
 protected:
     EventSourceTest()
         : m_page(DummyPageHolder::create())
         , m_source(EventSource::create(&document(), "https://localhost/", EventSourceInit(), m_exceptionState))
     {
         source()->setStateForTest(EventSource::OPEN);
         source()->setThreadableLoaderForTest(MockThreadableLoader::create());
+        source()->setParser(new EventSourceParser("https://localhost/", source()));
+        source()->setRequestInFlightForTest(true);
     }
     ~EventSourceTest() override
     {
         source()->setThreadableLoaderForTest(nullptr);
         source()->close();
 
         // We need this because there is
         // listener -> event -> source -> listener
         // reference cycle on non-oilpan build.
         m_source->removeAllEventListeners();
@@ skipping 334 matching lines @@
     EXPECT_EQ("4", bEvent->lastEventId());
 
     ASSERT_EQ(1u, messageListener->events().size());
     ASSERT_EQ("message", messageListener->events()[0]->type());
     RefPtrWillBeRawPtr<MessageEvent> messageEvent = static_cast<MessageEvent*>(messageListener->events()[0].get());
     ASSERT_EQ(MessageEvent::DataTypeSerializedScriptValue, messageEvent->dataType());
     EXPECT_EQ("bye", dataAsString(scriptState(), messageEvent.get()));
     EXPECT_EQ("8", messageEvent->lastEventId());
 }
 
+TEST_F(EventSourceTest, InvalidUTF8Sequence)
+{
+    RefPtrWillBeRawPtr<FakeEventListener> listener = FakeEventListener::create();
+
+    source()->addEventListener("message", listener);
+    enqueue("data:\xffhello\xc2\n\n");
+
+    ASSERT_EQ(1u, listener->events().size());
+    ASSERT_EQ("message", listener->events()[0]->type());
+    RefPtrWillBeRawPtr<MessageEvent> event = static_cast<MessageEvent*>(listener->events()[0].get());
+    ASSERT_EQ(MessageEvent::DataTypeSerializedScriptValue, event->dataType());
+    String expected = String() + replacementCharacter + "hello" + replacementCharacter;
+    EXPECT_EQ(expected, dataAsString(scriptState(), event.get()));
+}
+
 } // namespace
 
 } // namespace blink