"""Test Setup."""

from pytest import fixture


@fixture
def s3_event_fixture():
    """S3 event fixture."""
    return {
        'Records': [
            {
                'eventVersion': '2.0',
                'eventTime': '1970-01-01T00:00:00.000Z',
                's3': {
                    'object': {
                        'key': 'testfolder/L10303_217_attac1.txt',
                        'size': 1024,
                        'eTag': '0123456789abcdef0123456789abcdef',
                    },
                    'bucket': {'name': 'sourcebucket'},
                },
                'eventSource': 'aws:s3',
            }
        ]
    }


@fixture
def attachment_obj_from_s3():
    """Attachment object fixture."""
    return {
        'file_key': 'testfolder/L10303_217_attac1.txt',
        'file_name': 'attac1.txt',
        'original_file_name': 'attac1.txt',
        'file_type': 'txt',
        'bucket_name': 'sourcebucket',
        'file_size': '1024',
        'etag': '0123456789abcdef0123456789abcdef',
        'account_type': 'L',
        'contract_id': None,
        'label_id': '10303',
        'period_ids': '217',
        'status': 'COMPLETED',
        'upload_date': '1970-01-01T00:00:00.000Z',
    }


@fixture
def attachment_month_dynamodb_item():
    """Attachment month item fixture."""
    return {
        'file_key': '/testfolder/L10303_217_attac1.txt',
        'file_name': 'attac1.txt',
        'original_file_name': 'attac1.txt',
        'file_type': 'txt',
        'bucket_name': 'sourcebucket',
        'file_size': '1024',
        'etag': '0123456789abcdef0123456789abcdef',
        'account_type': 'L',
        'label_id': '10303',
        'period_ids': '217',
        'status': 'COMPLETED',
        'upload_date': '1970-01-01T00:00:00.000Z',
        'event_timestamp': '1970-01-01T00:00:00.000Z',
        'label_type_id_period_id': 'L10303_217',
    }


@fixture
def attachment_quarter_dynamodb_item():
    """Attachment quarter item fixture."""
    return {
        'file_key': '/testfolder/L10303_217_attac1.txt',
        'file_name': 'attac1.txt(217)',
        'original_file_name': 'attac1.txt',
        'file_type': 'txt',
        'bucket_name': 'sourcebucket',
        'file_size': '1024',
        'etag': '0123456789abcdef0123456789abcdef',
        'account_type': 'L',
        'label_id': '10303',
        'period_ids': '217,218,219',
        'status': 'COMPLETED',
        'upload_date': '1970-01-01T00:00:00.000Z',
        'event_timestamp': '1970-01-01T00:00:00.000Z',
        'label_type_id_period_id': 'L10303_217_218_219',
    }


@fixture
def mock_attachment_items():
    """Mock Attachment items by one file."""
    return [
        {'file_name': 'attachment_doc.pdf', 'label_type_id_period_id': 'L10303_217'},
        {'file_name': 'attachment_doc.pdf', 'label_type_id_period_id': 'L10303_217_218_219'},
    ]