MicroPythonOS/tests/test_graphical_imu_calibration.py

"""
Graphical test for IMU calibration activities.

Tests both CheckIMUCalibrationActivity and CalibrateIMUActivity
with mock data on desktop.

Usage:
    Desktop: ./tests/unittest.sh tests/test_graphical_imu_calibration.py
    Device:  ./tests/unittest.sh tests/test_graphical_imu_calibration.py --ondevice
"""

import unittest
import lvgl as lv
import mpos.ui
import os
import sys
import time
from mpos import (
    wait_for_render,
    capture_screenshot,
    find_label_with_text,
    verify_text_present,
    print_screen_labels,
    simulate_click,
    get_widget_coords,
    find_button_with_text,
    click_label,
    click_button,
    find_text_on_screen,
    AppManager
)


class TestIMUCalibration(unittest.TestCase):
    """Test suite for IMU calibration activities."""

    def setUp(self):
        """Set up test fixtures."""
        # Get screenshot directory
        if sys.platform == "esp32":
            self.screenshot_dir = "tests/screenshots"
        else:
            self.screenshot_dir = "../tests/screenshots" # it runs from internal_filesystem/

        # Ensure directory exists
        try:
            os.mkdir(self.screenshot_dir)
        except OSError:
            pass

    def tearDown(self):
        """Clean up after test."""
        # Navigate back to launcher
        try:
            for _ in range(3):  # May need multiple backs
                mpos.ui.back_screen()
                wait_for_render(5)
        except:
            pass

    def test_check_calibration_activity_loads(self):
        """Test that CheckIMUCalibrationActivity loads and displays."""
        print("\n=== Testing CheckIMUCalibrationActivity ===")

        # Navigate: Launcher -> Settings -> Check IMU Calibration
        result = AppManager.start_app("com.micropythonos.settings")
        self.assertTrue(result, "Failed to start Settings app")
        wait_for_render(15)

        # Initialize touch device with dummy click
        simulate_click(10, 10)
        wait_for_render(10)

        print("Clicking 'Check IMU Calibration' menu item...")
        self.assertTrue(click_label("Check IMU Calibration"), "Could not find Check IMU Calibration menu item")
        wait_for_render(iterations=20)

        # Verify key elements are present
        screen = lv.screen_active()
        print_screen_labels(screen)
        self.assertTrue(verify_text_present(screen, "Quality:"), "Quality label not found")
        self.assertTrue(verify_text_present(screen, "Accel."), "Accel. label not found")
        self.assertTrue(verify_text_present(screen, "Gyro"), "Gyro label not found")

        # Capture screenshot
        screenshot_path = f"{self.screenshot_dir}/check_imu_calibration.raw"
        print(f"Capturing screenshot: {screenshot_path}")
        capture_screenshot(screenshot_path)

        # Verify screenshot saved
        stat = os.stat(screenshot_path)
        self.assertTrue(stat[6] > 0, "Screenshot file is empty")

        print("=== CheckIMUCalibrationActivity test complete ===")

    def test_calibrate_activity_flow(self):
        """Test CalibrateIMUActivity full calibration flow."""
        print("\n=== Testing CalibrateIMUActivity Flow ===")

        # Navigate: Launcher -> Settings -> Calibrate IMU
        result = AppManager.start_app("com.micropythonos.settings")
        self.assertTrue(result, "Failed to start Settings app")
        wait_for_render(15)

        # Initialize touch device with dummy click
        simulate_click(10, 10)
        wait_for_render(10)

        print("Clicking 'Calibrate IMU' menu item...")
        self.assertTrue(click_label("Calibrate IMU"), "Could not find Calibrate IMU item")
        wait_for_render(iterations=20)

        # Verify activity loaded and shows instructions
        screen = lv.screen_active()
        print_screen_labels(screen)
        self.assertTrue(verify_text_present(screen, "IMU Calibration"),
                       "CalibrateIMUActivity title not found")
        self.assertTrue(verify_text_present(screen, "Place device on flat"),
                       "Instructions not shown")

        # Capture initial state
        screenshot_path = f"{self.screenshot_dir}/calibrate_imu_01_initial.raw"
        capture_screenshot(screenshot_path)

        # Click "Calibrate Now" button to start calibration
        calibrate_btn = find_button_with_text(screen, "Calibrate Now")
        self.assertIsNotNone(calibrate_btn, "Could not find 'Calibrate Now' button")
        # Use send_event instead of simulate_click (more reliable)
        calibrate_btn.send_event(lv.EVENT.CLICKED, None)
        wait_for_render(10)

        # Wait for calibration to complete (mock takes ~3 seconds)
        time.sleep(4)
        wait_for_render(40)

        # Verify calibration completed
        screen = lv.screen_active()
        print_screen_labels(screen)
        self.assertTrue(verify_text_present(screen, "Calibration successful!"),
                       "Calibration completion message not found")

        # Verify offsets are shown
        self.assertTrue(verify_text_present(screen, "Accel offsets") or
                       verify_text_present(screen, "offsets"),
                       "Calibration offsets not shown")

        # Capture completion state
        screenshot_path = f"{self.screenshot_dir}/calibrate_imu_02_complete.raw"
        capture_screenshot(screenshot_path)

        print("=== CalibrateIMUActivity flow test complete ===")

    def test_navigation_from_check_to_calibrate(self):
        """Test navigation from Check to Calibrate activity via button."""
        print("\n=== Testing Check -> Calibrate Navigation ===")

        # Navigate to Check activity
        result = AppManager.start_app("com.micropythonos.settings")
        self.assertTrue(result)
        wait_for_render(15)

        # Initialize touch device with dummy click
        simulate_click(10, 10)
        wait_for_render(10)

        print("Clicking 'Check IMU Calibration' menu item...")
        self.assertTrue(click_label("Check IMU Calibration"), "Could not find Check IMU Calibration menu item")
        wait_for_render(iterations=20)

        # Click "Calibrate" button to navigate to Calibrate activity
        screen = lv.screen_active()
        calibrate_btn = find_button_with_text(screen, "Calibrate")
        self.assertIsNotNone(calibrate_btn, "Could not find 'Calibrate' button")

        # Use send_event instead of simulate_click (more reliable for navigation)
        calibrate_btn.send_event(lv.EVENT.CLICKED, None)
        wait_for_render(30)

        # Verify CalibrateIMUActivity loaded
        screen = lv.screen_active()
        print_screen_labels(screen)
        self.assertTrue(verify_text_present(screen, "Calibrate Now"),
                       "Did not navigate to CalibrateIMUActivity")
        self.assertTrue(verify_text_present(screen, "Place device on flat"),
                       "CalibrateIMUActivity instructions not shown")

        print("=== Navigation test complete ===")