# 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()