Added support for registering additional shared memory objects (textures and ...

o3d/gpu_plugin/command_buffer_unittest.cc

 // Copyright (c) 2006-2008 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 "base/thread.h"
 #include "o3d/gpu_plugin/command_buffer.h"
 #include "o3d/gpu_plugin/np_utils/dynamic_np_object.h"
 #include "o3d/gpu_plugin/np_utils/np_browser_mock.h"
 #include "o3d/gpu_plugin/np_utils/np_object_mock.h"
 #include "o3d/gpu_plugin/np_utils/np_object_pointer.h"
 #include "o3d/gpu_plugin/system_services/shared_memory_mock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 using testing::_;
@@ skipping 136 matching lines @@
 
   command_buffer_->SetGetOffset(2);
   EXPECT_EQ(2, command_buffer_->GetGetOffset());
   EXPECT_CALL(*put_offset_change_callback, RunWithParams(_));
   EXPECT_EQ(2, command_buffer_->SyncOffsets(6));
 
   EXPECT_EQ(-1, command_buffer_->SyncOffsets(-1));
   EXPECT_EQ(-1, command_buffer_->SyncOffsets(1024));
 }
 
+TEST_F(CommandBufferTest, ZeroHandleMapsToNull) {
+  EXPECT_TRUE(NULL == command_buffer_->GetRegisteredObject(0).Get());
+}
+
+TEST_F(CommandBufferTest, RegisteringNullObjectReturnsZero) {
+  EXPECT_EQ(0, command_buffer_->RegisterObject(NPObjectPointer<NPObject>()));
+}
+
+TEST_F(CommandBufferTest, RegistersDistinctNonZeroHandlesForObject) {
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  EXPECT_EQ(2, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(2));
+}
+
+TEST_F(CommandBufferTest, RegisterObjectReusesUnregisteredHandles) {
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  EXPECT_EQ(2, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(2));
+  command_buffer_->UnregisterObject(window_object_, 1);
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  EXPECT_EQ(3, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(3));
+}
+
+TEST_F(CommandBufferTest, CannotUnregisterHandleZero) {
+  command_buffer_->UnregisterObject(window_object_, 0);
+  EXPECT_TRUE(NULL == command_buffer_->GetRegisteredObject(0).Get());
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+}
+
+TEST_F(CommandBufferTest, CannotUnregisterNegativeHandles) {
+  command_buffer_->UnregisterObject(window_object_, -1);
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+}
+
+TEST_F(CommandBufferTest, CannotUnregisterUnregisteredHandles) {
+  command_buffer_->UnregisterObject(window_object_, 1);
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+}
+
+TEST_F(CommandBufferTest,
+    CannotUnregisterHandleWithoutDemonstratingAccessToObject) {
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  command_buffer_->UnregisterObject(chromium_object_, 1);
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  EXPECT_EQ(2, command_buffer_->RegisterObject(window_object_));
+}
+
+// Testing this case specifically because there is an optimization that takes
+// a different code path in this case.
+TEST_F(CommandBufferTest, UnregistersLastRegisteredHandle) {
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  command_buffer_->UnregisterObject(window_object_, 1);
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+}
+
+// Testing this case specifically because there is an optimization that takes
+// a different code path in this case.
+TEST_F(CommandBufferTest, UnregistersTwoLastRegisteredHandles) {
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+  EXPECT_EQ(2, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(2));
+  command_buffer_->UnregisterObject(window_object_, 2);
+  command_buffer_->UnregisterObject(window_object_, 1);
+  EXPECT_EQ(1, command_buffer_->RegisterObject(window_object_));
+  EXPECT_EQ(window_object_, command_buffer_->GetRegisteredObject(1));
+}
+
 }  // namespace gpu_plugin
 }  // namespace o3d