OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "core/dom/DocumentStatisticsCollector.h" | 5 #include "core/dom/DocumentStatisticsCollector.h" |
6 | 6 |
7 #include "core/dom/Document.h" | 7 #include "core/dom/Document.h" |
8 #include "core/frame/FrameView.h" | 8 #include "core/frame/FrameView.h" |
9 #include "core/html/HTMLHeadElement.h" | 9 #include "core/html/HTMLHeadElement.h" |
10 #include "core/html/HTMLLinkElement.h" | 10 #include "core/html/HTMLLinkElement.h" |
11 #include "core/testing/DummyPageHolder.h" | 11 #include "core/testing/DummyPageHolder.h" |
12 #include "public/platform/WebDistillability.h" | 12 #include "public/platform/WebDistillability.h" |
13 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
14 #include "testing/gtest/include/gtest/gtest.h" | 14 #include "testing/gtest/include/gtest/gtest.h" |
15 #include "wtf/text/StringBuilder.h" | 15 #include "wtf/text/StringBuilder.h" |
| 16 #include <memory> |
16 | 17 |
17 namespace blink { | 18 namespace blink { |
18 | 19 |
19 // Saturate the length of a paragraph to save time. | 20 // Saturate the length of a paragraph to save time. |
20 const unsigned kTextContentLengthSaturation = 1000; | 21 const unsigned kTextContentLengthSaturation = 1000; |
21 | 22 |
22 // Filter out short P elements. The threshold is set to around 2 English sentenc
es. | 23 // Filter out short P elements. The threshold is set to around 2 English sentenc
es. |
23 const unsigned kParagraphLengthThreshold = 140; | 24 const unsigned kParagraphLengthThreshold = 140; |
24 | 25 |
25 class DocumentStatisticsCollectorTest : public ::testing::Test { | 26 class DocumentStatisticsCollectorTest : public ::testing::Test { |
26 protected: | 27 protected: |
27 void SetUp() override; | 28 void SetUp() override; |
28 | 29 |
29 void TearDown() override | 30 void TearDown() override |
30 { | 31 { |
31 ThreadHeap::collectAllGarbage(); | 32 ThreadHeap::collectAllGarbage(); |
32 } | 33 } |
33 | 34 |
34 Document& document() const { return m_dummyPageHolder->document(); } | 35 Document& document() const { return m_dummyPageHolder->document(); } |
35 | 36 |
36 void setHtmlInnerHTML(const String&); | 37 void setHtmlInnerHTML(const String&); |
37 | 38 |
38 private: | 39 private: |
39 OwnPtr<DummyPageHolder> m_dummyPageHolder; | 40 std::unique_ptr<DummyPageHolder> m_dummyPageHolder; |
40 }; | 41 }; |
41 | 42 |
42 void DocumentStatisticsCollectorTest::SetUp() | 43 void DocumentStatisticsCollectorTest::SetUp() |
43 { | 44 { |
44 m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600)); | 45 m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600)); |
45 } | 46 } |
46 | 47 |
47 void DocumentStatisticsCollectorTest::setHtmlInnerHTML(const String& htmlContent
) | 48 void DocumentStatisticsCollectorTest::setHtmlInnerHTML(const String& htmlContent
) |
48 { | 49 { |
49 document().documentElement()->setInnerHTML((htmlContent), ASSERT_NO_EXCEPTIO
N); | 50 document().documentElement()->setInnerHTML((htmlContent), ASSERT_NO_EXCEPTIO
N); |
(...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
143 ); | 144 ); |
144 WebDistillabilityFeatures features = DocumentStatisticsCollector::collectSta
tistics(document()); | 145 WebDistillabilityFeatures features = DocumentStatisticsCollector::collectSta
tistics(document()); |
145 | 146 |
146 double error = 1e-5; | 147 double error = 1e-5; |
147 EXPECT_NEAR(features.mozScore, 6 * sqrt(kTextContentLengthSaturation - kPara
graphLengthThreshold), error); | 148 EXPECT_NEAR(features.mozScore, 6 * sqrt(kTextContentLengthSaturation - kPara
graphLengthThreshold), error); |
148 EXPECT_NEAR(features.mozScoreAllSqrt, 6 * sqrt(kTextContentLengthSaturation)
, error); | 149 EXPECT_NEAR(features.mozScoreAllSqrt, 6 * sqrt(kTextContentLengthSaturation)
, error); |
149 EXPECT_NEAR(features.mozScoreAllLinear, 6 * kTextContentLengthSaturation, er
ror); | 150 EXPECT_NEAR(features.mozScoreAllLinear, 6 * kTextContentLengthSaturation, er
ror); |
150 } | 151 } |
151 | 152 |
152 } // namespace blink | 153 } // namespace blink |
OLD | NEW |