ios/build/bots/scripts/plugin/test_plugin_client_test.py

# Copyright 2023 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import unittest
import os
import sys
import mock

# if the current directory is in scripts (pwd), then we need to
# add plugin in order to import from that directory
if os.path.split(os.path.dirname(__file__))[1] != 'plugin':
  sys.path.append(
      os.path.join(os.path.abspath(os.path.dirname(__file__)), 'plugin'))
from test_plugin_client import TestPluginClient
from plugin_constants import PLUGIN_PROTOS_PATH, PLUGIN_SERVICE_ADDRESS

sys.path.append(PLUGIN_PROTOS_PATH)
import test_plugin_service_pb2
import test_plugin_service_pb2_grpc


class TestPluginClientTest(unittest.TestCase):

  def setUp(self):
    self.client = TestPluginClient(PLUGIN_SERVICE_ADDRESS)
    self.mock_channel_stub = mock.Mock()
    self.client.channel_stub = self.mock_channel_stub

  def test_ListEnabledPlugins(self):
    request = test_plugin_service_pb2.TestCaseWillStartRequest()
    self.client.ListEnabledPlugins(request)
    self.client.channel_stub.ListEnabledPlugins.assert_called_with(request)

  def test_TestCaseWillStart(self):
    request = test_plugin_service_pb2.TestCaseWillStartRequest()
    self.client.TestCaseWillStart(request)
    self.client.channel_stub.TestCaseWillStart.assert_called_with(request)

  def test_TestCaseDidFinish(self):
    request = test_plugin_service_pb2.TestCaseDidFinishRequest()
    self.client.TestCaseDidFinish(request)
    self.client.channel_stub.TestCaseDidFinish.assert_called_with(request)

  def test_TestCaseDidFail(self):
    request = test_plugin_service_pb2.TestCaseDidFailRequest()
    self.client.TestCaseDidFail(request)
    self.client.channel_stub.TestCaseDidFail.assert_called_with(request)


if __name__ == '__main__':
  unittest.main()