43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import OpenGL.GL as gl
|
|
|
|
def test_shader_manager_fbo_initialization():
|
|
with patch("src.shader_manager.gl") as mock_gl:
|
|
mock_gl.glGenFramebuffers.side_effect = [1, 2]
|
|
mock_gl.glGenTextures.side_effect = [3, 4]
|
|
mock_gl.glCheckFramebufferStatus.return_value = mock_gl.GL_FRAMEBUFFER_COMPLETE
|
|
|
|
from src.shader_manager import ShaderManager
|
|
manager = ShaderManager()
|
|
|
|
manager.setup_capture_fbo(800, 600)
|
|
|
|
assert manager.scene_fbo == 1
|
|
assert manager.blur_fbo == 2
|
|
assert manager.scene_tex == 3
|
|
assert manager.blur_tex == 4
|
|
assert mock_gl.glGenFramebuffers.call_count == 2
|
|
assert mock_gl.glGenTextures.call_count == 2
|
|
assert mock_gl.glCheckFramebufferStatus.call_count == 2
|
|
|
|
def test_shader_manager_capture_lifecycle():
|
|
with patch("src.shader_manager.gl") as mock_gl:
|
|
mock_gl.glCheckFramebufferStatus.return_value = mock_gl.GL_FRAMEBUFFER_COMPLETE
|
|
mock_gl.glGenFramebuffers.side_effect = [1, 2]
|
|
mock_gl.glGenTextures.side_effect = [3, 4]
|
|
from src.shader_manager import ShaderManager
|
|
manager = ShaderManager()
|
|
|
|
# Ensure setup is called on first capture
|
|
manager.capture_begin(1024, 768)
|
|
assert manager.fbo_width == 1024
|
|
assert manager.fbo_height == 768
|
|
# Should bind the blur FBO
|
|
mock_gl.glBindFramebuffer.assert_any_call(mock_gl.GL_FRAMEBUFFER, manager.blur_fbo)
|
|
|
|
mock_gl.glBindFramebuffer.reset_mock()
|
|
manager.capture_end()
|
|
# Verify unbind (glBindFramebuffer(..., 0))
|
|
mock_gl.glBindFramebuffer.assert_called_with(mock_gl.GL_FRAMEBUFFER, 0)
|