src/testing/end_to_end/honeydew/tests/functional_tests/affordance_tests/test_system_power_state_controller/test_system_power_state_controller.py

#!/usr/bin/env fuchsia-vendored-python
# Copyright 2024 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Mobly test for SystemPowerStateController affordance."""

import logging

from fuchsia_base_test import fuchsia_base_test
from mobly import asserts, test_runner

from honeydew import errors
from honeydew.interfaces.device_classes import fuchsia_device

_LOGGER: logging.Logger = logging.getLogger(__name__)


class SystemPowerStateControllerAffordanceTests(
    fuchsia_base_test.FuchsiaBaseTest
):
    """SystemPowerStateController affordance tests"""

    def setup_class(self) -> None:
        """setup_class is called once before running tests."""
        super().setup_class()
        self.device: fuchsia_device.FuchsiaDevice = self.fuchsia_devices[0]

    def test_idle_idle_suspend_auto_resume(self) -> None:
        """Test case for SystemPowerStateController.idle_suspend_auto_resume()"""
        if self.user_params["is_starnix_supported"]:
            self.device.system_power_state_controller.idle_suspend_auto_resume()
        else:
            with asserts.assert_raises(errors.NotSupportedError):
                self.device.system_power_state_controller.idle_suspend_auto_resume()


if __name__ == "__main__":
    test_runner.main()