def setUp(self):
self.cmd = LoadCommand()
self.name = 'ahoy'
self.url = 'https://south.amazonaws.com/jarbas/1970-01-01-ahoy.xz'
self.custom_settings = {
'AMAZON_S3_DATASET_DATE': '1970-01-01',
'AMAZON_S3_REGION': 'south',
'AMAZON_S3_BUCKET': 'jarbas'
}
class TestStaticMethods(TestCase):
def setUp(self):
self.cmd = LoadCommand()
def test_get_file_name(self):
expected = '1970-01-01-ahoy.xz'
with self.settings(AMAZON_S3_DATASET_DATE='1970-01-01'):
self.assertEqual(expected, self.cmd.get_file_name('ahoy'))
def test_get_model_name(self):
self.assertEqual('Activity', self.cmd.get_model_name(Activity))
class TestPrintCount(TestCase):
def setUp(self):
self.cmd = LoadCommand()
@patch('jarbas.core.management.commands.print')
def test_print_no_records(self, mock_print):
self.cmd.print_count(Activity)
arg = 'Current count: 0 Activitys '
kwargs = {'end': '\r'}
mock_print.assert_called_with(arg, **kwargs)
@patch('jarbas.core.management.commands.print')
def test_print_with_records(self, mock_print):
Activity.objects.create(**sample_activity_data)
self.cmd.print_count(Activity)
arg = 'Current count: 1 Activitys '
kwargs = {'end': '\r'}
mock_print.assert_called_with(arg, **kwargs)
@patch('jarbas.core.management.commands.print')
def test_print_with_permanent_keyword_arg(self, mock_print):
self.cmd.print_count(Activity, permanent=True)
arg = 'Current count: 0 Activitys '
kwargs = {'end': '\n'}
mock_print.assert_called_with(arg, **kwargs)
class TestPrintCount(TestCase):
def setUp(self):
self.cmd = LoadCommand()
@patch('jarbas.core.management.commands.print')
def test_print_no_records(self, mock_print):
self.cmd.print_count(Activity)
arg = 'Current count: 0 Activitys '
kwargs = {'end': '\r'}
mock_print.assert_called_with(arg, **kwargs)
@patch('jarbas.core.management.commands.print')
def test_print_with_records(self, mock_print):
Activity.objects.create(**sample_activity_data)
self.cmd.print_count(Activity)
arg = 'Current count: 1 Activitys '
kwargs = {'end': '\r'}
mock_print.assert_called_with(arg, **kwargs)
@patch('jarbas.core.management.commands.print')
def test_print_with_permanent_keyword_arg(self, mock_print):
self.cmd.print_count(Activity, permanent=True)
arg = 'Current count: 0 Activitys '
kwargs = {'end': '\n'}
mock_print.assert_called_with(arg, **kwargs)
class TestRemoteMethods(TestCase):
def setUp(self):
self.cmd = LoadCommand()
self.name = 'ahoy'
self.url = 'https://south.amazonaws.com/jarbas/1970-01-01-ahoy.xz'
self.custom_settings = {
'AMAZON_S3_DATASET_DATE': '1970-01-01',
'AMAZON_S3_REGION': 'south',
'AMAZON_S3_BUCKET': 'jarbas'
}
def test_get_url(self):
with self.settings(**self.custom_settings):
result = self.cmd.get_url(self.name)
self.assertEqual(self.url, result)
@patch('jarbas.core.management.commands.print')
@patch('jarbas.core.management.commands.urlretrieve')
def test_load_remote(self, mock_urlretrieve, mock_print):
with self.settings(**self.custom_settings):
result = self.cmd.load_remote(self.name)
self.assertEqual(self.url, mock_urlretrieve.call_args[0][0])
self.assertIsInstance(result, str)
class TestLocalMethods(TestCase):
def setUp(self):
self.cmd = LoadCommand()
self.source = '/whatever/works'
self.name = 'ahoy'
def test_get_path(self):
expected = '/whatever/works/1970-01-01-ahoy.xz'
with self.settings(AMAZON_S3_DATASET_DATE='1970-01-01'):
result = self.cmd.get_path(self.source, self.name)
self.assertEqual(expected, result)
@patch('jarbas.core.management.commands.print')
@patch('jarbas.core.management.commands.os.path.exists')
def test_load_local_exists(self, mock_exists, mock_print):
mock_exists.return_value = True
self.assertIsInstance(self.cmd.load_local(self.source, self.name), str)
@patch('jarbas.core.management.commands.print')
@patch('jarbas.core.management.commands.os.path.exists')
def test_load_local_fail(self, mock_exists, mock_print):
mock_exists.return_value = False
self.assertFalse(self.cmd.load_local(self.source, self.name))
def test_add_arguments_without_drop_all(self):
mock = Mock()
LoadCommand().add_arguments(mock, add_drop_all=False)
self.assertEqual(1, mock.add_argument.call_count)
def test_add_arguments(self):
mock = Mock()
LoadCommand().add_arguments(mock)
self.assertEqual(2, mock.add_argument.call_count)
def test_drop_all(self, mock_print):
self.assertEqual(0, Activity.objects.count())
Activity.objects.create(**sample_activity_data)
self.assertEqual(1, Activity.objects.count())
LoadCommand().drop_all(Activity)
self.assertEqual(0, Activity.objects.count())
def setUp(self):
self.cmd = LoadCommand()
class TestStaticMethods(TestCase):
def setUp(self):
self.cmd = LoadCommand()
def test_get_model_name(self):
self.assertEqual('Activity', self.cmd.get_model_name(Activity))
def test_to_date(self):
expected = date(1991, 7, 22)
self.assertEqual(self.cmd.to_date('22/7/91'), expected)
self.assertEqual(self.cmd.to_date('1991-07-22 03:15:00+0300'),
expected)
self.assertEqual(self.cmd.to_date('22/13/91'), None)
self.assertEqual(self.cmd.to_date('aa/7/91'), None)
self.assertEqual(self.cmd.to_date('22/07/16'), date(2016, 7, 22))
def test_to_number(self):
self.assertIsNone(self.cmd.to_number(''))
self.assertIsNone(self.cmd.to_number('NaN'))
self.assertIsNone(self.cmd.to_number('nan'))
self.assertEqual(1.0, self.cmd.to_number('1'))
self.assertEqual(1.2, self.cmd.to_number('1.2'))
self.assertEqual(1, self.cmd.to_number('1', int))
self.assertEqual(1, self.cmd.to_number('1.0', int))
def setUp(self):
self.cmd = LoadCommand()
self.source = '/whatever/works'
self.name = 'ahoy'
def setUp(self):
self.cmd = LoadCommand()
class TestStaticMethods(TestCase):
def setUp(self):
self.cmd = LoadCommand()
def test_get_model_name(self):
self.assertEqual('Activity', self.cmd.get_model_name(Activity))
def test_to_date(self):
expected = date(1991, 7, 22)
self.assertEqual(self.cmd.to_date('22/7/91'), expected)
self.assertEqual(self.cmd.to_date('1991-07-22 03:15:00+0300'), expected)
self.assertEqual(self.cmd.to_date('22/13/91'), None)
self.assertEqual(self.cmd.to_date('aa/7/91'), None)
self.assertEqual(self.cmd.to_date('22/07/16'), date(2016, 7, 22))
def test_to_number(self):
self.assertIsNone(self.cmd.to_number(''))
self.assertIsNone(self.cmd.to_number('NaN'))
self.assertIsNone(self.cmd.to_number('nan'))
self.assertEqual(1.0, self.cmd.to_number('1'))
self.assertEqual(1.2, self.cmd.to_number('1.2'))
self.assertEqual(1, self.cmd.to_number('1', int))
self.assertEqual(1, self.cmd.to_number('1.0', int))