import pytest
from vector_utils.connections import connection


@pytest.fixture
def sub_connection_class():
    """Create fake connection extension class."""
    class SubConnection(connection.Connection):
        def check_connection(self, remote_initial_dir):
            pass

        def close_connection(self):
            pass

        def delete(self, filename, remove_dir=False):
            pass

        def file_exists(self, file_name):
            pass

        def file_size(self, file_name):
            pass

        def is_dir(self, dirname):
            pass

        def login(self):
            pass

        def mkdir(self, dest_dir):
            pass

        def rmdir(self, dir_name):
            pass

        def reconnect(self):
            pass

        def scan_dir(self, directory, exceptions=[]):
            pass

        def transfer_files(self, file_list, transfer_mode='upload',
                           batch_file_id=None, pid=None, overwrite=None,
                           check_delivered_files=True, d_job=None):
            pass

    return SubConnection


def test_connection_base_class(sub_connection_class):
    """Test connection base class."""
    conn_object = {
        'domain_name': 'ihr-ingestion-pipeline-production-orchard'
    }
    c = sub_connection_class(conn_object)

    assert c.get_bucket() == 'ihr-ingestion-pipeline-production-orchard'
    assert c.conn_obj == conn_object