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())