import subprocess
import unittest
from unittest.mock import MagicMock
from steps import build_image
import inspect

class TestBuildImage(unittest.TestCase):
    def setUp(self):
        self.step = build_image.BuildImage()
        self.step.export_machine = MagicMock(return_value="")
        self.step.source_setup = MagicMock(return_value="")
        self.step.umask = MagicMock(return_value="")
        self.step.export_gbmc_config = MagicMock(return_value="")
        self.step.export_gbmc_version = MagicMock(return_value="")
        self.step.export_bb_env_extrawhite = MagicMock(return_value="")
        self.step.bitbake = MagicMock(return_value="")
        subprocess.run = MagicMock()

        self.step.run_script()

    def test_export_machine(self):
        """Tests that export_machine method is called"""
        self.step.export_machine.assert_called_with()

    def test_source_setup(self):
        """Tests that source_setup method is called"""
        self.step.source_setup.assert_called_with()

    def test_umask(self):
        """Tests that umask method i called"""
        self.step.umask.assert_called_with()

    def test_export_gbmc_config(self):
        """Tests that export_gbmc_config is called"""
        self.step.export_gbmc_config.assert_called_with()

    def test_export_gbmc_version(self):
        """Tests that export_gbmc_version is called"""
        self.step.export_gbmc_version.assert_called_with()

    def test_export_bb_env_extrawhite(self):
        """Tests that export_bb_env_extrawhite is called"""
        self.step.export_bb_env_extrawhite.assert_called_with()

    def test_bitbake(self):
        """Tests that bitbake method is called"""
        self.step.bitbake.assert_called_with()

    def test_subprocess_run(self):
        """Tests subprocess's arguments, tests the order in which all the previous
        tests are called
        """
        self.step = build_image.BuildImage()
        self.step.firmware_machine = "machine0"
        self.step.firmware_version = "x.y.z"
        self.step.gbmc_config = "test"
        subprocess.run = MagicMock()

        self.step.run_script()

        command =   "export MACHINE=machine0 " \
                    "&& source setup machine0 " \
                    "&& umask 002 " \
                    "&& export GBMC_CONFIG=test " \
                    "&& export GBMC_VERSION=x.y.z " \
                    "&& export BB_ENV_EXTRAWHITE=${BB_ENV_EXTRAWHITE} GBMC_CONFIG GBMC_VERSION " \
                    "&& bitbake -k obmc-phosphor-image"

        subprocess.run.assert_called_with(["bash", "-c", command])