From 981fd63963e47088ae62d637b37bfb49272be1d7 Mon Sep 17 00:00:00 2001
From: Nick Fisher <nick.fisher@avinium.com>
Date: Sat, 30 Nov 2024 16:02:10 +0800
Subject: [PATCH] update overlay tests

---
 thermion_dart/test/overlay_tests.dart | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/thermion_dart/test/overlay_tests.dart b/thermion_dart/test/overlay_tests.dart
index fed81484..58022c54 100644
--- a/thermion_dart/test/overlay_tests.dart
+++ b/thermion_dart/test/overlay_tests.dart
@@ -15,10 +15,14 @@ void main() async {
     group("grid", () {
       test('enable grid', () async {
         await testHelper.withViewer((viewer) async {
+          await viewer.showGridOverlay();
           await viewer.setLayerVisibility(VisibilityLayers.OVERLAY, true);
-          await testHelper.capture(viewer, "enable_grid");
+          await testHelper.capture(viewer, "grid_added_layer_visible");
           await viewer.setLayerVisibility(VisibilityLayers.OVERLAY, false);
-          await testHelper.capture(viewer, "disable_grid");
+          await testHelper.capture(viewer, "grid_added_layer_invisible");
+          await viewer.setLayerVisibility(VisibilityLayers.OVERLAY, true);
+          await viewer.removeGridOverlay();
+          await testHelper.capture(viewer, "grid_remove_layer_visible");
         }, postProcessing: true);
       });
     });
@@ -57,7 +61,6 @@ void main() async {
           await testHelper.capture(viewer, "glb_remove_stencil_highlight");
         }, postProcessing: true, bg: kWhite);
       });
-
     });
   });