Parse JSON in Blink for CopylessPaste.

third_party/WebKit/Source/modules/document_metadata/CopylessPasteExtractorTest.cpp

 // Copyright 2017 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 "modules/document_metadata/CopylessPasteExtractor.h"
 
 #include <memory>
+#include <string>
+#include <utility>
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/testing/DummyPageHolder.h"
+#include "modules/document_metadata/CopylessPasteExtractor.h"
+#include "platform/json/JSONValues.h"
+#include "public/platform/modules/document_metadata/copyless_paste.mojom-blink.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/text/StringBuilder.h"
 
 namespace blink {
 
 namespace {
 
+using mojom::document_metadata::blink::Entity;
+using mojom::document_metadata::blink::EntityPtr;
+using mojom::document_metadata::blink::Property;
+using mojom::document_metadata::blink::PropertyPtr;
+using mojom::document_metadata::blink::Values;
+using mojom::document_metadata::blink::ValuesPtr;
+using mojom::document_metadata::blink::WebPage;
+using mojom::document_metadata::blink::WebPagePtr;
+
 class CopylessPasteExtractorTest : public ::testing::Test {
  public:
-  CopylessPasteExtractorTest()
-      : content_(
-            "\n"
-            "\n"
-            "{\"@type\": \"NewsArticle\","
-            "\"headline\": \"Special characters for ya >_<;\"\n"
-            "}\n"
-            "\n") {}
+  CopylessPasteExtractorTest() {}
 
  protected:
   void SetUp() override;
 
   void TearDown() override { ThreadState::Current()->CollectAllGarbage(); }
 
   Document& GetDocument() const { return dummy_page_holder_->GetDocument(); }
 
-  String Extract() { return CopylessPasteExtractor::Extract(GetDocument()); }
+  WebPagePtr Extract() {
+    return CopylessPasteExtractor::extract(GetDocument());
+  }
 
-  void SetHtmlInnerHTML(const String&);
+  void SetHTMLInnerHTML(const String&);
 
-  String content_;
+  void SetURL(const String&);
+
+  void SetTitle(const String&);
+
+  PropertyPtr createStringProperty(const String& name, const String& value);
+
+  PropertyPtr createBooleanProperty(const String& name, const bool& value);
+
+  PropertyPtr createLongProperty(const String& name, const int64_t& value);
+
+  PropertyPtr createEntityProperty(const String& name, EntityPtr value);
+
+  WebPagePtr createWebPage(const String& url, const String& title);
 
  private:
   std::unique_ptr<DummyPageHolder> dummy_page_holder_;
 };
 
 void CopylessPasteExtractorTest::SetUp() {
   dummy_page_holder_ = DummyPageHolder::Create(IntSize(800, 600));
 }
 
-void CopylessPasteExtractorTest::SetHtmlInnerHTML(const String& html_content) {
+void CopylessPasteExtractorTest::SetHTMLInnerHTML(const String& html_content) {
   GetDocument().documentElement()->setInnerHTML((html_content));
 }
 
+void CopylessPasteExtractorTest::SetURL(const String& url) {
+  GetDocument().SetURL(blink::KURL(blink::kParsedURLString, url));
+}
+
+void CopylessPasteExtractorTest::SetTitle(const String& title) {
+  GetDocument().setTitle(title);
+}
+
+PropertyPtr CopylessPasteExtractorTest::createStringProperty(
+    const String& name,
+    const String& value) {
+  PropertyPtr property = Property::New();
+  property->name = name;
+  property->values = Values::New();
+  property->values->set_string_values({value});
+  return property;
+}
+
+PropertyPtr CopylessPasteExtractorTest::createBooleanProperty(
+    const String& name,
+    const bool& value) {
+  PropertyPtr property = Property::New();
+  property->name = name;
+  property->values = Values::New();
+  property->values->set_bool_values({value});
+  return property;
+}
+
+PropertyPtr CopylessPasteExtractorTest::createLongProperty(
+    const String& name,
+    const int64_t& value) {
+  PropertyPtr property = Property::New();
+  property->name = name;
+  property->values = Values::New();
+  property->values->set_long_values({value});
+  return property;
+}
+
+PropertyPtr CopylessPasteExtractorTest::createEntityProperty(const String& name,
+                                                             EntityPtr value) {
+  PropertyPtr property = Property::New();
+  property->name = name;
+  property->values = Values::New();
+  property->values->set_entity_values(Vector<EntityPtr>());
+  property->values->get_entity_values().push_back(std::move(value));
+  return property;
+}
+
+WebPagePtr CopylessPasteExtractorTest::createWebPage(const String& url,
+                                                     const String& title) {
+  WebPagePtr page = WebPage::New();
+  page->url = blink::KURL(blink::kParsedURLString, url);
+  page->title = title;
+  return page;
+}
+
 TEST_F(CopylessPasteExtractorTest, empty) {
-  String extracted = Extract();
-  String expected = "[]";
-  EXPECT_EQ(expected, extracted);
+  ASSERT_TRUE(Extract().is_null());
 }
 
 TEST_F(CopylessPasteExtractorTest, basic) {
-  SetHtmlInnerHTML(
-      "<body>"
-      "<script type=\"application/ld+json\">" +
-      content_ +
-      "</script>"
-      "</body>");
-
-  String extracted = Extract();
-  String expected = "[" + content_ + "]";
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Special characters for ya >_<;"));
+
+  expected->entities.push_back(std::move(restaurant));
   EXPECT_EQ(expected, extracted);
 }
 
 TEST_F(CopylessPasteExtractorTest, header) {
-  SetHtmlInnerHTML(
+  SetHTMLInnerHTML(
       "<head>"
-      "<script type=\"application/ld+json\">" +
-      content_ +
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
       "</script>"
       "</head>");
 
-  String extracted = Extract();
-  String expected = "[" + content_ + "]";
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Special characters for ya >_<;"));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, booleanValue) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"open\": true"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(createBooleanProperty("open", true));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, longValue) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"long\": 1"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(createLongProperty("long", 1ll));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, doubleValue) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"double\": 1.5"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(createStringProperty("double", "1.5"));
+
+  expected->entities.push_back(std::move(restaurant));
   EXPECT_EQ(expected, extracted);
 }
 
 TEST_F(CopylessPasteExtractorTest, multiple) {
-  SetHtmlInnerHTML(
+  SetHTMLInnerHTML(
       "<head>"
-      "<script type=\"application/ld+json\">" +
-      content_ +
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
       "</script>"
       "</head>"
       "<body>"
-      "<script type=\"application/ld+json\">" +
-      content_ +
-      "</script>"
-      "<script type=\"application/ld+json\">" +
-      content_ +
-      "</script>"
-      "</body>");
-
-  String extracted = Extract();
-  String expected = "[" + content_ + "," + content_ + "," + content_ + "]";
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
+      "</script>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  for (int i = 0; i < 3; ++i) {
+    EntityPtr restaurant = Entity::New();
+    restaurant->type = "Restaurant";
+    restaurant->properties.push_back(
+        createStringProperty("name", "Special characters for ya >_<;"));
+
+    expected->entities.push_back(std::move(restaurant));
+  }
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, nested) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Ye ol greasy diner\","
+      "\"address\": {"
+      "\n"
+      "  \"streetAddress\": \"123 Big Oak Road\","
+      "  \"addressLocality\": \"San Francisco\""
+      "  }\n"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Ye ol greasy diner"));
+
+  EntityPtr address = Entity::New();
+  address->type = "Thing";
+  address->properties.push_back(
+      createStringProperty("streetAddress", "123 Big Oak Road"));
+  address->properties.push_back(
+      createStringProperty("addressLocality", "San Francisco"));
+
+  restaurant->properties.push_back(
+      createEntityProperty("address", std::move(address)));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, repeated) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": [ \"First name\", \"Second name\" ]"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  PropertyPtr name = Property::New();
+  name->name = "name";
+  name->values = Values::New();
+  Vector<String> nameValues;
+  nameValues.push_back("First name");
+  nameValues.push_back("Second name");
+  name->values->set_string_values(nameValues);
+
+  restaurant->properties.push_back(std::move(name));
+
+  expected->entities.push_back(std::move(restaurant));
+
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, repeatedObject) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Ye ol greasy diner\","
+      "\"address\": ["
+      "\n"
+      "  {"
+      "  \"streetAddress\": \"123 Big Oak Road\","
+      "  \"addressLocality\": \"San Francisco\""
+      "  },\n"
+      "  {"
+      "  \"streetAddress\": \"123 Big Oak Road\","
+      "  \"addressLocality\": \"San Francisco\""
+      "  }\n"
+      "]\n"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Ye ol greasy diner"));
+
+  PropertyPtr addressProperty = Property::New();
+  addressProperty->name = "address";
+  addressProperty->values = Values::New();
+  addressProperty->values->set_entity_values(Vector<EntityPtr>());
+  for (int i = 0; i < 2; ++i) {
+    EntityPtr address = Entity::New();
+    address->type = "Thing";
+    address->properties.push_back(
+        createStringProperty("streetAddress", "123 Big Oak Road"));
+    address->properties.push_back(
+        createStringProperty("addressLocality", "San Francisco"));
+    addressProperty->values->get_entity_values().push_back(std::move(address));
+  }
+  restaurant->properties.push_back(std::move(addressProperty));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, truncateLongString) {
+  String maxLengthString;
+  for (int i = 0; i < 200; ++i) {
+    maxLengthString.Append("a");
+  }
+  String tooLongString(maxLengthString);
+  tooLongString.Append("a");
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"" +
+      tooLongString +
+      "\""
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", maxLengthString));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, enforceTypeExists) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_TRUE(extracted.is_null());
+}
+
+TEST_F(CopylessPasteExtractorTest, enforceTypeWhitelist) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"UnsupportedType\","
+      "\"name\": \"Special characters for ya >_<;\""
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_TRUE(extracted.is_null());
+}
+
+TEST_F(CopylessPasteExtractorTest, truncateTooManyValuesInField) {
+  String largeRepeatedField = "[";
+  for (int i = 0; i < 101; ++i) {
+    largeRepeatedField.Append("\"a\"");
+    if (i != 100) {
+      largeRepeatedField.Append(", ");
+    }
+  }
+  largeRepeatedField.Append("]");
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": " +
+      largeRepeatedField +
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  PropertyPtr name = Property::New();
+  name->name = "name";
+  name->values = Values::New();
+  Vector<String> nameValues;
+  for (int i = 0; i < 100; ++i) {
+    nameValues.push_back("a");
+  }
+  name->values->set_string_values(nameValues);
+
+  restaurant->properties.push_back(std::move(name));
+
+  expected->entities.push_back(std::move(restaurant));
+
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, truncateTooManyFields) {
+  String tooManyFields;
+  for (int i = 0; i < 20; ++i) {
+    tooManyFields.Append(String::Format("\"%d\": \"a\"", i));
+    if (i != 19) {
+      tooManyFields.Append(",\n");
+    }
+  }
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\"," +
+      tooManyFields +
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  for (int i = 0; i < 19; ++i) {
+    restaurant->properties.push_back(
+        createStringProperty(String::Number(i), "a"));
+  }
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, ignorePropertyWithEmptyArray) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": []"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  expected->entities.push_back(std::move(restaurant));
+
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, ignorePropertyWithMixedTypes) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": [ \"Name\", 1 ]"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  expected->entities.push_back(std::move(restaurant));
+
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, ignorePropertyWithNestedArray) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": [ [ \"Name\" ] ]"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+
+  expected->entities.push_back(std::move(restaurant));
+
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, enforceMaxNestingDepth) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Ye ol greasy diner\","
+      "\"1\": {"
+      "  \"2\": {"
+      "    \"3\": {"
+      "      \"4\": {"
+      "        \"5\": 6"
+      "      }\n"
+      "    }\n"
+      "  }\n"
+      "}\n"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Ye ol greasy diner"));
+
+  EntityPtr entity1 = Entity::New();
+  entity1->type = "Thing";
+
+  EntityPtr entity2 = Entity::New();
+  entity2->type = "Thing";
+
+  EntityPtr entity3 = Entity::New();
+  entity3->type = "Thing";
+
+  entity2->properties.push_back(createEntityProperty("3", std::move(entity3)));
+
+  entity1->properties.push_back(createEntityProperty("2", std::move(entity2)));
+
+  restaurant->properties.push_back(
+      createEntityProperty("1", std::move(entity1)));
+
+  expected->entities.push_back(std::move(restaurant));
+  EXPECT_EQ(expected, extracted);
+}
+
+TEST_F(CopylessPasteExtractorTest, maxNestingDepthWithTerminalProperty) {
+  SetHTMLInnerHTML(
+      "<body>"
+      "<script type=\"application/ld+json\">"
+      "\n"
+      "\n"
+      "{\"@type\": \"Restaurant\","
+      "\"name\": \"Ye ol greasy diner\","
+      "\"1\": {"
+      "  \"2\": {"
+      "    \"3\": {"
+      "      \"4\": 5"
+      "    }\n"
+      "  }\n"
+      "}\n"
+      "}\n"
+      "\n"
+      "</script>"
+      "</body>");
+  SetURL("http://www.test.com/");
+  SetTitle("My neat website about cool stuff");
+
+  WebPagePtr extracted = Extract();
+  ASSERT_FALSE(extracted.is_null());
+
+  WebPagePtr expected =
+      createWebPage("http://www.test.com/", "My neat website about cool stuff");
+
+  EntityPtr restaurant = Entity::New();
+  restaurant->type = "Restaurant";
+  restaurant->properties.push_back(
+      createStringProperty("name", "Ye ol greasy diner"));
+
+  EntityPtr entity1 = Entity::New();
+  entity1->type = "Thing";
+
+  EntityPtr entity2 = Entity::New();
+  entity2->type = "Thing";
+
+  EntityPtr entity3 = Entity::New();
+  entity3->type = "Thing";
+
+  entity3->properties.push_back(createLongProperty("4", 5));
+
+  entity2->properties.push_back(createEntityProperty("3", std::move(entity3)));
+
+  entity1->properties.push_back(createEntityProperty("2", std::move(entity2)));
+
+  restaurant->properties.push_back(
+      createEntityProperty("1", std::move(entity1)));
+
+  expected->entities.push_back(std::move(restaurant));
   EXPECT_EQ(expected, extracted);
 }
 
 }  // namespace
-
 }  // namespace blink