server-backend/base_external_system/tests/test_external_system_adapter.py

# Copyright 2017 LasLabs Inc.
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl).

from odoo.exceptions import UserError

from .common import Common


class TestExternalSystemAdapter(Common):
    def setUp(self):
        super(TestExternalSystemAdapter, self).setUp()
        self.system = self.env.ref("base_external_system.external_system_os")
        self.record = self.env["external.system.adapter"].new(
            {"system_id": self.system.id}
        )

    def test_client_yields_client(self):
        """It should yield the client."""
        with self._mock_method("external_get_client") as magic:
            with self.record.client() as client:
                self.assertEqual(client, magic())

    def test_client_destroys_client(self):
        """It should destroy the client after use."""
        with self._mock_method("external_destroy_client") as magic:
            with self.record.client() as client:
                self.assertFalse(magic.call_count)
            magic.assert_called_once_with(client)

    def test_external_get_client_ensure_one(self):
        """It should assert singletons."""
        with self.assertRaises(ValueError):
            self.env["external.system.adapter"].external_get_client()

    def test_external_destroy_client_ensure_one(self):
        """It should assert singletons."""
        with self.assertRaises(ValueError):
            self.env["external.system.adapter"].external_destroy_client(None)

    def test_external_test_connection(self):
        """It should raise a UserError."""
        with self.assertRaises(UserError):
            self.record.external_test_connection()