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)