import unittest

from datetime import datetime as dt

import pytz

import logging

import singer.utils as u

import json

class TestFormat(unittest.TestCase):

def test_small_years(self):

self.assertEqual(u.strftime(dt(90, 1, 1, tzinfo=pytz.UTC)),

"0090-01-01T00:00:00.000000Z")

def test_round_trip(self):

now = dt.utcnow().replace(tzinfo=pytz.UTC)

dtime = u.strftime(now)

pdtime = u.strptime_to_utc(dtime)

fdtime = u.strftime(pdtime)

self.assertEqual(dtime, fdtime)

class TestHandleException(unittest.TestCase):

def setUp(self):

self.logger = logging.getLogger(__name__)

def test_successful_fn(self):

@u.handle_top_exception(self.logger)

def foo():

return 3

self.assertEqual(foo(), 3)

def test_exception_fn(self):

@u.handle_top_exception(self.logger)

def foo():

raise RuntimeError("foo")

self.assertRaises(RuntimeError, foo)

class TestUpdateConfig(unittest.TestCase):

def test_file_changed(self):

original_config = {'key_a': 'val_a'}

with open('config.json', 'w') as file:

file.write(json.dumps(original_config))

changed_config = {'key_b': 'val_b'}

u.update_config_file('config.json', changed_config)

read_config = u.load_json('config.json')

self.assertEqual(changed_config.get('key_b'), read_config.get('key_b'))

def tearDown(self):

import os

os.remove('config.json')