Fix successful Apache Druid task submissions reported as failed (apac…

…he#36813)

* Accept 202 status code response

* Fix status message

* Addresses [comment](apache#36813 (comment))

* Make tests cover actual behaviour

airflow/providers/apache/druid/hooks/druid.py

@@ -107,9 +107,10 @@ def submit_indexing_job(
         req_index = requests.post(url, data=json_index_spec, headers=self.header, auth=self.get_auth())
 
         code = req_index.status_code
-        if code != 200:
+        not_accepted = not (200 <= code < 300)
+        if not_accepted:
             self.log.error("Error submitting the Druid job to %s (%s) %s", url, code, req_index.content)
-            raise AirflowException(f"Did not get 200 when submitting the Druid job to {url}")
+            raise AirflowException(f"Did not get 200 or 202 when submitting the Druid job to {url}")
 
         req_json = req_index.json()
         # Wait until the job is completed

tests/providers/apache/druid/hooks/test_druid.py

@@ -65,6 +65,7 @@ def test_submit_gone_wrong(self, requests_mock):
     def test_submit_ok(self, requests_mock):
         task_post = requests_mock.post(
             "http://druid-overlord:8081/druid/indexer/v1/task",
+            status_code=200,
             text='{"task":"9f8a7359-77d4-4612-b0cd-cc2f6a3c28de"}',
         )
         status_check = requests_mock.get(
@@ -81,6 +82,7 @@ def test_submit_ok(self, requests_mock):
     def test_submit_sql_based_ingestion_ok(self, requests_mock):
         task_post = requests_mock.post(
             "http://druid-overlord:8081/druid/v2/sql/task",
+            status_code=202,
             text='{"taskId":"9f8a7359-77d4-4612-b0cd-cc2f6a3c28de"}',
         )
         status_check = requests_mock.get(