def test_bad_url(self):
self.assertEqual(
req_hydrodata(self.sites, self.H_START_DT, self.H_END_DT,
'badurl.noob'), None)
self.assertEqual(
req_hydrodata(self.sites, self.H_START_DT, self.H_END_DT, ''),
None)
def test_bad_list(self):
self.assertEqual(
req_hydrodata(['x'], self.H_START_DT, self.H_END_DT, self.prefix),
None)
self.assertEqual(
req_hydrodata([123456, 100], self.H_START_DT, self.H_END_DT,
self.prefix), None)
self.assertEqual(
req_hydrodata([], self.H_START_DT, self.H_END_DT, self.prefix),
None)
def test_valid_data(self):
with requests_mock.Mocker() as m:
m.get(self.mock_url, json=self.mock_response)
self.assertEqual(
req_hydrodata(['05463500'], self.H_START_DT, self.H_END_DT,
self.prefix), self.valid_return)
def test_bad_status_code(self):
with requests_mock.Mocker() as m:
m.get(self.mock_url, status_code=404)
self.assertEqual(
req_hydrodata(['05463500'], self.H_START_DT, self.H_END_DT,
self.prefix), None)
