cornflow 2.0.0a11__py3-none-any.whl → 2.0.0a13__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (52) hide show
  1. airflow_config/airflow_local_settings.py +1 -1
  2. cornflow/app.py +8 -3
  3. cornflow/cli/migrations.py +23 -3
  4. cornflow/cli/service.py +18 -18
  5. cornflow/cli/utils.py +16 -1
  6. cornflow/commands/dag.py +1 -1
  7. cornflow/config.py +13 -8
  8. cornflow/endpoints/__init__.py +8 -2
  9. cornflow/endpoints/alarms.py +66 -2
  10. cornflow/endpoints/data_check.py +53 -26
  11. cornflow/endpoints/execution.py +387 -132
  12. cornflow/endpoints/login.py +81 -63
  13. cornflow/endpoints/meta_resource.py +11 -3
  14. cornflow/migrations/versions/999b98e24225.py +34 -0
  15. cornflow/models/base_data_model.py +4 -32
  16. cornflow/models/execution.py +2 -3
  17. cornflow/models/meta_models.py +28 -22
  18. cornflow/models/user.py +7 -10
  19. cornflow/schemas/alarms.py +8 -0
  20. cornflow/schemas/execution.py +2 -1
  21. cornflow/schemas/query.py +2 -1
  22. cornflow/schemas/user.py +5 -20
  23. cornflow/shared/authentication/auth.py +201 -264
  24. cornflow/shared/const.py +3 -14
  25. cornflow/shared/databricks.py +5 -1
  26. cornflow/tests/const.py +2 -0
  27. cornflow/tests/custom_test_case.py +77 -26
  28. cornflow/tests/unit/test_actions.py +2 -2
  29. cornflow/tests/unit/test_alarms.py +55 -1
  30. cornflow/tests/unit/test_apiview.py +108 -3
  31. cornflow/tests/unit/test_cases.py +20 -29
  32. cornflow/tests/unit/test_cli.py +6 -5
  33. cornflow/tests/unit/test_commands.py +3 -3
  34. cornflow/tests/unit/test_dags.py +5 -6
  35. cornflow/tests/unit/test_executions.py +491 -118
  36. cornflow/tests/unit/test_instances.py +14 -2
  37. cornflow/tests/unit/test_instances_file.py +1 -1
  38. cornflow/tests/unit/test_licenses.py +1 -1
  39. cornflow/tests/unit/test_log_in.py +230 -207
  40. cornflow/tests/unit/test_permissions.py +8 -8
  41. cornflow/tests/unit/test_roles.py +48 -10
  42. cornflow/tests/unit/test_schemas.py +1 -1
  43. cornflow/tests/unit/test_tables.py +7 -7
  44. cornflow/tests/unit/test_token.py +19 -5
  45. cornflow/tests/unit/test_users.py +22 -6
  46. cornflow/tests/unit/tools.py +75 -10
  47. {cornflow-2.0.0a11.dist-info → cornflow-2.0.0a13.dist-info}/METADATA +16 -15
  48. {cornflow-2.0.0a11.dist-info → cornflow-2.0.0a13.dist-info}/RECORD +51 -51
  49. {cornflow-2.0.0a11.dist-info → cornflow-2.0.0a13.dist-info}/WHEEL +1 -1
  50. cornflow/endpoints/execution_databricks.py +0 -808
  51. {cornflow-2.0.0a11.dist-info → cornflow-2.0.0a13.dist-info}/entry_points.txt +0 -0
  52. {cornflow-2.0.0a11.dist-info → cornflow-2.0.0a13.dist-info}/top_level.txt +0 -0
cornflow/tests/unit/test_dags.py CHANGED
@@ -234,7 +234,6 @@ class TestDagDetailEndpoint(TestExecutionsDetailEndpointMock):
234
234
  )
235
235
  self.assertEqual(data["data"], instance_data["data"])
236
236
  self.assertEqual(data["config"], self.payload["config"])
237
- return
238
237
 
239
238
  def test_get_no_dag(self):
240
239
  """
@@ -372,7 +371,7 @@ class TestDeployedDAG(TestCase):
372
371
  follow_redirects=True,
373
372
  headers={
374
373
  "Content-Type": "application/json",
375
- "Authorization": "Bearer " + self.token,
374
+ "Authorization": f"Bearer {self.token}",
376
375
  },
377
376
  )
378
377
 
@@ -403,7 +402,7 @@ class TestDeployedDAG(TestCase):
403
402
  follow_redirects=True,
404
403
  headers={
405
404
  "Content-Type": "application/json",
406
- "Authorization": "Bearer " + self.token,
405
+ "Authorization": f"Bearer {self.token}",
407
406
  },
408
407
  )
409
408
 
@@ -428,7 +427,7 @@ class TestDeployedDAG(TestCase):
428
427
  follow_redirects=True,
429
428
  headers={
430
429
  "Content-Type": "application/json",
431
- "Authorization": "Bearer " + self.token,
430
+ "Authorization": f"Bearer {self.token}",
432
431
  },
433
432
  )
434
433
 
@@ -440,7 +439,7 @@ class TestDeployedDAG(TestCase):
440
439
  follow_redirects=True,
441
440
  headers={
442
441
  "Content-Type": "application/json",
443
- "Authorization": "Bearer " + self.token,
442
+ "Authorization": f"Bearer {self.token}",
444
443
  },
445
444
  )
446
445
 
@@ -455,7 +454,7 @@ class TestDeployedDAG(TestCase):
455
454
  follow_redirects=True,
456
455
  headers={
457
456
  "Content-Type": "application/json",
458
- "Authorization": "Bearer " + self.token,
457
+ "Authorization": f"Bearer {self.token}",
459
458
  },
460
459
  )
461
460
  self.assertEqual(response.status_code, 400)