def test_no_overwrite(self):
task = ImportLastCountryOfUserToHiveTask(**self._get_kwargs())
with patch('edx.analytics.tasks.database_imports.HivePartitionTarget') as mock_target:
output = mock_target()
# Make MagicMock act more like a regular mock, so that flatten() does the right thing.
del output.__iter__
del output.__getitem__
output.exists = Mock(return_value=False)
self.assertFalse(task.complete())
self.assertTrue(output.exists.called)
output.exists = Mock(return_value=True)
self.assertTrue(task.complete())
self.assertTrue(output.exists.called)
def test_no_overwrite(self):
task = ImportLastCountryOfUserToHiveTask(**self._get_kwargs())
with patch('edx.analytics.tasks.database_imports.HivePartitionTarget'
) as mock_target:
output = mock_target()
# Make MagicMock act more like a regular mock, so that flatten() does the right thing.
del output.__iter__
del output.__getitem__
output.exists = Mock(return_value=False)
self.assertFalse(task.complete())
self.assertTrue(output.exists.called)
output.exists = Mock(return_value=True)
self.assertTrue(task.complete())
self.assertTrue(output.exists.called)
def test_overwrite(self):
kwargs = self._get_kwargs()
kwargs['overwrite'] = True
task = ImportLastCountryOfUserToHiveTask(**kwargs)
self.assertFalse(task.complete())
def test_overwrite(self):
kwargs = self._get_kwargs()
kwargs['overwrite'] = True
task = ImportLastCountryOfUserToHiveTask(**kwargs)
self.assertFalse(task.complete())
