1#!/usr/bin/env python3 2# 3# Copyright 2022 - The Android Open Source Project 4# 5# Licensed under the Apache License, Version 2.0 (the "License"); 6# you may not use this file except in compliance with the License. 7# You may obtain a copy of the License at 8# 9# http://www.apache.org/licenses/LICENSE-2.0 10# 11# Unless required by applicable law or agreed to in writing, software 12# distributed under the License is distributed on an "AS IS" BASIS, 13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 14# See the License for the specific language governing permissions and 15# limitations under the License. 16 17"""Unit tests for RemoteHostClient.""" 18 19import unittest 20from unittest import mock 21 22from acloud.internal.lib import driver_test_lib 23from acloud.internal.lib import remote_host_client 24 25 26class RemoteHostClientTest(driver_test_lib.BaseDriverTest): 27 """Unit tests for RemoteHostClient.""" 28 29 _IP_ADDRESS = "192.0.2.1" 30 31 def testGetInstanceIP(self): 32 """Test GetInstanceIP.""" 33 client = remote_host_client.RemoteHostClient(self._IP_ADDRESS) 34 ip_addr = client.GetInstanceIP("name") 35 self.assertEqual(ip_addr.external, self._IP_ADDRESS) 36 self.assertEqual(ip_addr.internal, self._IP_ADDRESS) 37 38 def testRecordTime(self): 39 """Test RecordTime and execution_time.""" 40 client = remote_host_client.RemoteHostClient(self._IP_ADDRESS) 41 self.assertFalse(client.execution_time) 42 with mock.patch( 43 "acloud.internal.lib.remote_host_client.time") as mock_time: 44 mock_time.time.return_value = 1.0 45 self.assertEqual(1.0, client.RecordTime("TIME", 0.25)) 46 self.assertDictEqual({"TIME": 0.75}, client.execution_time) 47 48 49if __name__ == "__main__": 50 unittest.main() 51