From d8a0859f165cdbddc7b8ab6f943cfe850e3e1732 Mon Sep 17 00:00:00 2001
From: Nick Fisher <nick.fisher@avinium.com>
Date: Thu, 19 Sep 2024 13:00:40 +0800
Subject: [PATCH] update test

---
 thermion_dart/test/integration_test.dart | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/thermion_dart/test/integration_test.dart b/thermion_dart/test/integration_test.dart
index e2b5e77f..a8fcc755 100644
--- a/thermion_dart/test/integration_test.dart
+++ b/thermion_dart/test/integration_test.dart
@@ -328,15 +328,16 @@ void main() async {
           .setCameraRotation(Quaternion.axisAngle(Vector3(1, 0, 0), -pi / 8));
       await viewer.setBackgroundColor(1.0, 0.0, 1.0, 1.0);
 
-      // var materialInstance =
-      //     await viewer.createUbershaderMaterialInstance(unlit: true);
+      var materialInstance =
+          await viewer.createUbershaderMaterialInstance(unlit: true);
       final cube = await viewer.createGeometry(
           GeometryHelper.cube(uvs: false, normals: true),
-          materialInstance: null);
-      // await viewer.setMaterialPropertyFloat4(
-      //     cube, "baseColorFactor", 0, 0.0, 1.0, 0.0, 0.0);
-      // await viewer.destroyMaterialInstance(materialInstance);
+          materialInstance: materialInstance);
+      await viewer.setMaterialPropertyFloat4(
+          cube, "baseColorFactor", 0, 0.0, 1.0, 0.0, 0.0);
       await _capture(viewer, "geometry_cube_with_custom_material");
+      await viewer.removeEntity(cube);
+      await viewer.destroyMaterialInstance(materialInstance);
     });
 
     test('create sphere (no normals)', () async {