class TestCrawlResourceGetRequiredArgument(unittest.TestCase): def setUp(self): self.resource = CrawlResource() self.url = 'http://localhost:1234' self.data = {'url': self.url} def test_get_argument(self): self.assertEqual( self.resource.get_required_argument(self.data, 'url'), self.url) def test_raise_error(self): exception = self.assertRaises( Error, self.resource.get_required_argument, self.data, 'key') self.assertEqual(exception.status, '400') def test_empty_argument(self): self.data['url'] = '' exception = self.assertRaises( Error, self.resource.get_required_argument, self.data, 'url') self.assertEqual(exception.status, '400')
def setUp(self): self.resource = CrawlResource() self.url = 'http://localhost:1234' self.data = {'url': self.url}
def resource(): return CrawlResource()