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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
from __future__ import annotations
import inspect
import sys
import uuid
import pytest
from globus_sdk import ConnectorTable, GlobusConnectServerConnector
@pytest.mark.parametrize("connector_data", ConnectorTable._connectors)
def test_lookup_by_attribute(connector_data):
attrname, connector_name, _ = connector_data
connector = getattr(ConnectorTable, attrname)
assert connector.name == connector_name
@pytest.mark.parametrize("connector_data", ConnectorTable._connectors)
@pytest.mark.parametrize("as_uuid", (True, False))
def test_lookup_by_id(connector_data, as_uuid):
_, connector_name, connector_id = connector_data
if as_uuid:
connector_id = uuid.UUID(connector_id)
connector = ConnectorTable.lookup(connector_id)
assert connector.name == connector_name
@pytest.mark.parametrize("connector_data", ConnectorTable._connectors)
def test_lookup_by_name(connector_data):
_, connector_name, connector_id = connector_data
connector = ConnectorTable.lookup(connector_name)
assert connector.connector_id == connector_id
@pytest.mark.parametrize(
"lookup_name, expect_name",
(
("Google Drive", "Google Drive"),
("google drive", "Google Drive"),
("google_drive", "Google Drive"),
("google-drive", "Google Drive"),
("google-----drive", "Google Drive"),
("google-_-drive", "Google Drive"), # moody
(" google_-drIVE", "Google Drive"),
("google_-drIVE ", "Google Drive"),
(" GOOGLE DRIVE ", "Google Drive"),
),
)
def test_lookup_by_name_normalization(lookup_name, expect_name):
connector = ConnectorTable.lookup(lookup_name)
assert connector.name == expect_name
@pytest.mark.parametrize("name", [c.name for c in ConnectorTable.all_connectors()])
def test_all_connector_names_map_to_attributes(name):
connector = ConnectorTable.lookup(name)
assert connector is not None
name = name.replace(" ", "_").upper()
assert getattr(ConnectorTable, name) == connector
@pytest.mark.skipif(
sys.version_info < (3, 10), reason="inspect.get_annotations added in 3.10"
)
def test_all_connector_attributes_are_assigned():
# build a list of attribute names annotated with
# `t.ClassVar[GlobusConnectServerConnector]`
annotated_attributes = []
for attribute, annotation in inspect.get_annotations(ConnectorTable).items():
# get_annotations does not interpret string-ized annotations by default, so we
# receive the relevant values as strings, making comparison simple
if annotation != "t.ClassVar[GlobusConnectServerConnector]":
continue
annotated_attributes.append(attribute)
# confirm that we got the right number of annotated items
assert len(annotated_attributes) == len(ConnectorTable._connectors)
# now confirm that all of these are assigned values
for attribute in annotated_attributes:
instance = getattr(ConnectorTable, attribute)
assert isinstance(instance, GlobusConnectServerConnector)
def test_table_can_be_extended_with_simple_item():
# don't think too hard about the ethical implications of transporter clones
connector_name = "Star Trek Transporter"
connector_id = uuid.uuid1()
ExtendedTable = ConnectorTable.extend(
connector_name=connector_name, connector_id=connector_id
)
# we get a new subclass named 'ExtendedConnectorTable'
assert issubclass(ExtendedTable, ConnectorTable)
assert ExtendedTable.__name__ == "ExtendedConnectorTable"
# access via name, attribute, and ID all resolve to the same object
data_object = ExtendedTable.lookup(connector_id)
assert isinstance(data_object, GlobusConnectServerConnector)
assert ExtendedTable.STAR_TREK_TRANSPORTER is data_object
assert ExtendedTable.lookup(connector_name) is data_object
def test_table_extended_twice():
connector_id1 = uuid.uuid1()
connector_id2 = uuid.uuid1()
ExtendedTable1 = ConnectorTable.extend(
connector_name="Star Trek Transporter", connector_id=connector_id1
)
ExtendedTable2 = ExtendedTable1.extend(
connector_name="Battlestar Galactica FTL", connector_id=connector_id2
)
# we get new subclasses named 'ExtendedConnectorTable'
assert issubclass(ExtendedTable1, ConnectorTable)
assert ExtendedTable1.__name__ == "ExtendedConnectorTable"
assert issubclass(ExtendedTable2, ExtendedTable1)
assert ExtendedTable2.__name__ == "ExtendedConnectorTable"
# both tables get the same object for connector1
# only table2 has connector2
data_object = ExtendedTable1.lookup(connector_id1)
assert isinstance(data_object, GlobusConnectServerConnector)
assert ExtendedTable2.lookup(connector_id1) is data_object
assert ExtendedTable1.lookup(connector_id2) is None
data_object2 = ExtendedTable2.lookup(connector_id2)
assert isinstance(data_object2, GlobusConnectServerConnector)
|
