from datetime import datetime
from datetime import timedelta

from types import StringType
from types import DictType

from ago import human
from ago import delta2dict

PRESENT = datetime.now()
PAST    = PRESENT - timedelta( 492, 58711, 45 ) # days, secs, ms
FUTURE  = PRESENT + timedelta( 2, 12447, 963 ) # days, secs, ms

PAST_DELTA   = PRESENT - PAST
FUTURE_DELTA = PRESENT - FUTURE

def test_human_is_string():
    assert type(human( PAST )) is StringType

def test_delta2dict_is_dict():
    assert type(delta2dict( PAST_DELTA )) is DictType

def test_ago_in_past_human():
    assert 'ago' in human( PAST )

def test_in_in_future_human():
    assert 'in' in human( FUTURE )

def test_no_coma_in_one_precision():
    assert ',' not in human( PAST, precision = 1 )
    assert ',' not in human( FUTURE, precision = 1 )

def test_coma_in_three_precision():
    assert ',' in human( PAST, precision = 3 )
    assert ',' in human( FUTURE, precision = 3 )

def test_one_day_singular():
    assert 's' not in human( timedelta(1) )

def test_two_day_plural():
    assert 's' in human( timedelta(2) )

def test_past_tense():
    output = human( PAST,
      past_tense = 'titanic sunk {} ago',
      future_tense = 'titanic will sink in {} from now'
    )
    assert 'titanic sunk' in output

def test_future_tense():
    output = human( FUTURE,
      past_tense = 'titanic sunk {} ago',
      future_tense = 'titanic will sink in {} from now'
    )
    assert 'titanic will sink in' in output

def test_valid_past_dict():
    past_dict = delta2dict( PAST_DELTA )
    assert past_dict['year'] == 1
    assert past_dict['day'] == 127
    assert past_dict['hour'] == 16
    assert past_dict['minute'] == 18
    assert past_dict['microsecond'] == 45

def test_valid_future_dict():
    past_dict = delta2dict( FUTURE_DELTA )
    assert past_dict['year'] == 0
    assert past_dict['day'] == 2
    assert past_dict['hour'] == 3
    assert past_dict['minute'] == 27
    assert past_dict['microsecond'] == 963

def example_usage():
    """Test and example usage"""

    print '\nTest past tense:\n'
    print delta2dict( PAST_DELTA )
    print 'Commented ' + human( PAST_DELTA, 1 )
    print human( PAST, past_tense = "Commented {} ago" )

    print '\nTest future tense:\n'
    print delta2dict( FUTURE_DELTA )
    print 'Shutdown ' + human( FUTURE_DELTA, 5 )
    print human( FUTURE, future_tense = 'Shutdown in {} from now' )
    print ''