Free the resources used by a context group.

gpu/command_buffer/service/program_manager_unittest.cc

 // Copyright (c) 2010 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 "gpu/command_buffer/service/program_manager.h"
 #include "app/gfx/gl/gl_mock.h"
 #include "base/scoped_ptr.h"
 #include "base/string_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::gfx::MockGLInterface;
 using ::testing::_;
 using ::testing::DoAll;
 using ::testing::InSequence;
 using ::testing::MatcherCast;
 using ::testing::Pointee;
 using ::testing::Return;
 using ::testing::SetArrayArgument;
 using ::testing::SetArgumentPointee;
 using ::testing::StrEq;
 using ::testing::StrictMock;
 
 namespace gpu {
 namespace gles2 {
 
 class ProgramManagerTest : public testing::Test {
  public:
   ProgramManagerTest() { }
+  ~ProgramManagerTest() {
+    manager_.Destroy(false);
+  }
 
  protected:
   virtual void SetUp() {
+    gl_.reset(new ::testing::StrictMock< ::gfx::MockGLInterface>());
+    ::gfx::GLInterface::SetGLInterface(gl_.get());
   }
 
   virtual void TearDown() {
+    ::gfx::GLInterface::SetGLInterface(NULL);
+    gl_.reset();
   }
 
+  // Use StrictMock to make 100% sure we know how GL will be called.
+  scoped_ptr< ::testing::StrictMock< ::gfx::MockGLInterface> > gl_;
   ProgramManager manager_;
 };
 
 TEST_F(ProgramManagerTest, Basic) {
   const GLuint kClient1Id = 1;
   const GLuint kService1Id = 11;
   const GLuint kClient2Id = 2;
   // Check we can create program.
   manager_.CreateProgramInfo(kClient1Id, kService1Id);
   // Check program got created.
   ProgramManager::ProgramInfo* info1 = manager_.GetProgramInfo(kClient1Id);
   ASSERT_TRUE(info1 != NULL);
   EXPECT_EQ(kService1Id, info1->service_id());
   EXPECT_FALSE(info1->CanLink());
   EXPECT_STREQ("", info1->log_info().c_str());
   GLuint client_id = 0;
   EXPECT_TRUE(manager_.GetClientId(info1->service_id(), &client_id));
   EXPECT_EQ(kClient1Id, client_id);
   // Check we get nothing for a non-existent program.
   EXPECT_TRUE(manager_.GetProgramInfo(kClient2Id) == NULL);
   // Check trying to a remove non-existent programs does not crash.
   manager_.RemoveProgramInfo(kClient2Id);
   // Check we can't get the program after we remove it.
   manager_.RemoveProgramInfo(kClient1Id);
   EXPECT_TRUE(manager_.GetProgramInfo(kClient1Id) == NULL);
 }
 
+TEST_F(ProgramManagerTest, Destroy) {
+  const GLuint kClient1Id = 1;
+  const GLuint kService1Id = 11;
+  // Check we can create program.
+  manager_.CreateProgramInfo(kClient1Id, kService1Id);
+  // Check program got created.
+  ProgramManager::ProgramInfo* info1 =
+      manager_.GetProgramInfo(kClient1Id);
+  ASSERT_TRUE(info1 != NULL);
+  EXPECT_CALL(*gl_, DeleteProgram(kService1Id))
+      .Times(1)
+      .RetiresOnSaturation();
+  manager_.Destroy(true);
+  // Check the resources were released.
+  info1 = manager_.GetProgramInfo(kClient1Id);
+  ASSERT_TRUE(info1 == NULL);
+}
+
 class ProgramManagerWithShaderTest : public testing::Test {
  public:
   ProgramManagerWithShaderTest()
       : program_info_(NULL) {
   }
 
+  ~ProgramManagerWithShaderTest() {
+    manager_.Destroy(false);
+  }
+
   static const GLint kNumVertexAttribs = 16;
 
   static const GLuint kClientProgramId = 123;
   static const GLuint kServiceProgramId = 456;
 
   static const char* kAttrib1Name;
   static const char* kAttrib2Name;
   static const char* kAttrib3Name;
   static const GLint kAttrib1Size = 1;
   static const GLint kAttrib2Size = 1;
@@ skipping 323 matching lines @@
   program_info->AttachShader(fshader);
   EXPECT_TRUE(program_info->CanLink());
   vshader->SetStatus(false, "");
   EXPECT_FALSE(program_info->CanLink());
   vshader->SetStatus(true, "");
   EXPECT_TRUE(program_info->CanLink());
   fshader->SetStatus(false, "");
   EXPECT_FALSE(program_info->CanLink());
   fshader->SetStatus(true, "");
   EXPECT_TRUE(program_info->CanLink());
+  shader_manager.Destroy(false);
 }
 
 TEST_F(ProgramManagerWithShaderTest, GetUniformLocation) {
   const ProgramManager::ProgramInfo* program_info =
       manager_.GetProgramInfo(kClientProgramId);
   ASSERT_TRUE(program_info != NULL);
   EXPECT_EQ(kUniform1Location, program_info->GetUniformLocation(kUniform1Name));
   EXPECT_EQ(kUniform2Location, program_info->GetUniformLocation(kUniform2Name));
   EXPECT_EQ(kUniform3Location, program_info->GetUniformLocation(kUniform3Name));
   // Check we can get uniform2 as "uniform2" even though the name is
@@ skipping 59 matching lines @@
   // as the "gl_" uniform we skipped.
   // +4u is to account for "gl_" and NULL terminator.
   program_info->GetProgramiv(GL_ACTIVE_UNIFORM_MAX_LENGTH, &value);
   EXPECT_EQ(strlen(kUniform3Name) + 4u, static_cast<size_t>(value));
 }
 
 }  // namespace gles2
 }  // namespace gpu