| import subprocess |
| import sh |
| import unittest |
| from unittest.mock import MagicMock |
| from steps import machine_setup |
| |
| class TestMachineSetup(unittest.TestCase): |
| def setUp(self): |
| self.init_step_mock() |
| self.step.export_machine = MagicMock(return_value="") |
| self.step.source_setup = MagicMock(return_value="") |
| self.step.umask = MagicMock(return_value="") |
| |
| self.step.run_script() |
| |
| def init_step_mock(self): |
| self.step = machine_setup.MachineSetup() |
| self.step.firmware_machine = "machine0" |
| self.step.main_repo_name = "repo0" |
| sh.cd = MagicMock() |
| subprocess.run = MagicMock() |
| |
| 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 is called""" |
| self.step.umask.assert_called_with() |
| |
| def test_cd(self): |
| """Tests that run_script changes into the main_repo_name dir""" |
| sh.cd.assert_called_with(self.step.main_repo_name) |
| |
| def test_subprocess(self): |
| """Tests the arguments of subprocess, and the order in which each |
| sub-commands are called (separated by &&) |
| """ |
| self.init_step_mock() |
| |
| self.step.run_script() |
| |
| command = "export MACHINE=machine0 " \ |
| "&& source setup machine0 " \ |
| "&& umask 002" |
| |
| subprocess.run.assert_called_with(["bash", "-c", command]) |
