# Copyright 2016 LasLabs Inc.

import mock

from odoo.tests import common

ADAPTER = (
    "odoo.addons.base_external_dbsource_mssql.models" ".base_external_dbsource.pymssql"
)


class TestBaseExternalDbsource(common.TransactionCase):
    def setUp(self):
        super().setUp()
        self.dbsource = self.env.ref("base_external_dbsource_mssql.demo_mssql")

    def test_connection_close_mssql(self):
        """It should close the connection"""
        connection = mock.MagicMock()
        res = self.dbsource.connection_close_mssql(connection)
        self.assertEqual(res, connection.close())

    def test_connection_open_mssql(self):
        """It should call SQLAlchemy open"""
        with mock.patch.object(
            type(self.dbsource), "_connection_open_sqlalchemy"
        ) as parent_method:
            self.dbsource.connection_open_mssql()
            parent_method.assert_called_once_with()

    def test_excecute_mssql(self):
        """It should pass args to SQLAlchemy execute"""
        expect = "sqlquery", "sqlparams", "metadata"
        with mock.patch.object(
            type(self.dbsource), "_execute_sqlalchemy"
        ) as parent_method:
            self.dbsource.execute_mssql(*expect)
            parent_method.assert_called_once_with(*expect)