tests/crossbench/pinpoint/test_list_bots.py

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

from __future__ import annotations

from unittest import mock

import requests

from crossbench.exception import MultiException
from crossbench.pinpoint.api import PINPOINT_CONFIG_API_URL
from crossbench.pinpoint.list_bots import fetch_bots
from tests import test_helper
from tests.crossbench.pinpoint.requests_mixin import MockRequestsMixin


class ListBotsTest(MockRequestsMixin):

  def setUp(self):
    super().setUp()

    def mock_post_side_effect(url, *args, **kwargs):
      self.assertEqual(url, PINPOINT_CONFIG_API_URL,
                       f"Unexpected URL called: {url}")

      mock_response = mock.Mock()
      mock_response.json.return_value = {
          "configurations": ["bot1", "bot2", "bot3"]
      }
      mock_response.raise_for_status.return_value = None
      return mock_response

    self.mock_post.side_effect = mock_post_side_effect

  def test_fetch_bots(self):
    bots = fetch_bots()

    self.assertEqual(bots, ["bot1", "bot2", "bot3"])
    self.mock_post.assert_called_once_with(PINPOINT_CONFIG_API_URL)

  def test_fetch_bots_api_error(self):
    self.mock_post.side_effect = requests.exceptions.HTTPError("API Error")

    with self.assertRaises(MultiException):
      fetch_bots()

    self.mock_post.assert_called_once_with(PINPOINT_CONFIG_API_URL)


if __name__ == "__main__":
  test_helper.run_pytest(__file__)