def test_enableDisableAttribute(type, offset, stride, type_result, tuple_size):
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
attribute_index = MagicMock()
mocked_shader_program.attributeLocation = MagicMock(
return_value=attribute_index)
shader.enableAttribute("blorp", type, offset, stride)
# Validate
mocked_shader_program.setAttributeBuffer.assert_called_once_with(
attribute_index, type_result, offset, tuple_size, stride)
mocked_shader_program.enableAttributeArray.assert_called_once_with(
attribute_index)
# Disable it again
shader.disableAttribute("blorp")
mocked_shader_program.disableAttributeArray.assert_called_once_with(
attribute_index)
# Disable unknown attribute
shader.disableAttribute("BEEP")
mocked_shader_program.disableAttributeArray.assert_called_once_with(
attribute_index)
def test_load():
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "test.shader"))
# It should be called 3 times, once for vertex, once for fragment and once for geometry
call_arg_list = mocked_shader_program.addShaderFromSourceCode.call_args_list
assert call(QOpenGLShader.Vertex, "vertex_code") in call_arg_list
assert call(QOpenGLShader.Fragment, "fragment_code") in call_arg_list
assert call(QOpenGLShader.Geometry, "geometry_code") in call_arg_list
def createShaderProgram(self, file_name: str) -> ShaderProgram:
shader = ShaderProgram()
# The version_string must match the keys in shader files.
if OpenGLContext.isLegacyOpenGL():
version_string = "" # Nothing is added to "fragment" and "vertex"
else:
version_string = "41core"
try:
shader.load(file_name, version = version_string)
except InvalidShaderProgramError:
# If the loading failed, it could be that there is no specific shader for this version.
# Try again without a version nr to get the generic one.
if version_string != "":
shader.load(file_name, version = "")
return shader
def test_loadInvalid():
shader = ShaderProgram()
with pytest.raises(InvalidShaderProgramError):
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "invalid.shader"))
with pytest.raises(InvalidShaderProgramError):
shader.load(os.path.join(os.path.dirname(os.path.abspath(__file__)), "Shaders", "invalid2.shader"))
def createShaderProgram(self, file_name: str) -> ShaderProgram:
shader = ShaderProgram()
# The version_string must match the keys in shader files.
if OpenGLContext.isLegacyOpenGL():
version_string = "" # Nothing is added to "fragment" and "vertex"
else:
version_string = "41core"
try:
shader.load(file_name, version=version_string)
except InvalidShaderProgramError:
# If the loading failed, it could be that there is no specific shader for this version.
# Try again without a version nr to get the generic one.
if version_string != "":
shader.load(file_name, version="")
return shader
def createShaderProgram(self, file_name: str) -> Optional[ShaderProgram]:
"""Create a ShaderProgram Object.
This should return an implementation-specifc ShaderProgram subclass.
"""
shader = ShaderProgram()
# The version_string must match the keys in shader files.
if OpenGLContext.isLegacyOpenGL():
version_string = "" # Nothing is added to "fragment" and "vertex"
else:
version_string = "41core"
try:
shader.load(file_name, version=version_string)
except InvalidShaderProgramError:
# If the loading failed, it could be that there is no specific shader for this version.
# Try again without a version nr to get the generic one.
if version_string != "":
try:
shader.load(file_name, version="")
except InvalidShaderProgramError as e:
Logger.logException("e", str(e))
return None
return shader
def test_ShaderProgramInit():
# Creating it just shouldn't fail.
shader = ShaderProgram()
def test_bindAndRelease():
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
shader.bind()
assert mocked_shader_program.bind.call_count == 1
# Doing it without releasing in between shouldn't cause another bind.
shader.bind()
assert mocked_shader_program.bind.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
# We left it unbound, so now binding should work.
shader.bind()
assert mocked_shader_program.bind.call_count == 2
def test_bindAndRelease():
shader = ShaderProgram()
mocked_shader_program = MagicMock()
shader._shader_program = mocked_shader_program
shader.bind()
assert mocked_shader_program.bind.call_count == 1
# Doing it without releasing in between shouldn't cause another bind.
shader.bind()
assert mocked_shader_program.bind.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
shader.release()
assert mocked_shader_program.release.call_count == 1
# We left it unbound, so now binding should work.
shader.bind()
assert mocked_shader_program.bind.call_count == 2