#!/usr/bin/env vpython3
# Copyright 2025 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import logging
import unittest

import exception_utils


class DeviceExceptionCheckerTest(unittest.TestCase):
  def setUp(self):
    self.exception_checker = exception_utils.DeviceExceptionChecker()

  def test_device_setup_not_complete_error_raised(self):
    log_message = "(Underlying Error: The operation couldn’t be completed. Device setup is not yet complete.)"

    self.exception_checker.check_line(log_message)

    self.assertRaises(
      exception_utils.DeviceSetupNotCompleteError,
      self.exception_checker.throw_first,
    )

  def test_failed_to_install_embedded_profile_error_raised(self):
    log_message = "Recovery Suggestion: Failed to install embedded profile for"

    self.exception_checker.check_line(log_message)

    self.assertRaises(
      exception_utils.FailedToInstallEmbeddedProfileError,
      self.exception_checker.throw_first,
    )

  def test_developer_mode_disabled_error_raised(self):
    log_message = (
      "name:iPhone, error:Developer Mode disabled To use iPhone"
      + " for development, enable Developer Mode in Settings → "
      + "Privacy & Security."
    )

    self.exception_checker.check_line(log_message)

    self.assertRaises(
      exception_utils.DeveloperModeDisabledError,
      self.exception_checker.throw_first,
    )


if __name__ == '__main__':
  logging.basicConfig(
    format='[%(asctime)s:%(levelname)s] %(message)s',
    level=logging.DEBUG,
    datefmt='%I:%M:%S',
  )
  unittest.main()