1 files changed, 38 insertions, 0 deletions
diff --git a/src/tests/test_shader_composer.cc b/src/tests/test_shader_composer.cc
new file mode 100644
index 0000000..4a5cb8b
--- /dev/null
+++ b/src/tests/test_shader_composer.cc
@@ -0,0 +1,38 @@
+// This file is part of the 64k demo project.
+// It tests the ShaderComposer utility.
+
+#include "gpu/effects/shader_composer.h"
+#include <cassert>
+#include <iostream>
+#include <string>
+
+void test_composition() {
+    std::cout << "Testing Shader Composition..." << std::endl;
+    auto& sc = ShaderComposer::Get();
+    
+    sc.RegisterSnippet("math", "fn add(a: f32, b: f32) -> f32 { return a + b; }");
+    sc.RegisterSnippet("util", "fn square(a: f32) -> f32 { return a * a; }");
+    
+    std::string main_code = "fn main() { let x = add(1.0, square(2.0)); }";
+    std::string result = sc.Compose({"math", "util"}, main_code);
+    
+    // Verify order and presence
+    assert(result.find("Snippet: math") != std::string::npos);
+    assert(result.find("Snippet: util") != std::string::npos);
+    assert(result.find("Main Code") != std::string::npos);
+    
+    size_t pos_math = result.find("Snippet: math");
+    size_t pos_util = result.find("Snippet: util");
+    size_t pos_main = result.find("Main Code");
+    
+    assert(pos_math < pos_util);
+    assert(pos_util < pos_main);
+    
+    std::cout << "Composition logic verified." << std::endl;
+}
+
+int main() {
+    test_composition();
+    std::cout << "--- ALL SHADER COMPOSER TESTS PASSED ---" << std::endl;
+    return 0;
+}