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
|
import pytest
from influxdb_client.rest import ApiException
from tests.base_test import BaseTest
class TestLabelsApi(BaseTest):
def setUp(self) -> None:
super(TestLabelsApi, self).setUp()
self.organization = self.find_my_org()
labels = self.labels_api.find_labels()
for label in labels:
self.labels_api.delete_label(label)
def test_create_label_unicode(self):
name = self.generate_name("Cool resource 🍺")
properties = {"color": "red 🍺", "source": "remote api"}
label = self.labels_api.create_label(name, self.organization.id, properties)
self.assertIsNotNone(label)
self.assertIsNotNone(label.id)
self.assertEqual(label.name, name)
self.assertEqual(label.properties["color"], "red 🍺")
self.assertEqual(label.properties["source"], "remote api")
def test_create_label_empty_properties(self):
name = self.generate_name("Cool resource")
label = self.labels_api.create_label(name=name, org_id=self.organization.id)
self.assertIsNotNone(label)
self.assertIsNotNone(label.id)
self.assertEqual(label.name, name)
self.assertEqual(label.properties, None)
def test_update_label(self):
label = self.labels_api.create_label(self.generate_name("Cool Resource"), org_id=self.organization.id)
self.assertEqual(label.properties, None)
label.properties = {"color": "blue"}
label = self.labels_api.update_label(label)
self.assertEqual(label.properties["color"], "blue")
self.assertEqual(len(label.properties), 1)
label.properties["type"] = "free"
label = self.labels_api.update_label(label)
self.assertEqual(label.properties["color"], "blue")
self.assertEqual(label.properties["type"], "free")
self.assertEqual(len(label.properties), 2)
label.properties["type"] = "paid"
label.properties["color"] = ""
label = self.labels_api.update_label(label)
self.assertEqual(label.properties["type"], "paid")
self.assertEqual(len(label.properties), 1)
def test_delete_label(self):
created_label = self.labels_api.create_label(self.generate_name("Cool Resource"), org_id=self.organization.id)
self.assertIsNotNone(created_label)
found_label = self.labels_api.find_label_by_id(label_id=created_label.id)
self.assertIsNotNone(found_label)
self.labels_api.delete_label(created_label)
with pytest.raises(ApiException) as e:
assert self.labels_api.find_label_by_id(label_id=created_label.id)
assert "label not found" in e.value.body
def test_clone_label(self):
name = self.generate_name("cloned")
properties = {
"color": "green",
"location": "west"
}
label = self.labels_api.create_label(name, self.organization.id, properties)
cloned = self.labels_api.clone_label(cloned_name=name+"_clone", label=label)
self.assertEqual(cloned.name, name+"_clone")
self.assertEqual(cloned.properties, properties)
def test_find_label_by_id(self):
label = self.labels_api.create_label(self.generate_name("Cool Resource"), org_id=self.organization.id)
label_by_id = self.labels_api.find_label_by_id(label_id=label.id)
self.assertEqual(label, label_by_id)
def test_find_label_by_id_none(self):
with pytest.raises(ApiException) as e:
assert self.labels_api.find_label_by_id(label_id="020f755c3c082000")
assert "label not found" in e.value.body
def test_find_labels(self):
size = len(self.labels_api.find_labels())
self.labels_api.create_label(self.generate_name("Cool Resource"), org_id=self.organization.id)
labels = self.labels_api.find_labels()
self.assertEqual(len(labels), size + 1)
def test_find_label_by_org(self):
organization = self.organizations_api.create_organization(self.generate_name("org"))
labels = self.labels_api.find_label_by_org(org_id=organization.id)
self.assertEqual(len(labels), 0)
self.labels_api.create_label(self.generate_name("Cool Resource"), org_id=organization.id)
labels = self.labels_api.find_label_by_org(org_id=organization.id)
self.assertEqual(len(labels), 1)
self.organizations_api.delete_organization(organization.id)
with pytest.raises(ApiException) as e:
assert self.organizations_api.find_organization(org_id=organization.id)
assert "organization not found" in e.value.body
|
