import pytest

from ooze import exceptions
from ooze import sanitation

def test_string():

    sanitizer = sanitation.STRING

    assert sanitizer.sanitize('The quick brown fox jumps over the lazy dog') == 'The quick brown fox jumps over the lazy dog'

def test_integer():

    sanitizer = sanitation.INTEGER

    assert sanitizer.sanitize('23') == 23

    with pytest.raises(exceptions.SanitationException):
        sanitizer.sanitize('Fnord')

    with pytest.raises(exceptions.SanitationException):
        sanitizer.sanitize([])

def test_bool():

    sanitizer = sanitation.BOOL

    assert sanitizer.sanitize('true') == True
    assert sanitizer.sanitize('True') == True
    assert sanitizer.sanitize('TRUE') == True
    assert sanitizer.sanitize('yes') == True
    assert sanitizer.sanitize('Yes') == True
    assert sanitizer.sanitize('YES') == True
    assert sanitizer.sanitize('false') == False
    assert sanitizer.sanitize('False') == False
    assert sanitizer.sanitize('FALSE') == False
    assert sanitizer.sanitize('no') == False
    assert sanitizer.sanitize('No') == False
    assert sanitizer.sanitize('NO') == False
    assert sanitizer.sanitize('1') == True
    assert sanitizer.sanitize('0') == False

    with pytest.raises(exceptions.SanitationException):
        sanitizer.sanitize('Fnord')

    with pytest.raises(exceptions.SanitationException):
        sanitizer.sanitize(23)

    with pytest.raises(exceptions.SanitationException):
        sanitizer.sanitize([])