def setup_dag(self, configured_app):
with DAG(self.dag_id, start_date=datetime(2020, 6, 15), doc_md="details") as dag:
DummyOperator(task_id=self.task_id)
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
configured_app.dag_bag = dag_bag # type:ignore
def setUpClass(cls) -> None:
super().setUpClass()
with conf_vars({
("api", "auth_backend"):
"tests.test_utils.remote_user_api_auth_backend"
}):
cls.app = app.create_app(testing=True) # type:ignore
create_user(
cls.app, # type: ignore
username="******",
role_name="Test",
permissions=[('can_read', 'Dag'), ('can_read', 'DagRun'),
('can_read', 'Task')],
)
create_user(cls.app,
username="******",
role_name="TestNoPermissions") # type: ignore
with DAG(cls.dag_id,
start_date=datetime(2020, 6, 15),
doc_md="details") as dag:
DummyOperator(task_id=cls.task_id)
cls.dag = dag # type:ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
cls.app.dag_bag = dag_bag # type:ignore
def setUpClass(cls) -> None:
super().setUpClass()
with conf_vars({("api", "auth_backend"): "tests.test_utils.remote_user_api_auth_backend"}):
cls.app = app.create_app(testing=True) # type:ignore
create_user(
cls.app, # type: ignore
username="******",
role_name="Test",
permissions=[
(permissions.ACTION_CAN_CREATE, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_READ, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_EDIT, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_DELETE, permissions.RESOURCE_DAG),
],
)
create_user(cls.app, username="******", role_name="TestNoPermissions") # type: ignore
create_user(
cls.app, username="******", role_name="TestGranularDag" # type: ignore
)
cls.app.appbuilder.sm.sync_perm_for_dag( # type: ignore # pylint: disable=no-member
"TEST_DAG_1",
access_control={'TestGranularDag': [permissions.ACTION_CAN_EDIT, permissions.ACTION_CAN_READ]},
)
with DAG(cls.dag_id, start_date=datetime(2020, 6, 15), doc_md="details") as dag:
DummyOperator(task_id=cls.task_id)
with DAG(cls.dag2_id, start_date=datetime(2020, 6, 15)) as dag2: # no doc_md
DummyOperator(task_id=cls.task_id)
cls.dag = dag # type:ignore
cls.dag2 = dag2 # type: ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag, dag2.dag_id: dag2}
cls.app.dag_bag = dag_bag # type:ignore
def __init__(self, configured_app):
app = configured_app
app.appbuilder.sm.sync_perm_for_dag( # type: ignore # pylint: disable=no-member
"TEST_DAG_1",
access_control={'TestGranularDag': [permissions.ACTION_CAN_EDIT, permissions.ACTION_CAN_READ]},
)
with DAG(
self.dag_id,
start_date=datetime(2020, 6, 15),
doc_md="details",
params={"foo": 1},
tags=['example'],
) as dag:
DummyOperator(task_id=self.task_id)
with DAG(self.dag2_id, start_date=datetime(2020, 6,
15)) as dag2: # no doc_md
DummyOperator(task_id=self.task_id)
with DAG(self.dag3_id) as dag3: # DAG start_date set to None
DummyOperator(task_id=self.task_id,
start_date=datetime(2019, 6, 12))
self.dag = dag # type:ignore
self.dag2 = dag2 # type: ignore
self.dag3 = dag3 # tupe: ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag, dag2.dag_id: dag2, dag3.dag_id: dag3}
configured_app.dag_bag = dag_bag # type:ignore
def setUpClass(cls) -> None:
super().setUpClass()
with conf_vars({("api", "auth_backend"): "tests.test_utils.remote_user_api_auth_backend"}):
cls.app = app.create_app(testing=True) # type:ignore
create_user(
cls.app, # type: ignore
username="******",
role_name="Test",
permissions=[
("can_create", "Dag"),
("can_read", "Dag"),
("can_edit", "Dag"),
("can_delete", "Dag"),
],
)
create_user(cls.app, username="******", role_name="TestNoPermissions") # type: ignore
create_user(
cls.app, username="******", role_name="TestGranularDag" # type: ignore
)
cls.app.appbuilder.sm.sync_perm_for_dag( # type: ignore # pylint: disable=no-member
"TEST_DAG_1", access_control={'TestGranularDag': ['can_edit', 'can_read']}
)
with DAG(cls.dag_id, start_date=datetime(2020, 6, 15), doc_md="details") as dag:
DummyOperator(task_id=cls.task_id)
cls.dag = dag # type:ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
cls.app.dag_bag = dag_bag # type:ignore
def configured_app(minimal_app_for_api):
app = minimal_app_for_api
create_user(
app, # type: ignore
username="******",
role_name="Test",
permissions=[
(permissions.ACTION_CAN_CREATE, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_READ, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_EDIT, permissions.RESOURCE_DAG),
(permissions.ACTION_CAN_DELETE, permissions.RESOURCE_DAG),
],
)
create_user(app,
username="******",
role_name="TestNoPermissions") # type: ignore
create_user(app,
username="******",
role_name="TestGranularDag") # type: ignore
app.appbuilder.sm.sync_perm_for_dag( # type: ignore # pylint: disable=no-member
"TEST_DAG_1",
access_control={'TestGranularDag': [permissions.ACTION_CAN_EDIT, permissions.ACTION_CAN_READ]},
)
app.appbuilder.sm.sync_perm_for_dag( # type: ignore # pylint: disable=no-member
"TEST_DAG_1",
access_control={'TestGranularDag': [permissions.ACTION_CAN_EDIT, permissions.ACTION_CAN_READ]},
)
with DAG(
DAG_ID,
start_date=datetime(2020, 6, 15),
doc_md="details",
params={"foo": 1},
tags=['example'],
) as dag:
DummyOperator(task_id=TASK_ID)
with DAG(DAG2_ID, start_date=datetime(2020, 6, 15)) as dag2: # no doc_md
DummyOperator(task_id=TASK_ID)
with DAG(DAG3_ID) as dag3: # DAG start_date set to None
DummyOperator(task_id=TASK_ID, start_date=datetime(2019, 6, 12))
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag, dag2.dag_id: dag2, dag3.dag_id: dag3}
app.dag_bag = dag_bag
yield app
delete_user(app, username="******") # type: ignore
delete_user(app, username="******") # type: ignore
delete_user(app, username="******") # type: ignore
def setUpClass(cls) -> None:
super().setUpClass()
cls.app = app.create_app(testing=True) # type:ignore
with DAG(cls.dag_id, start_date=datetime(2020, 6, 15), doc_md="details") as dag:
DummyOperator(task_id=cls.task_id)
cls.dag = dag # type:ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
cls.app.dag_bag = dag_bag # type:ignore
def setup_dag(self, configured_app):
with DAG(self.dag_id,
start_date=self.task1_start_date,
doc_md="details") as dag:
task1 = DummyOperator(task_id=self.task_id)
task2 = DummyOperator(task_id=self.task_id2,
start_date=self.task2_start_date)
task1 >> task2
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
configured_app.dag_bag = dag_bag # type:ignore
def setUpClass(cls) -> None:
super().setUpClass()
with conf_vars({("api", "auth_backend"): "tests.test_utils.remote_user_api_auth_backend"}):
cls.app = app.create_app(testing=True) # type:ignore
# TODO: Add new role for each view to test permission.
create_user(cls.app, username="******", role="Admin") # type: ignore
with DAG(cls.dag_id, start_date=datetime(2020, 6, 15), doc_md="details") as dag:
DummyOperator(task_id=cls.task_id)
cls.dag = dag # type:ignore
dag_bag = DagBag(os.devnull, include_examples=False)
dag_bag.dags = {dag.dag_id: dag}
cls.app.dag_bag = dag_bag # type:ignore
