"""Test for a vendor model."""
from oto import response
import pytest

from content_review.models import review_type
from content_review.utils import db_operations


@pytest.fixture
def db_fixture():
    """Set up the vendor table."""
    db_operations.create_tables()
    db_operations.seed_review_type_table()


@pytest.fixture()
def get_all_review_types():
    """Successful response for getting all review types."""
    return response.Response([
        {
            'review_type_id': 1,
            'name': 'MENA',
            'description': 'Middle Eastern and North African territories'
        }
    ])


@pytest.fixture()
def get_all_review_types_flipped_dict():
    """Successful response for getting all review status types flipped."""
    return response.Response({
        'MENA': 1
    })


def test_get_review_types(db_fixture, get_all_review_types):
    """Test getting review types."""
    actual_response = review_type.get_review_types()
    assert get_all_review_types.message == actual_response.message


def test_get_review_types_flipped_dict(
        db_fixture, get_all_review_types_flipped_dict):
    """Test getting review types flipped."""
    actual_response = review_type.get_review_types(flipped_dict=True)
    assert get_all_review_types_flipped_dict.message == actual_response.message