| import sh |
| import unittest |
| from unittest.mock import MagicMock |
| from steps import gsutil_setup |
| |
| |
| class TestGsutilSetup(unittest.TestCase): |
| def setUp(self): |
| self.init_step_mock() |
| |
| self.step.use_cache = True |
| self.step.create_sstate_cache_dir = MagicMock() |
| self.step.create_sources_dir = MagicMock() |
| self.step.fetch_sstate_cache = MagicMock() |
| self.step.fetch_sources = MagicMock() |
| self.step.chown_sstate_cache = MagicMock() |
| |
| self.step.run_script() |
| |
| def init_step_mock(self): |
| self.step = gsutil_setup.GsutilSetup() |
| self.step.firmware_machine = "machine0" |
| self.step.gcs_bucket = "a-bucket" |
| |
| def test_create_sstate_cache_dir(self): |
| """Tests that sstate_cache_dir method is called""" |
| self.step.create_sstate_cache_dir.assert_called_with() |
| |
| def test_create_sources_dir(self): |
| """Tests that create_sources method is called""" |
| self.step.create_sources_dir.assert_called_with() |
| |
| def test_fetch_sstate_cache(self): |
| """Tests that fetch_sstate_cache method is called""" |
| self.step.fetch_sstate_cache.assert_called_with() |
| |
| def test_fetch_sources(self): |
| """Tests that fetch_sources method is called""" |
| self.step.fetch_sources.assert_called_with() |
| |
| def test_disable_sstate_cache(self): |
| """Tests the program state when use_cache=False""" |
| self.init_step_mock() |
| |
| self.step.use_cache = False |
| self.step.create_sstate_cache_dir = MagicMock() |
| self.step.create_sources_dir = MagicMock() |
| self.step.fetch_sstate_cache = MagicMock() |
| self.step.fetch_sources = MagicMock() |
| self.step.chown_sstate_cache = MagicMock() |
| |
| self.step.run_script() |
| |
| self.step.create_sstate_cache_dir.assert_not_called() |
| self.step.create_sources_dir.assert_called_with() |
| self.step.fetch_sstate_cache.assert_not_called() |
| self.step.fetch_sources.assert_called_with() |
| self.step.chown_sstate_cache.assert_not_called() |
| |
| def test_mkdir(self): |
| """Tests the call order of the mkdir calls""" |
| self.init_step_mock() |
| self.step.fetch_sstate_cache = MagicMock() |
| self.step.fetch_sources = MagicMock() |
| self.step.chown_sstate_cache = MagicMock() |
| self.step.use_cache = True |
| sh.mkdir = MagicMock() |
| |
| self.step.run_script() |
| |
| actual_calls = sh.mkdir.call_args_list |
| expected_calls = [(("-p", "build/machine0/sstate-cache"),), |
| (("-p", "build/sources"),),] |
| |
| self.assertEqual(actual_calls, expected_calls) |
| |
| def test_gsutil(self): |
| """Tests the order in which gsutil is called, and their arguments""" |
| self.init_step_mock() |
| self.step.use_cache = True |
| self.step.create_sstate_cache_dir = MagicMock() |
| self.step.create_sources_dir = MagicMock() |
| self.step.gsutil = MagicMock() |
| self.step.chown_sstate_cache = MagicMock() |
| |
| self.step.run_script() |
| |
| actual_calls = self.step.gsutil.call_args_list |
| expected_calls = [(("rsync", "gs://a-bucket/cache-unzip/machine0", |
| "build/machine0/sstate-cache"),), |
| (("rsync", "gs://a-bucket/sources", "build/sources"),)] |
| |
| self.assertEqual(actual_calls, expected_calls) |
| |
| def test_chown(self): |
| """Tests that sh.chown is called with the right arguments when use_cache=True""" |
| self.init_step_mock() |
| self.step.use_cache = True |
| self.step.create_sstate_cache_dir = MagicMock() |
| self.step.create_sources_dir = MagicMock() |
| self.step.fetch_sstate_cache = MagicMock() |
| self.step.fetch_sources = MagicMock() |
| sh.chown = MagicMock() |
| |
| self.step.run_script() |
| |
| sh.chown("-R", "build", "build/machine0/sstate-cache") |
| |
| self.step.chown_sstate_cache = MagicMock() |
| |
| self.step.run_script() |
| |
| self.step.chown_sstate_cache.assert_called_with() |
