MicroPythonOS/tests/test_graphical_osupdate.py

import unittest
import lvgl as lv
import mpos

# Import graphical test helper
from mpos import (
    wait_for_render,
    find_label_with_text,
    verify_text_present,
    print_screen_labels,
    BuildInfo,
    AppManager
)


class TestOSUpdateGraphicalUI(unittest.TestCase):
    """Graphical tests for OSUpdate app UI state."""

    def tearDown(self):
        """Clean up after each test method."""
        # Navigate back to launcher
        mpos.ui.back_screen()
        wait_for_render(5)

    def test_app_launches_successfully(self):
        """Test that OSUpdate app launches without errors."""
        result = AppManager.start_app("com.micropythonos.osupdate")

        self.assertTrue(result, "Failed to start OSUpdate app")
        wait_for_render(10)

        # Get active screen
        screen = lv.screen_active()
        self.assertIsNotNone(screen, "No active screen after launch")

    def test_ui_elements_exist(self):
        """Test that all required UI elements are created."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Find UI elements by searching for labels/text
        current_version_label = find_label_with_text(screen, "Installed OS version")
        self.assertIsNotNone(current_version_label, "Current version label not found")

        # Check for update button text (case insensitive)
        # Button text will be "Update OS", "Reinstall\nsame version", or "Install\nolder version"
        update_button_found = verify_text_present(screen, "Update") or verify_text_present(screen, "update") or \
                             verify_text_present(screen, "Reinstall") or verify_text_present(screen, "Install")
        self.assertTrue(update_button_found, "Update button text not found")


    def test_install_button_text_exists(self):
        """Test that install button with update text exists on screen."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Verify the button text is present - it will be "Update OS" initially
        # (or "Reinstall\nsame version" or "Install\nolder version" depending on version comparison)
        button_text_found = verify_text_present(screen, "Update OS") or \
                           verify_text_present(screen, "Reinstall") or \
                           verify_text_present(screen, "Install")
        self.assertTrue(button_text_found, "Install button text should be present on screen")

    def test_current_version_displayed(self):
        """Test that current OS version is displayed correctly."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Find version label
        version_label = find_label_with_text(screen, "Installed OS version:")
        self.assertIsNotNone(version_label, "Version label not found")

        # Check that it contains the current version
        label_text = version_label.get_text()
        current_version = BuildInfo.version.release
        self.assertIn(current_version, label_text,
                     f"Current version {current_version} not in label text: {label_text}")

    def test_initial_status_message_without_wifi(self):
        """Test status message when wifi is not connected."""
        # This test assumes desktop mode where wifi check returns True
        # On actual hardware without wifi, it would show error
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Should show either "Checking for OS updates..." or update info
        # or wifi error depending on network state
        checking_found = verify_text_present(screen, "Checking") or \
                        verify_text_present(screen, "version") or \
                        verify_text_present(screen, "WiFi")
        self.assertTrue(checking_found, "Should show some status message")

    def test_initial_state_labels(self):
        """Print initial app labels for debugging."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(20)

        screen = lv.screen_active()

        # Print labels for debugging
        print("\n=== OSUpdate Initial State Labels ===")
        print_screen_labels(screen)

class TestOSUpdateGraphicalStatusMessages(unittest.TestCase):
    """Graphical tests for OSUpdate status messages."""

    def tearDown(self):
        """Clean up after test."""
        mpos.ui.back_screen()
        wait_for_render(5)

    def test_status_label_exists(self):
        """Test that status label is created and visible."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Status label should exist and show some text
        # Look for common status messages
        has_status = (
            verify_text_present(screen, "Checking") or
            verify_text_present(screen, "version") or
            verify_text_present(screen, "WiFi") or
            verify_text_present(screen, "Error") or
            verify_text_present(screen, "Update")
        )
        self.assertTrue(has_status, "Status label should be present with some message")

    def test_all_labels_readable(self):
        """Test that all labels are readable (no truncation issues)."""
        result = AppManager.start_app("com.micropythonos.osupdate")
        self.assertTrue(result)
        wait_for_render(15)

        screen = lv.screen_active()

        # Print all labels to verify they're readable
        print("\n=== All OSUpdate Labels ===")
        print_screen_labels(screen)

        # At minimum, should have version label
        version_found = verify_text_present(screen, "Installed OS version")
        self.assertTrue(version_found, "Version label should be present and readable")