src/gpu/command_buffer/service/shader_manager_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "gpu/command_buffer/service/shader_manager.h"
   6
   7 #include "base/memory/scoped_ptr.h"
   8 #include "gpu/command_buffer/service/mocks.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10 #include "ui/gl/gl_mock.h"
  11
  12 using ::testing::Return;
  13 using ::testing::ReturnRef;
  14
  15 namespace gpu {
  16 namespace gles2 {
  17
  18 class ShaderManagerTest : public testing::Test {
  19  public:
  20   ShaderManagerTest() {
  21   }
  22
  23   virtual ~ShaderManagerTest() {
  24     manager_.Destroy(false);
  25   }
  26
  27  protected:
  28   virtual void SetUp() {
  29     gl_.reset(new ::testing::StrictMock< ::gfx::MockGLInterface>());
  30     ::gfx::GLInterface::SetGLInterface(gl_.get());
  31   }
  32
  33   virtual void TearDown() {
  34     ::gfx::GLInterface::SetGLInterface(NULL);
  35     gl_.reset();
  36   }
  37
  38   // Use StrictMock to make 100% sure we know how GL will be called.
  39   scoped_ptr< ::testing::StrictMock< ::gfx::MockGLInterface> > gl_;
  40   ShaderManager manager_;
  41 };
  42
  43 TEST_F(ShaderManagerTest, Basic) {
  44   const GLuint kClient1Id = 1;
  45   const GLuint kService1Id = 11;
  46   const GLenum kShader1Type = GL_VERTEX_SHADER;
  47   const GLuint kClient2Id = 2;
  48   // Check we can create shader.
  49   Shader* info0 = manager_.CreateShader(
  50       kClient1Id, kService1Id, kShader1Type);
  51   // Check shader got created.
  52   ASSERT_TRUE(info0 != NULL);
  53   Shader* shader1 = manager_.GetShader(kClient1Id);
  54   ASSERT_EQ(info0, shader1);
  55   // Check we get nothing for a non-existent shader.
  56   EXPECT_TRUE(manager_.GetShader(kClient2Id) == NULL);
  57   // Check we can't get the shader after we remove it.
  58   manager_.MarkAsDeleted(shader1);
  59   EXPECT_TRUE(manager_.GetShader(kClient1Id) == NULL);
  60 }
  61
  62 TEST_F(ShaderManagerTest, Destroy) {
  63   const GLuint kClient1Id = 1;
  64   const GLuint kService1Id = 11;
  65   const GLenum kShader1Type = GL_VERTEX_SHADER;
  66   // Check we can create shader.
  67   Shader* shader1 = manager_.CreateShader(
  68       kClient1Id, kService1Id, kShader1Type);
  69   // Check shader got created.
  70   ASSERT_TRUE(shader1 != NULL);
  71   EXPECT_CALL(*gl_, DeleteShader(kService1Id))
  72       .Times(1)
  73       .RetiresOnSaturation();
  74   manager_.Destroy(true);
  75   // Check that resources got freed.
  76   shader1 = manager_.GetShader(kClient1Id);
  77   ASSERT_TRUE(shader1 == NULL);
  78 }
  79
  80 TEST_F(ShaderManagerTest, DeleteBug) {
  81   const GLuint kClient1Id = 1;
  82   const GLuint kClient2Id = 2;
  83   const GLuint kService1Id = 11;
  84   const GLuint kService2Id = 12;
  85   const GLenum kShaderType = GL_VERTEX_SHADER;
  86   // Check we can create shader.
  87   scoped_refptr<Shader> shader1(
  88       manager_.CreateShader(kClient1Id, kService1Id, kShaderType));
  89   scoped_refptr<Shader> shader2(
  90       manager_.CreateShader(kClient2Id, kService2Id, kShaderType));
  91   ASSERT_TRUE(shader1.get());
  92   ASSERT_TRUE(shader2.get());
  93   manager_.UseShader(shader1.get());
  94   manager_.MarkAsDeleted(shader1.get());
  95   manager_.MarkAsDeleted(shader2.get());
  96   EXPECT_TRUE(manager_.IsOwned(shader1.get()));
  97   EXPECT_FALSE(manager_.IsOwned(shader2.get()));
  98 }
  99
 100 TEST_F(ShaderManagerTest, Shader) {
 101   const GLuint kClient1Id = 1;
 102   const GLuint kService1Id = 11;
 103   const GLenum kShader1Type = GL_VERTEX_SHADER;
 104   const char* kClient1Source = "hello world";
 105   // Check we can create shader.
 106   Shader* shader1 = manager_.CreateShader(
 107       kClient1Id, kService1Id, kShader1Type);
 108   // Check shader got created.
 109   ASSERT_TRUE(shader1 != NULL);
 110   EXPECT_EQ(kService1Id, shader1->service_id());
 111   // Check if the shader has correct type.
 112   EXPECT_EQ(kShader1Type, shader1->shader_type());
 113   EXPECT_FALSE(shader1->IsValid());
 114   EXPECT_FALSE(shader1->InUse());
 115   EXPECT_TRUE(shader1->source() == NULL);
 116   EXPECT_TRUE(shader1->log_info() == NULL);
 117   const char* kLog = "foo";
 118   shader1->SetStatus(true, kLog, NULL);
 119   EXPECT_TRUE(shader1->IsValid());
 120   EXPECT_STREQ(kLog, shader1->log_info()->c_str());
 121   // Check we can set its source.
 122   shader1->UpdateSource(kClient1Source);
 123   EXPECT_STREQ(kClient1Source, shader1->source()->c_str());
 124   EXPECT_EQ(NULL, shader1->translated_source());
 125   // Check we can set its translated source.
 126   shader1->UpdateTranslatedSource(kClient1Source);
 127   EXPECT_STREQ(kClient1Source,
 128                shader1->translated_source()->c_str());
 129 }
 130
 131 TEST_F(ShaderManagerTest, GetInfo) {
 132   const GLuint kClient1Id = 1;
 133   const GLuint kService1Id = 11;
 134   const GLenum kShader1Type = GL_VERTEX_SHADER;
 135   const GLenum kAttrib1Type = GL_FLOAT_VEC2;
 136   const GLsizei kAttrib1Size = 2;
 137   const int kAttrib1Precision = SH_PRECISION_MEDIUMP;
 138   const char* kAttrib1Name = "attr1";
 139   const GLenum kAttrib2Type = GL_FLOAT_VEC3;
 140   const GLsizei kAttrib2Size = 4;
 141   const int kAttrib2Precision = SH_PRECISION_HIGHP;
 142   const char* kAttrib2Name = "attr2";
 143   const int kAttribStaticUse = 0;
 144   const GLenum kUniform1Type = GL_FLOAT_MAT2;
 145   const GLsizei kUniform1Size = 3;
 146   const int kUniform1Precision = SH_PRECISION_LOWP;
 147   const int kUniform1StaticUse = 1;
 148   const char* kUniform1Name = "uni1";
 149   const GLenum kUniform2Type = GL_FLOAT_MAT3;
 150   const GLsizei kUniform2Size = 5;
 151   const int kUniform2Precision = SH_PRECISION_MEDIUMP;
 152   const int kUniform2StaticUse = 0;
 153   const char* kUniform2Name = "uni2";
 154
 155   MockShaderTranslator shader_translator;
 156   ShaderTranslator::VariableMap attrib_map;
 157   attrib_map[kAttrib1Name] = ShaderTranslatorInterface::VariableInfo(
 158       kAttrib1Type, kAttrib1Size, kAttrib1Precision,
 159       kAttribStaticUse, kAttrib1Name);
 160   attrib_map[kAttrib2Name] = ShaderTranslatorInterface::VariableInfo(
 161       kAttrib2Type, kAttrib2Size, kAttrib2Precision,
 162       kAttribStaticUse, kAttrib2Name);
 163   ShaderTranslator::VariableMap uniform_map;
 164   uniform_map[kUniform1Name] = ShaderTranslatorInterface::VariableInfo(
 165       kUniform1Type, kUniform1Size, kUniform1Precision,
 166       kUniform1StaticUse, kUniform1Name);
 167   uniform_map[kUniform2Name] = ShaderTranslatorInterface::VariableInfo(
 168       kUniform2Type, kUniform2Size, kUniform2Precision,
 169       kUniform2StaticUse, kUniform2Name);
 170   EXPECT_CALL(shader_translator, attrib_map())
 171       .WillRepeatedly(ReturnRef(attrib_map));
 172   EXPECT_CALL(shader_translator, uniform_map())
 173       .WillRepeatedly(ReturnRef(uniform_map));
 174   ShaderTranslator::VariableMap varying_map;
 175   EXPECT_CALL(shader_translator, varying_map())
 176       .WillRepeatedly(ReturnRef(varying_map));
 177   ShaderTranslator::NameMap name_map;
 178   EXPECT_CALL(shader_translator, name_map())
 179       .WillRepeatedly(ReturnRef(name_map));
 180   // Check we can create shader.
 181   Shader* shader1 = manager_.CreateShader(
 182       kClient1Id, kService1Id, kShader1Type);
 183   // Check shader got created.
 184   ASSERT_TRUE(shader1 != NULL);
 185   // Set Status
 186   shader1->SetStatus(true, "", &shader_translator);
 187   // Check attrib and uniform infos got copied.
 188   for (ShaderTranslator::VariableMap::const_iterator it = attrib_map.begin();
 189        it != attrib_map.end(); ++it) {
 190     const Shader::VariableInfo* variable_info =
 191         shader1->GetAttribInfo(it->first);
 192     ASSERT_TRUE(variable_info != NULL);
 193     EXPECT_EQ(it->second.type, variable_info->type);
 194     EXPECT_EQ(it->second.size, variable_info->size);
 195     EXPECT_EQ(it->second.precision, variable_info->precision);
 196     EXPECT_EQ(it->second.static_use, variable_info->static_use);
 197     EXPECT_EQ(it->second.name, variable_info->name);
 198   }
 199   for (ShaderTranslator::VariableMap::const_iterator it = uniform_map.begin();
 200        it != uniform_map.end(); ++it) {
 201     const Shader::VariableInfo* variable_info =
 202         shader1->GetUniformInfo(it->first);
 203     ASSERT_TRUE(variable_info != NULL);
 204     EXPECT_EQ(it->second.type, variable_info->type);
 205     EXPECT_EQ(it->second.size, variable_info->size);
 206     EXPECT_EQ(it->second.precision, variable_info->precision);
 207     EXPECT_EQ(it->second.static_use, variable_info->static_use);
 208     EXPECT_EQ(it->second.name, variable_info->name);
 209   }
 210   // Check attrib and uniform get cleared.
 211   shader1->SetStatus(true, NULL, NULL);
 212   EXPECT_TRUE(shader1->log_info() == NULL);
 213   for (ShaderTranslator::VariableMap::const_iterator it = attrib_map.begin();
 214        it != attrib_map.end(); ++it) {
 215     const Shader::VariableInfo* variable_info =
 216         shader1->GetAttribInfo(it->first);
 217     EXPECT_TRUE(variable_info == NULL);
 218   }
 219   for (ShaderTranslator::VariableMap::const_iterator it = uniform_map.begin();
 220        it != uniform_map.end(); ++it) {
 221     const Shader::VariableInfo* variable_info =
 222         shader1->GetUniformInfo(it->first);
 223     ASSERT_TRUE(variable_info == NULL);
 224   }
 225 }
 226
 227 TEST_F(ShaderManagerTest, ShaderInfoUseCount) {
 228   const GLuint kClient1Id = 1;
 229   const GLuint kService1Id = 11;
 230   const GLenum kShader1Type = GL_VERTEX_SHADER;
 231   // Check we can create shader.
 232   Shader* shader1 = manager_.CreateShader(
 233       kClient1Id, kService1Id, kShader1Type);
 234   // Check shader got created.
 235   ASSERT_TRUE(shader1 != NULL);
 236   EXPECT_FALSE(shader1->InUse());
 237   EXPECT_FALSE(shader1->IsDeleted());
 238   manager_.UseShader(shader1);
 239   EXPECT_TRUE(shader1->InUse());
 240   manager_.UseShader(shader1);
 241   EXPECT_TRUE(shader1->InUse());
 242   manager_.MarkAsDeleted(shader1);
 243   EXPECT_TRUE(shader1->IsDeleted());
 244   Shader* shader2 = manager_.GetShader(kClient1Id);
 245   EXPECT_EQ(shader1, shader2);
 246   manager_.UnuseShader(shader1);
 247   EXPECT_TRUE(shader1->InUse());
 248   manager_.UnuseShader(shader1);  // this should delete the info.
 249   shader2 = manager_.GetShader(kClient1Id);
 250   EXPECT_TRUE(shader2 == NULL);
 251
 252   shader1 = manager_.CreateShader(kClient1Id, kService1Id, kShader1Type);
 253   ASSERT_TRUE(shader1 != NULL);
 254   EXPECT_FALSE(shader1->InUse());
 255   manager_.UseShader(shader1);
 256   EXPECT_TRUE(shader1->InUse());
 257   manager_.UseShader(shader1);
 258   EXPECT_TRUE(shader1->InUse());
 259   manager_.UnuseShader(shader1);
 260   EXPECT_TRUE(shader1->InUse());
 261   manager_.UnuseShader(shader1);
 262   EXPECT_FALSE(shader1->InUse());
 263   shader2 = manager_.GetShader(kClient1Id);
 264   EXPECT_EQ(shader1, shader2);
 265   manager_.MarkAsDeleted(shader1);  // this should delete the shader.
 266   shader2 = manager_.GetShader(kClient1Id);
 267   EXPECT_TRUE(shader2 == NULL);
 268 }
 269
 270 }  // namespace gles2
 271 }  // namespace gpu