"""Tests for processing module."""

from unittest import mock

from flexmock import flexmock
import pytest
from transcoding.logic import processing_sample
from transcoding.utils import message_extractor


@pytest.mark.parametrize('message_body, expected_result_status', [
    ({'message_id': 1}, 200), ({}, 400)])
def test_sample_processing(message_body, expected_result_status):
    """Test for getting topic from order data."""
    # mocking
    message_mock = mock.MagicMock()
    (flexmock(message_extractor)
     .should_receive('get_message_body')
     .with_args(message_mock)
     .and_return(message_body))
    # test functional call
    result_response = processing_sample.sample_processing(message_mock)
    # checking
    assert result_response.status == expected_result_status