summaryrefslogtreecommitdiff
path: root/harnesses/host_controller/tfc/tfc_client_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'harnesses/host_controller/tfc/tfc_client_test.py')
-rw-r--r--harnesses/host_controller/tfc/tfc_client_test.py130
1 files changed, 0 insertions, 130 deletions
diff --git a/harnesses/host_controller/tfc/tfc_client_test.py b/harnesses/host_controller/tfc/tfc_client_test.py
deleted file mode 100644
index 10db850..0000000
--- a/harnesses/host_controller/tfc/tfc_client_test.py
+++ /dev/null
@@ -1,130 +0,0 @@
-#!/usr/bin/env python
-#
-# Copyright (C) 2017 The Android Open Source Project
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-
-import unittest
-
-try:
- from unittest import mock
-except ImportError:
- import mock
-
-from host_controller.tfc import tfc_client
-from host_controller.tfc import command_attempt
-from host_controller.tfc import device_info
-from host_controller.tfc import request
-
-
-class TfcClientTest(unittest.TestCase):
- """A test for tfc_client.TfcClient.
-
- Attributes:
- _client: The tfc_client.TfcClient being tested.
- _service: The mock service that _client connects to.
- """
- _DEVICE_INFOS = [device_info.DeviceInfo(
- device_serial="ABCDEF", group_name="group0",
- run_target="sailfish", state="Available")]
-
- def setUp(self):
- """Creates a TFC client connecting to a mock service."""
- self._service = mock.Mock()
- self._client = tfc_client.TfcClient(self._service)
-
- def testNewRequest(self):
- """Tests requests.new."""
- req = request.Request(cluster="cluster0",
- command_line="vts-codelab",
- run_target="sailfish",
- user="user0")
- self._client.NewRequest(req)
- self._service.assert_has_calls([
- mock.call.requests().new().execute()])
-
- def testLeaseHostTasks(self):
- """Tests tasks.leasehosttasks."""
- tasks = {"tasks": [{"request_id": "2",
- "command_line": "vts-codelab --serial ABCDEF",
- "task_id": "1-0",
- "device_serials": ["ABCDEF"],
- "command_id": "1"}]}
- self._service.tasks().leasehosttasks().execute.return_value = tasks
- self._client.LeaseHostTasks("cluster0", ["cluster1", "cluster2"],
- "host0", self._DEVICE_INFOS)
- self._service.assert_has_calls([
- mock.call.tasks().leasehosttasks().execute()])
-
- def testHostEvents(self):
- """Tests host_events.submit."""
- device_snapshots = [
- self._client.CreateDeviceSnapshot("vts-staging", "host0",
- self._DEVICE_INFOS),
- self._client.CreateDeviceSnapshot("vts-presubmit", "host0",
- self._DEVICE_INFOS)]
- self._client.SubmitHostEvents(device_snapshots)
- self._service.assert_has_calls([
- mock.call.host_events().submit().execute()])
-
- def testCommandEvents(self):
- """Tests command_events.submit."""
- cmd = command_attempt.CommandAttempt(
- task_id="321-0",
- attempt_id="abcd-1234",
- hostname="host0",
- device_serial="ABCDEF")
- expected_event = {
- "task_id": "321-0",
- "attempt_id": "abcd-1234",
- "hostname": "host0",
- "device_serial": "ABCDEF",
- "time": 1}
-
- normal_event = cmd.CreateCommandEvent(
- command_attempt.EventType.INVOCATION_STARTED,
- event_time=1)
- expected_event["type"] = command_attempt.EventType.INVOCATION_STARTED
- self.assertDictEqual(expected_event, normal_event)
-
- error_event = cmd.CreateCommandEvent(
- command_attempt.EventType.EXECUTE_FAILED,
- error="unit test", event_time=1.1)
- expected_event["type"] = command_attempt.EventType.EXECUTE_FAILED
- expected_event["data"] = {"error":"unit test"}
- self.assertDictEqual(expected_event, error_event)
-
- complete_event = cmd.CreateInvocationCompletedEvent(
- summary="complete", total_test_count=2, failed_test_count=1,
- error="unit test")
- expected_event["type"] = command_attempt.EventType.INVOCATION_COMPLETED
- expected_event["data"] = {"summary": "complete", "error": "unit test",
- "total_test_count": 2, "failed_test_count": 1}
- del expected_event["time"]
- self.assertDictContainsSubset(expected_event, complete_event)
- self.assertIn("time", complete_event)
-
- self._client.SubmitCommandEvents([
- normal_event, error_event, complete_event])
- self._service.assert_has_calls([
- mock.call.command_events().submit().execute()])
-
- def testWrongParameter(self):
- """Tests raising exception for wrong parameter name."""
- self.assertRaisesRegexp(KeyError, "sdk", device_info.DeviceInfo,
- device_serial="123", sdk="25")
-
-
-if __name__ == "__main__":
- unittest.main()