#
# Copyright (c) 2020, 2024 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#

import uuid

from dcmanager.common import consts
from dcmanager.tests.unit.audit.test_firmware_audit_manager import DeviceImageState
from dcmanager.tests.unit.fakes import FakeVimStrategy
from dcmanager.tests.unit.orchestrator.states.fakes import FakeDevice
from dcmanager.tests.unit.orchestrator.states.fakes import FakeDeviceImage
from dcmanager.tests.unit.orchestrator.states.fakes import FakeDeviceLabel
from dcmanager.tests.unit.orchestrator.test_base import TestSwUpdate


class TestFwUpdateState(TestSwUpdate):

    # Setting DEFAULT_STRATEGY_TYPE to firmware will setup the firmware
    # orchestration worker, and will mock away the other orch threads
    DEFAULT_STRATEGY_TYPE = consts.SW_UPDATE_TYPE_FIRMWARE

    def setUp(self):
        super().setUp()

    def _create_fake_strategy(self, state):
        return FakeVimStrategy(state=state)

    def _create_fake_device(self, pvendor_id, pdevice_id, enabled=True):
        return FakeDevice(
            str(uuid.uuid4()),
            pvendor_id=pvendor_id,
            pdevice_id=pdevice_id,
            enabled=enabled
        )

    def _create_fake_device_label(self, label_key, label_value, pcidevice_uuid):
        return FakeDeviceLabel(
            label_key=label_key,
            label_value=label_value,
            pcidevice_uuid=pcidevice_uuid
        )

    def _create_fake_device_image(
        self, pci_vendor, pci_device, applied, applied_labels
    ):
        return FakeDeviceImage(
            str(uuid.uuid4()),
            pci_vendor=pci_vendor,
            pci_device=pci_device,
            applied=applied,
            applied_labels=applied_labels
        )

    def _create_fake_device_image_state(self, pcidevice_uuid, image_uuid, status):
        return DeviceImageState(
            pcidevice_uuid,
            image_uuid,
            status
        )