def test_run_harvester_daysback(monkeypatch):
mock_harvest = mock.MagicMock()
mock_begin_norm = mock.MagicMock()
monkeypatch.setattr('scrapi.tasks.harvest', mock_harvest)
monkeypatch.setattr('scrapi.tasks.begin_normalization', mock_begin_norm)
tasks.run_harvester('test', days_back=10)
assert mock_harvest.si.called
assert mock_begin_norm.s.called
mock_begin_norm.s.assert_called_once_with('test')
mock_harvest.si.assert_called_once_with('test', 'TIME', days_back=10)
def test_run_harvester_calls(monkeypatch):
mock_harvest = mock.MagicMock()
mock_begin_norm = mock.MagicMock()
monkeypatch.setattr("scrapi.tasks.harvest", mock_harvest)
monkeypatch.setattr("scrapi.tasks.begin_normalization", mock_begin_norm)
tasks.run_harvester("test")
assert mock_harvest.si.called
assert mock_begin_norm.s.called
end_date = date(2015, 03, 16)
start_date = end_date - timedelta(settings.DAYS_BACK)
mock_begin_norm.s.assert_called_once_with("test")
mock_harvest.si.assert_called_once_with("test", "TIME", start_date=start_date, end_date=end_date)
def test_run_harvester_calls(monkeypatch):
mock_harvest = mock.MagicMock()
mock_begin_norm = mock.MagicMock()
monkeypatch.setattr('scrapi.tasks.harvest', mock_harvest)
monkeypatch.setattr('scrapi.tasks.begin_normalization', mock_begin_norm)
tasks.run_harvester('test')
assert mock_harvest.si.called
assert mock_begin_norm.s.called
end_date = date(2015, 03, 16)
start_date = end_date - timedelta(settings.DAYS_BACK)
mock_begin_norm.s.assert_called_once_with('test')
mock_harvest.si.assert_called_once_with('test', 'TIME', start_date=start_date, end_date=end_date)
def test_run_harvester_daysback(monkeypatch):
mock_harvest = mock.MagicMock()
mock_begin_norm = mock.MagicMock()
monkeypatch.setattr("scrapi.tasks.harvest", mock_harvest)
monkeypatch.setattr("scrapi.tasks.begin_normalization", mock_begin_norm)
start_date = date(2015, 03, 14)
end_date = date(2015, 03, 16)
tasks.run_harvester("test", start_date=start_date, end_date=end_date)
assert mock_harvest.si.called
assert mock_begin_norm.s.called
mock_begin_norm.s.assert_called_once_with("test")
mock_harvest.si.assert_called_once_with("test", "TIME", start_date=start_date, end_date=end_date)
def test_run_harvester_daysback(monkeypatch):
mock_harvest = mock.MagicMock()
mock_begin_norm = mock.MagicMock()
monkeypatch.setattr('scrapi.tasks.harvest', mock_harvest)
monkeypatch.setattr('scrapi.tasks.begin_normalization', mock_begin_norm)
start_date = date(2015, 03, 14)
end_date = date(2015, 03, 16)
tasks.run_harvester('test', start_date=start_date, end_date=end_date)
assert mock_harvest.si.called
assert mock_begin_norm.s.called
mock_begin_norm.s.assert_called_once_with('test')
mock_harvest.si.assert_called_once_with('test', 'TIME', start_date=start_date, end_date=end_date)
