Python SambaHook.get_conn примеры использования

Пример #1

    def test_get_conn(self, get_conn_mock):
        get_conn_mock.return_value = connection
        hook = SambaHook('samba_default')

        self.assertEqual(smbclient.SambaClient, type(hook.get_conn()))
        get_conn_mock.assert_called_once_with('samba_default')

Пример #2

Файл: test_samba.py Проект: ysktir/airflow-1

    def test_get_conn(self, get_conn_mock):
        get_conn_mock.return_value = connection
        hook = SambaHook('samba_default')

        assert isinstance(hook.get_conn(), smbclient.SambaClient)
        get_conn_mock.assert_called_once_with('samba_default')