old_contents: import unittest
  
  from requests.exceptions import ConnectionError
  
  from betfairlightweight import APIClient
  from betfairlightweight.endpoints.keepalive import KeepAlive, APIError
  from betfairlightweight.exceptions import KeepAliveError
  from betfairlightweight.resources.authresources import KeepAliveResource
  from tests import mock
  from tests.unit.tools import create_mock_json
  
  
  class KeepAliveTest(unittest.TestCase):
  
      def setUp(self):
          client = APIClient('username', 'password', 'app_key', 'UK')
          self.keep_alive = KeepAlive(client)
  
      @mock.patch('betfairlightweight.endpoints.keepalive.KeepAlive.request')
      def test_call(self, mock_response):
          mock = create_mock_json('tests/resources/keep_alive_success.json')
          mock_response.return_value = (mock.json(), 1.3)
          response = self.keep_alive()
  
          assert isinstance(response, KeepAliveResource)
          assert self.keep_alive.client.session_token == mock.json().get('token')
  
      @mock.patch('betfairlightweight.baseclient.BaseClient.cert')
      @mock.patch('betfairlightweight.baseclient.BaseClient.keep_alive_headers')
      @mock.patch('betfairlightweight.baseclient.requests.post')
      def test_request(self, mock_post, mock_keep_alive_headers, mock_cert):
          mock_response = create_mock_json('tests/resources/logout_success.json')
          mock_post.return_value = mock_response
  
          url = 'https://identitysso.betfair.com/api/keepAlive'
          response = self.keep_alive.request()
  
          mock_post.assert_called_once_with(url, headers=mock_keep_alive_headers, cert=mock_cert)
          assert response[0] == mock_response.json()
  
      @mock.patch('betfairlightweight.baseclient.BaseClient.cert')
      @mock.patch('betfairlightweight.baseclient.BaseClient.keep_alive_headers')
      @mock.patch('betfairlightweight.baseclient.requests.post')
      def test_request_error(self, mock_post, mock_keep_alive_headers, mock_cert):
          mock_post.side_effect = ValueError()
  
          with self.assertRaises(APIError):
              self.keep_alive.request()
  
          mock_post.side_effect = ConnectionError()
  
          with self.assertRaises(APIError):
              self.keep_alive.request()
  
      def test_keep_alive_error_handler(self):
          mock_response = create_mock_json('tests/resources/keep_alive_success.json')
          assert self.keep_alive._error_handler(mock_response.json()) is None
  
          mock_response = create_mock_json('tests/resources/keep_alive_fail.json')
          with self.assertRaises(KeepAliveError):
              self.keep_alive._error_handler(mock_response.json())
  
      def test_url(self):
          assert self.keep_alive.url == 'https://identitysso.betfair.com/api/keepAlive'
  
new_contents: import unittest
  
  from requests.exceptions import ConnectionError
  
  from betfairlightweight import APIClient
  from betfairlightweight.endpoints.keepalive import KeepAlive, APIError
  from betfairlightweight.exceptions import KeepAliveError
  from betfairlightweight.resources.authresources import KeepAliveResource
  from tests import mock
  from tests.unit.tools import create_mock_json
  
  
  class KeepAliveTest(unittest.TestCase):
  
      def setUp(self):
          client = APIClient('username', 'password', 'app_key', 'UK')
          self.keep_alive = KeepAlive(client)
  
      @mock.patch('betfairlightweight.endpoints.keepalive.KeepAlive.request')
      def test_call(self, mock_response):
          mock = create_mock_json('tests/resources/keep_alive_success.json')
          mock_response.return_value = (mock.json(), 1.3)
          response = self.keep_alive()
  
          assert isinstance(response, KeepAliveResource)
          assert self.keep_alive.client.session_token == mock.json().get('token')
  
      @mock.patch('betfairlightweight.baseclient.BaseClient.keep_alive_headers')
      @mock.patch('betfairlightweight.baseclient.requests.post')
      def test_request(self, mock_post, mock_keep_alive_headers):
          mock_response = create_mock_json('tests/resources/logout_success.json')
          mock_post.return_value = mock_response
  
          url = 'https://identitysso.betfair.com/api/keepAlive'
          response = self.keep_alive.request()
  
          mock_post.assert_called_once_with(url, headers=mock_keep_alive_headers)
          assert response[0] == mock_response.json()
  
      @mock.patch('betfairlightweight.baseclient.BaseClient.keep_alive_headers')
      @mock.patch('betfairlightweight.baseclient.requests.post')
      def test_request_error(self, mock_post, mock_keep_alive_headers):
          mock_post.side_effect = ValueError()
  
          with self.assertRaises(APIError):
              self.keep_alive.request()
  
          mock_post.side_effect = ConnectionError()
  
          with self.assertRaises(APIError):
              self.keep_alive.request()
  
      def test_keep_alive_error_handler(self):
          mock_response = create_mock_json('tests/resources/keep_alive_success.json')
          assert self.keep_alive._error_handler(mock_response.json()) is None
  
          mock_response = create_mock_json('tests/resources/keep_alive_fail.json')
          with self.assertRaises(KeepAliveError):
              self.keep_alive._error_handler(mock_response.json())
  
      def test_url(self):
          assert self.keep_alive.url == 'https://identitysso.betfair.com/api/keepAlive'
  
subject: Remove cert bits from KeepAlive tests.