Draw: Added testcases for hair refine shaders.

1 files changed, 22 insertions, 1 deletions

diff --git a/source/blender/draw/tests/shaders_test.cc b/source/blender/draw/tests/shaders_test.cc
index c96f22859ca..cce69714f5e 100644
--- a/source/blender/draw/tests/shaders_test.cc
+++ b/source/blender/draw/tests/shaders_test.cc
@@ -3,17 +3,22 @@
 #include "testing/testing.h"
 
 #include "draw_testing.hh"
-#include "intern/draw_manager_testing.h"
 
 #include "GPU_context.h"
+#include "GPU_index_buffer.h"
 #include "GPU_init_exit.h"
 #include "GPU_shader.h"
+#include "GPU_texture.h"
+#include "GPU_vertex_buffer.h"
+
+#include "intern/draw_manager_testing.h"
 
 #include "engines/eevee/eevee_private.h"
 #include "engines/gpencil/gpencil_engine.h"
 #include "engines/image/image_private.h"
 #include "engines/overlay/overlay_private.h"
 #include "engines/workbench/workbench_private.h"
+#include "intern/draw_shader.h"
 
 namespace blender::draw {
 
@@ -366,4 +371,20 @@ TEST_F(DrawTest, eevee_glsl_shaders_static)
   EEVEE_shaders_free();
 }
 
+static void test_draw_shaders(eParticleRefineShaderType sh_type)
+{
+  DRW_shaders_free();
+  EXPECT_NE(DRW_shader_hair_refine_get(PART_REFINE_CATMULL_ROM, sh_type), nullptr);
+  DRW_shaders_free();
+}
+
+TEST_F(DrawTest, draw_glsl_shaders)
+{
+#ifndef __APPLE__
+  test_draw_shaders(PART_REFINE_SHADER_TRANSFORM_FEEDBACK);
+  test_draw_shaders(PART_REFINE_SHADER_COMPUTE);
+#endif
+  test_draw_shaders(PART_REFINE_SHADER_TRANSFORM_FEEDBACK_WORKAROUND);
+}
+
 }  // namespace blender::draw