File: test_run_flow.py

package info (click to toggle)
python-globus-sdk 4.3.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,172 kB
  • sloc: python: 35,227; sh: 44; makefile: 35
file content (81 lines) | stat: -rw-r--r-- 2,740 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
from __future__ import annotations

import json

import pytest

import globus_sdk
from globus_sdk import FlowsAPIError, SpecificFlowClient
from globus_sdk.testing import get_last_request, load_response


def test_run_flow(specific_flow_client_class: type[SpecificFlowClient]):
    metadata = load_response(SpecificFlowClient.run_flow).metadata

    flow_client = specific_flow_client_class(flow_id=metadata["flow_id"])

    resp = flow_client.run_flow(**metadata["request_params"])
    assert resp.http_status == 200


def test_run_flow_missing_scope(specific_flow_client_class: type[SpecificFlowClient]):
    metadata = load_response(
        SpecificFlowClient.run_flow, case="missing_scope_error"
    ).metadata

    flow_client = specific_flow_client_class(flow_id=metadata["flow_id"])

    with pytest.raises(FlowsAPIError) as excinfo:
        flow_client.run_flow(**metadata["request_params"])

    err = excinfo.value
    assert err.http_status == 403
    assert err.code == "MISSING_SCOPE"
    assert (
        err.message == "This action requires the following scope: frobulate[demuddle]"
    )


def test_run_flow_without_activity_notification_policy(
    specific_flow_client_class,
):
    metadata = load_response(SpecificFlowClient.run_flow).metadata
    flow_client = specific_flow_client_class(flow_id=metadata["flow_id"])
    resp = flow_client.run_flow({})
    assert resp.http_status == 200

    last_req = get_last_request()
    sent_payload = json.loads(last_req.body)
    assert "activity_notification_policy" not in sent_payload


def test_run_flow_with_empty_activity_notification_policy(
    specific_flow_client_class,
):
    metadata = load_response(SpecificFlowClient.run_flow).metadata
    flow_client = specific_flow_client_class(flow_id=metadata["flow_id"])

    policy = globus_sdk.RunActivityNotificationPolicy()
    flow_client.run_flow({}, activity_notification_policy=policy)

    last_req = get_last_request()
    sent_payload = json.loads(last_req.body)
    assert "activity_notification_policy" in sent_payload
    assert sent_payload["activity_notification_policy"] == {}


def test_run_flow_with_activity_notification_policy(
    specific_flow_client_class,
):
    metadata = load_response(SpecificFlowClient.run_flow).metadata
    flow_client = specific_flow_client_class(flow_id=metadata["flow_id"])

    policy = globus_sdk.RunActivityNotificationPolicy(status=["FAILED", "INACTIVE"])
    flow_client.run_flow({}, activity_notification_policy=policy)

    last_req = get_last_request()
    sent_payload = json.loads(last_req.body)
    assert "activity_notification_policy" in sent_payload
    assert sent_payload["activity_notification_policy"] == {
        "status": ["FAILED", "INACTIVE"]
    }