test: implement kittest ui testing

.gitignore

@@ -30,3 +30,6 @@ test_db
 **/.DS_Store
 explorer.log
 .gitaipconfig
+
+# Ignore egui kittest snapshots
+tests/snapshots/

Cargo.toml

@@ -69,4 +69,4 @@ raw-cpuid = "11.5.0"
 [dev-dependencies]
 
 tempfile = { version = "3.20.0" }
-egui_kittest = { version = "0.31.1", features = ["eframe"] }
+egui_kittest = { version = "0.31.1", features = ["eframe", "snapshot", "wgpu"] }

src/logging.rs

@@ -8,13 +8,13 @@
     let log_file_path = app_user_data_file_path("det.log").expect("should create log file path");
     let log_file = match std::fs::File::create(&log_file_path) {
         Ok(file) => file,
         Err(e) => panic!("Failed to create log file: {:?}", e),
     };
 
     let filter = EnvFilter::try_new(
         "info,dash_evo_tool=trace,dash_sdk=debug,tenderdash_abci=debug,drive=debug,drive_proof_verifier=debug,rs_dapi_client=debug,h2=warn",
     )
         .unwrap_or_else(|e| panic!("Failed to create EnvFilter: {:?}", e));
 
     let subscriber = tracing_subscriber::fmt()
         .with_env_filter(filter)
@@ -24,7 +24,11 @@
 
     // Set global subscriber with proper error handling
     if let Err(e) = tracing::subscriber::set_global_default(subscriber) {
-        panic!("Unable to set global default subscriber: {:?}", e);
+        // log to stderr and ignore the error
+        eprintln!(
+            "WARNING: cannot initialize logging, logging might not work correctly: {:?}",
+            e
+        );
     }
 
     // Log panic events

src/ui/components/left_panel.rs

@@ -1,6 +1,7 @@
 use crate::app::AppAction;
 use crate::context::AppContext;
 use crate::ui::components::styled::GradientButton;
+use crate::ui::components::test_label::TestableWidget;
 use crate::ui::theme::{DashColors, Shadow, Shape, Spacing};
 use crate::ui::RootScreenType;
 use dash_sdk::dashcore_rpc::dashcore::Network;
@@ -38,11 +39,11 @@
 
                     Some(texture)
                 } else {
                     eprintln!("Failed to load image from embedded data at path: {}", path);
                     None
                 }
             } else {
                 eprintln!("Image not found in embedded assets at path: {}", path);
                 None
             }
         })
@@ -56,7 +57,8 @@
     let mut action = AppAction::None;
 
     // Define the button details directly in this function
-    let buttons = [
+    // Contains (label, screen type, icon path)
+    let buttons: [(&'static str, RootScreenType, &'static str); 7] = [
         ("I", RootScreenType::RootScreenIdentities, "identity.png"),
         ("Q", RootScreenType::RootScreenDocumentQuery, "doc.png"),
         ("O", RootScreenType::RootScreenMyTokenBalances, "tokens.png"),
@@ -112,10 +114,13 @@
 
                             // Add icon-based button if texture is loaded
                             if let Some(ref texture) = texture {
-                                let button =
-                                    ImageButton::new(texture).frame(false).tint(button_color);
+                                let button = ImageButton::new(texture)
+                                    .frame(false)
+                                    .tint(button_color)
+                                    .test_label(label);
 
                                 let added = ui.add(button);
+
                                 if added.clicked() {
                                     action =
                                         AppAction::SetMainScreenThenGoToMainScreen(*screen_type);

src/ui/components/mod.rs

@@ -4,6 +4,7 @@ pub mod entropy_grid;
 pub mod left_panel;
 pub mod left_wallet_panel;
 pub mod styled;
+pub mod test_label;
 pub mod tokens_subscreen_chooser_panel;
 pub mod tools_subscreen_chooser_panel;
 pub mod top_panel;

src/ui/components/styled.rs

@@ -6,7 +6,7 @@ use crate::{
 };
 use egui::{
     Button, CentralPanel, Color32, Context, Frame, Margin, Response, RichText, Stroke, TextEdit,
-    Ui, Vec2,
+    Ui, Vec2, Widget,
 };
 
 /// Styled button variants
@@ -217,6 +217,12 @@ impl<'a> StyledCheckbox<'a> {
     }
 }
 
+impl Widget for StyledCheckbox<'_> {
+    fn ui(self, ui: &mut Ui) -> Response {
+        self.show(ui)
+    }
+}
+
 /// Gradient button with animated effects
 pub(crate) struct GradientButton {
     text: String,

src/ui/components/test_label.rs

@@ -0,0 +1,60 @@
+use egui::{Sense, Widget, WidgetInfo};
+
+/// A wrapper widget that adds a label to any `egui::Widget` for testing purposes.
+///
+/// This widget allows you to attach a label to any widget, which can be used in tests
+/// to identify and interact with the widget from egui_kittest.
+/// Only use when there is no other way to identify the widget in tests.
+///
+/// ## Example usage:
+///
+/// ```rust
+/// use egui::Widget;
+/// use dash_evo_tool::ui::components::test_label::{TestLabel, TestableWidget};
+/// fn my_widget(ui: &mut egui::Ui) {
+///     let my_button = egui::Button::new("Click me");
+///     ui.add(my_button.test_label("my_button"));
+/// }
+/// ```
+pub struct TestLabel<T: Widget> {
+    pub label: String,
+    pub inner: T,
+}
+
+impl<T: Widget> TestLabel<T> {
+    pub fn new(inner: T, label: &str) -> Self {
+        Self {
+            label: label.to_string(),
+            inner,
+        }
+    }
+}
+
+impl<T: Widget> egui::Widget for TestLabel<T> {
+    fn ui(self, ui: &mut egui::Ui) -> egui::Response {
+        let scope = ui.scope(|ui| self.inner.ui(ui));
+        let response = scope.response.interact(Sense::click());
+
+        response.widget_info(move || {
+            let label = self.label.clone();
+            WidgetInfo::labeled(egui::WidgetType::Other, ui.is_enabled(), label)
+        });
+
+        // Pass all interactions from the inner widget
+        scope.inner.union(response)
+    }
+}
+/// Trait to allow any widget to be tested with a label
+///
+/// This trait provides a method to attach a test label to any widget.
+/// The label can be used to identify the widget in tests, making it easier to interact with
+/// and assert conditions on the widget during testing.
+pub trait TestableWidget<T: Widget> {
+    fn test_label(self, label: &str) -> TestLabel<T>;
+}
+
+impl<T: Widget> TestableWidget<T> for T {
+    fn test_label(self, label: &str) -> TestLabel<T> {
+        TestLabel::new(self, label)
+    }
+}

src/ui/network_chooser_screen.rs

@@ -6,12 +6,14 @@
 use crate::context::AppContext;
 use crate::ui::components::left_panel::add_left_panel;
 use crate::ui::components::styled::{island_central_panel, StyledCard, StyledCheckbox};
+use crate::ui::components::test_label::TestableWidget;
 use crate::ui::components::top_panel::add_top_panel;
 use crate::ui::theme::{DashColors, ThemeMode};
 use crate::ui::{RootScreenType, ScreenLike};
 use dash_sdk::dpp::dashcore::Network;
 use dash_sdk::dpp::identity::TimestampMillis;
 use eframe::egui::{self, Context, Ui};
+use egui::Widget;
 use std::path::PathBuf;
 use std::sync::Arc;
 use std::time::{Duration, SystemTime, UNIX_EPOCH};
@@ -257,10 +259,10 @@
                                                         .expect("Expected to save db settings");
                                                 } else {
                                                     self.custom_dash_qt_error_message =
                                                         Some(format!(
                                                     "Invalid file: Please select a valid '{}'.",
                                                     required_file_name
                                                 ));
                                                 }
                                             }
                                         }
@@ -456,10 +458,10 @@
                                     };
 
                                     ui.label(
                                         egui::RichText::new(format!(
                                             "Add these lines to your {} dash.conf:",
                                             network_name
                                         ))
                                         .color(DashColors::TEXT_PRIMARY),
                                     );
 
@@ -538,7 +540,11 @@
 
         // Network selection
         let mut is_selected = self.current_network == network;
-        if StyledCheckbox::new(&mut is_selected, "").show(ui).clicked() && is_selected {
+
+        let checkbox = StyledCheckbox::new(&mut is_selected, "")
+            .test_label(&format!("select_network_{}", network.magic()));
+
+        if checkbox.ui(ui).clicked() && is_selected {
             self.current_network = network;
             app_action = AppAction::SwitchNetwork(network);
             // Recheck in 1 second
@@ -609,7 +615,7 @@
                             if let Err(e) =
                                 Arc::clone(local_app_context).reinit_core_client_and_sdk()
                             {
                                 eprintln!("Failed to re-init local RPC client and sdk: {}", e);
                             } else {
                                 // Trigger SwitchNetworks
                                 app_action = AppAction::SwitchNetwork(Network::Regtest);