def test_blueprint_resource(self): test_script_path = 'my_script.py' def mock_download_resource(script_path): self.assertEqual(script_path, test_script_path) return script_path result = tasks.download_resource(mock_download_resource, test_script_path) self.assertEqual(result, test_script_path)
def test_blueprint_resource(self): test_script_path = 'my_script.py' def mock_download_resource(script_path, target_path): self.assertEqual(script_path, test_script_path) return target_path result = tasks.download_resource(mock_download_resource, test_script_path) self.assertTrue(result.endswith(test_script_path))
def _test_url(self, url): script_path = url original_requests_get = requests.get try: requests.get = self._mock_requests_get result = tasks.download_resource(None, script_path) with open(result) as f: self.assertEqual(script_path, f.read()) self.assertTrue(result.endswith('-some_script.py')) finally: requests.get = original_requests_get