1 files changed, 13 insertions, 5 deletions
diff --git a/src/tests/test_3d_render.cc b/src/tests/test_3d_render.cc
index 41bffe6..4be7153 100644
--- a/src/tests/test_3d_render.cc
+++ b/src/tests/test_3d_render.cc
@@ -140,19 +140,27 @@ void init_wgpu() {
 
 void setup_scene() {
   g_scene.clear();
-  // Center Red Cube
-  Object3D center;
+  // Center Red Cube (Wireframe Proxy)
+  Object3D center(ObjectType::CUBE);
   center.position = vec3(0, 0, 0);
   center.color = vec4(1, 0, 0, 1);
   g_scene.add_object(center);
 
-  // Orbiting Green Cubes
+  // Orbiting Objects
   for (int i = 0; i < 8; ++i) {
-    Object3D obj;
+    ObjectType type = ObjectType::SPHERE;
+    if (i % 3 == 1) type = ObjectType::TORUS;
+    if (i % 3 == 2) type = ObjectType::BOX;
+    
+    Object3D obj(type);
     float angle = (i / 8.0f) * 6.28318f;
     obj.position = vec3(std::cos(angle) * 4.0f, 0, std::sin(angle) * 4.0f);
     obj.scale = vec3(0.5f, 0.5f, 0.5f);
-    obj.color = vec4(0, 1, 0, 1);
+    
+    if (type == ObjectType::SPHERE) obj.color = vec4(0, 1, 0, 1);
+    else if (type == ObjectType::TORUS) obj.color = vec4(0, 0.5, 1, 1);
+    else obj.color = vec4(1, 1, 0, 1);
+    
     g_scene.add_object(obj);
   }
 }