| # Copyright 2026 The Fuchsia Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import unittest |
| from unittest import mock |
| |
| import find_affected |
| |
| |
| class TestFindAffected(unittest.IsolatedAsyncioTestCase): |
| def test_format_affected_targets(self) -> None: |
| """Tests the label aggregation and mapping algorithm to execution commands.""" |
| input_dict = { |
| "//src/my_test:foo": find_affected.FormattedResult( |
| False, ["core.x64", "core.arm64"] |
| ), |
| "//src/other:bar": find_affected.FormattedResult( |
| False, ["core.x64"] |
| ), |
| "//src/host:x64": find_affected.FormattedResult(True, ["core.x64"]), |
| "//src/arm:arm64": find_affected.FormattedResult( |
| False, ["core.arm64"] |
| ), |
| } |
| |
| output = find_affected.format_affected_targets(input_dict) |
| self.assertEqual(len(output), 4) |
| |
| # Expected to be sorted alphabetically by label! |
| self.assertEqual(output[0].pure_label, "//src/arm:arm64") |
| self.assertEqual(output[0].command, "fx add-test //src/arm:arm64") |
| self.assertListEqual(output[0].pb_configs, ["core.arm64"]) |
| |
| self.assertEqual(output[1].pure_label, "//src/host:x64") |
| self.assertEqual(output[1].command, "fx add-host-test //src/host:x64") |
| self.assertListEqual(output[1].pb_configs, ["core.x64"]) |
| |
| self.assertEqual(output[2].pure_label, "//src/my_test:foo") |
| self.assertEqual(output[2].command, "fx add-test //src/my_test:foo") |
| self.assertListEqual(output[2].pb_configs, ["core.x64", "core.arm64"]) |
| |
| self.assertEqual(output[3].pure_label, "//src/other:bar") |
| self.assertEqual(output[3].command, "fx add-test //src/other:bar") |
| self.assertListEqual(output[3].pb_configs, ["core.x64"]) |
| |
| def test_clean_gathered_results(self) -> None: |
| """Tests merging results from multiple build invocations into a target-to-product mapping.""" |
| # Simulated raw output from run_find_affected |
| results = [ |
| find_affected.GatheredResult( |
| "core.x64", |
| [ |
| find_affected.AffectedResult("//src/my_test:foo", False), |
| find_affected.AffectedResult("//src/other:bar", False), |
| ], |
| ), |
| find_affected.GatheredResult( |
| "core.arm64", |
| [ |
| find_affected.AffectedResult("//src/my_test:foo", False), |
| find_affected.AffectedResult("not_a_label", False), |
| ], |
| ), |
| find_affected.GatheredResult("minimal.x64", []), |
| ] |
| |
| mapped = find_affected.clean_gathered_results(results) |
| |
| self.assertEqual(len(mapped), 2) |
| # Should drop 'not_a_label' and aggregate the configs. |
| self.assertListEqual( |
| mapped["//src/my_test:foo"].pb_configs, ["core.x64", "core.arm64"] |
| ) |
| self.assertFalse(mapped["//src/my_test:foo"].is_host) |
| self.assertListEqual(mapped["//src/other:bar"].pb_configs, ["core.x64"]) |
| self.assertFalse(mapped["//src/other:bar"].is_host) |
| |
| @mock.patch("find_affected.FxCmd") |
| @mock.patch("execution.run_command") |
| async def test_find_affected_tests( |
| self, mock_run_command: mock.AsyncMock, mock_fx_cmd_class: mock.Mock |
| ) -> None: |
| """Tests the full find_affected_tests flow with successful command execution.""" |
| mock_fx = mock_fx_cmd_class.return_value |
| mock_fx.start = mock.AsyncMock() |
| mock_fx_running = mock_fx.start.return_value |
| mock_fx_running.run_to_completion = mock.AsyncMock( |
| return_value=mock.Mock(return_code=0) |
| ) |
| |
| mock_run_command.return_value = mock.Mock( |
| stdout="//src/test1:foo,device\n//src/test2:bar,host\n", |
| return_code=0, |
| ) |
| |
| gathered = await find_affected.find_affected_tests( |
| "/fuchsia", "core.x64", "/out/core_x64", ["//bundle"], "/tmp/files" |
| ) |
| |
| self.assertEqual(gathered.product_board, "core.x64") |
| self.assertListEqual( |
| gathered.affected_results, |
| [ |
| find_affected.AffectedResult("//src/test1:foo", False), |
| find_affected.AffectedResult("//src/test2:bar", True), |
| ], |
| ) |
| |
| # Verify FxCmd was initialized correctly |
| mock_fx_cmd_class.assert_called_once_with( |
| build_directory="/out/core_x64" |
| ) |
| # Verify fx set was called |
| mock_fx.start.assert_called_once_with( |
| "set", |
| "core.x64", |
| "--no-change-env", |
| "--rbe-mode=off", |
| "--with", |
| "//bundle", |
| ) |
| # Verify build-api-client was called |
| mock_run_command.assert_called_once() |
| args = mock_run_command.call_args[0] |
| self.assertIn("/fuchsia/build/api/client", args[0]) |
| self.assertIn("affected_tests", args) |
| |
| @mock.patch("find_affected.FxCmd") |
| async def test_find_affected_tests_fx_set_failure( |
| self, mock_fx_cmd_class: mock.Mock |
| ) -> None: |
| """Tests that find_affected_tests returns empty results if fx set fails.""" |
| mock_fx = mock_fx_cmd_class.return_value |
| mock_fx.start = mock.AsyncMock() |
| mock_fx_running = mock_fx.start.return_value |
| # Return non-zero return code for fx set. |
| mock_fx_running.run_to_completion = mock.AsyncMock( |
| return_value=mock.Mock(return_code=1) |
| ) |
| |
| gathered = await find_affected.find_affected_tests( |
| "/fuchsia", "core.x64", "/out/core_x64", [], "/tmp/files" |
| ) |
| |
| self.assertEqual(gathered.product_board, "core.x64") |
| self.assertListEqual(gathered.affected_results, []) |
| |
| @mock.patch("find_affected.FxCmd") |
| async def test_find_affected_tests_exception( |
| self, mock_fx_cmd_class: mock.Mock |
| ) -> None: |
| """Tests that find_affected_tests handles unexpected exceptions gracefully.""" |
| mock_fx = mock_fx_cmd_class.return_value |
| # Raise an exception when starting the command. |
| mock_fx.start = mock.AsyncMock( |
| side_effect=RuntimeError("Generic error") |
| ) |
| |
| gathered = await find_affected.find_affected_tests( |
| "/fuchsia", "core.x64", "/out/core_x64", [], "/tmp/files" |
| ) |
| |
| self.assertEqual(gathered.product_board, "core.x64") |
| self.assertListEqual(gathered.affected_results, []) |
