from datetime import date, datetime, timedelta
from decimal import Decimal
import unittest

from dateutil.tz import tzutc
import six

from yolk import utils


class TestUtils(unittest.TestCase):

    def test_timezone_utils(self):
        now = datetime.now()
        utcnow = datetime.now(tz=tzutc())
        self.assertTrue(utils.is_naive(now))
        self.assertFalse(utils.is_naive(utcnow))

        fixed = utils.guess_timezone(now)
        self.assertFalse(utils.is_naive(fixed))

        shouldnt_be_edited = utils.guess_timezone(utcnow)
        self.assertEqual(utcnow, shouldnt_be_edited)

    def test_clean(self):
        simple = {
            'decimal': Decimal('0.142857'),
            'unicode': six.u('woo'),
            'date': datetime.now(),
            'long': 200000000,
            'integer': 1,
            'float': 2.0,
            'bool': True,
            'str': 'woo',
            'none': None
        }

        complicated = {
            'exception': Exception('This should show up'),
            'timedelta': timedelta(microseconds=20),
            'list': [1, 2, 3]
        }

        combined = dict(simple.items())
        combined.update(complicated.items())

        pre_clean_keys = combined.keys()

        utils.clean(combined)
        self.assertEqual(combined.keys(), pre_clean_keys)

    def test_clean_with_dates(self):
        dict_with_dates = {
            'birthdate': date(1980, 1, 1),
            'registration': datetime.utcnow(),
        }
        self.assertEqual(dict_with_dates, utils.clean(dict_with_dates))

    @classmethod
    def test_bytes(cls):
        if six.PY3:
            item = bytes(10)
        else:
            item = bytearray(10)

        utils.clean(item)

    def test_clean_fn(self):
        cleaned = utils.clean({'fn': lambda x: x, 'number': 4})
        self.assertEqual(cleaned['number'], 4)
        if 'fn' in cleaned:
            self.assertEqual(cleaned['fn'], None)

    def test_remove_slash(self):
        self.assertEqual('http://segment.io',
                         utils.remove_trailing_slash('http://segment.io/'))
        self.assertEqual('http://segment.io',
                         utils.remove_trailing_slash('http://segment.io'))

[(14, 'datetime.datetime.now', 'datetime.now', ([], {}), False, 'from datetime import date, datetime, timedelta\n'), (19, 'yolk.utils.guess_timezone', 'utils.guess_timezone', (['now'], {}), False, 'from yolk import utils\n'), (22, 'yolk.utils.guess_timezone', 'utils.guess_timezone', (['utcnow'], {}), False, 'from yolk import utils\n'), (49, 'yolk.utils.clean', 'utils.clean', (['combined'], {}), False, 'from yolk import utils\n'), (66, 'yolk.utils.clean', 'utils.clean', (['item'], {}), False, 'from yolk import utils\n'), (69, 'yolk.utils.clean', 'utils.clean', (["{'fn': lambda x: x, 'number': 4}"], {}), False, 'from yolk import utils\n'), (16, 'yolk.utils.is_naive', 'utils.is_naive', (['now'], {}), False, 'from yolk import utils\n'), (17, 'yolk.utils.is_naive', 'utils.is_naive', (['utcnow'], {}), False, 'from yolk import utils\n'), (20, 'yolk.utils.is_naive', 'utils.is_naive', (['fixed'], {}), False, 'from yolk import utils\n'), (27, 'decimal.Decimal', 'Decimal', (['"""0.142857"""'], {}), False, 'from decimal import Decimal\n'), (28, 'six.u', 'six.u', (['"""woo"""'], {}), False, 'import six\n'), (29, 'datetime.datetime.now', 'datetime.now', ([], {}), False, 'from datetime import date, datetime, timedelta\n'), (40, 'datetime.timedelta', 'timedelta', ([], {'microseconds': '(20)'}), False, 'from datetime import date, datetime, timedelta\n'), (54, 'datetime.date', 'date', (['(1980)', '(1)', '(1)'], {}), False, 'from datetime import date, datetime, timedelta\n'), (55, 'datetime.datetime.utcnow', 'datetime.utcnow', ([], {}), False, 'from datetime import date, datetime, timedelta\n'), (57, 'yolk.utils.clean', 'utils.clean', (['dict_with_dates'], {}), False, 'from yolk import utils\n'), (76, 'yolk.utils.remove_trailing_slash', 'utils.remove_trailing_slash', (['"""http://segment.io/"""'], {}), False, 'from yolk import utils\n'), (78, 'yolk.utils.remove_trailing_slash', 'utils.remove_trailing_slash', (['"""http://segment.io"""'], {}), False, 'from yolk import utils\n'), (15, 'dateutil.tz.tzutc', 'tzutc', ([], {}), False, 'from dateutil.tz import tzutc\n')]