"""Shared test mocks."""

import json

from kafka_utils.consumer.source.msk import MSKMessage
import pytest


def load_file_json(filepath):
    """Load json from file."""
    with open(filepath) as event_file:
        read_data = event_file.read()
    return json.loads(read_data)


def get_single_event(info):
    """Get individual event."""
    events = info.get('records').get('stream.asset.mezzanine.audio.flac-1')
    return MSKMessage(events[0])


@pytest.fixture
def mock_event():
    """Mock event."""
    return load_file_json('tests/sample_event.json')


@pytest.fixture()
def mock_msk_event(mock_event):
    """Mock msk event."""
    return get_single_event(mock_event)