"""Tests for file_validator module."""

import pytest

import file_validator


@pytest.mark.parametrize('mime_file_type, expected_result', [
    ('test_correct_type.txt', True),
    ('test_correct_type.csv', True),
    ('test_correct_type.tsv', True),
    ('Something', False),
])
def test_is_valid_file_type(mime_file_type, expected_result):
    """Test file type validation."""
    result = file_validator.is_valid_file_type(mime_file_type)
    assert result == expected_result


@pytest.mark.parametrize('file_size, expected_result', [
    (1, True),
    (111222, True),
    (444555666777888, False),
])
def test_is_valid_file_size(file_size, expected_result):
    """Test file type validation."""
    result = file_validator.is_valid_file_size(file_size)
    assert result == expected_result