1# Copyright 2021 - The Android Open Source Project
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#     http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14"""Tests for hostcleanup."""
15import unittest
16
17from unittest import mock
18
19from acloud.internal.lib import driver_test_lib
20from acloud.hostcleanup import hostcleanup
21from acloud.hostcleanup import host_cleanup_runner
22
23
24class HostcleanupTest(driver_test_lib.BaseDriverTest):
25    """Test hostcleanup."""
26
27    # pylint: disable=no-self-use
28    @mock.patch.object(host_cleanup_runner, "PackagesUninstaller")
29    def testRun(self, mock_uninstallpkgs):
30        """test Run."""
31        args = mock.MagicMock()
32        hostcleanup.Run(args)
33        mock_uninstallpkgs.assert_called_once()
34
35
36if __name__ == '__main__':
37    unittest.main()
38