Manage registering composited elements in PaintArtifactCompositor.

third_party/WebKit/Source/platform/graphics/compositing/PaintArtifactCompositorTest.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 "platform/graphics/compositing/PaintArtifactCompositor.h"
 
 #include <memory>
 
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
@@ skipping 78 matching lines @@
     web_layer_tree_view_ =
         WTF::MakeUnique<WebLayerTreeViewWithCompositorFrameSink>(settings);
     web_layer_tree_view_->SetRootLayer(
         *paint_artifact_compositor_->GetWebLayer());
   }
 
   const cc::PropertyTrees& GetPropertyTrees() {
     return *web_layer_tree_view_->GetLayerTreeHost()->property_trees();
   }
 
+  const cc::LayerTreeHost& GetLayerTreeHost() {
+    return *web_layer_tree_view_->GetLayerTreeHost();
+  }
+
   int ElementIdToEffectNodeIndex(CompositorElementId element_id) {
     return web_layer_tree_view_->GetLayerTreeHost()
         ->property_trees()
         ->element_id_to_effect_node_index[element_id];
   }
 
   int ElementIdToTransformNodeIndex(CompositorElementId element_id) {
     return web_layer_tree_view_->GetLayerTreeHost()
         ->property_trees()
         ->element_id_to_transform_node_index[element_id];
@@ skipping 2150 matching lines @@
                                        tinyEffect);
   TestPaintArtifact artifact;
   artifact
       .Chunk(TransformPaintPropertyNode::Root(), ClipPaintPropertyNode::Root(),
              visibleEffect)
       .RectDrawing(FloatRect(0, 0, 100, 100), Color::kBlack);
   Update(artifact.Build());
   ASSERT_EQ(0u, ContentLayerCount());
 }
 
+TEST_F(PaintArtifactCompositorTestWithPropertyTrees,
+       UpdateManagesLayerElementIds) {
+  RefPtr<TransformPaintPropertyNode> transform =
+      CreateSampleTransformNodeWithElementId();
+  CompositorElementId element_id = transform->GetCompositorElementId();
+
+  {
+    TestPaintArtifact artifact;
+    artifact
+        .Chunk(transform, ClipPaintPropertyNode::Root(),
+               EffectPaintPropertyNode::Root())
+        .RectDrawing(FloatRect(0, 0, 100, 100), Color::kBlack);
+
+    Update(artifact.Build());
+    ASSERT_EQ(1u, ContentLayerCount());
+    ASSERT_TRUE(GetLayerTreeHost().LayerByElementId(element_id));
+  }
+
+  {
+    TestPaintArtifact artifact;
+    ASSERT_TRUE(GetLayerTreeHost().LayerByElementId(element_id));
+    Update(artifact.Build());
+    ASSERT_EQ(0u, ContentLayerCount());
+    ASSERT_FALSE(GetLayerTreeHost().LayerByElementId(element_id));
+  }
+}
+
 }  // namespace blink