forked from python/release-tools
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_buildbotapi.py
More file actions
140 lines (113 loc) · 3.96 KB
/
test_buildbotapi.py
File metadata and controls
140 lines (113 loc) · 3.96 KB
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
135
136
137
138
139
140
from functools import cache
from unittest.mock import AsyncMock
import aiohttp
import pytest
import buildbotapi
def test_builder_class() -> None:
# Arrange / Act
builder = buildbotapi.Builder(
builderid=123,
description="my description",
name="my name",
tags=["tag1", "tag2"],
)
# Assert
assert builder.builderid == 123
assert builder.description == "my description"
assert builder.name == "my name"
assert builder.tags == ["tag1", "tag2"]
assert hash(builder) == 123
@cache
def load(filename: str) -> str:
with open(filename) as f:
return f.read()
@pytest.mark.asyncio
async def test_buildbotapi_authenticate() -> None:
# Arrange
async with AsyncMock(aiohttp.ClientSession) as mock_session:
api = buildbotapi.BuildBotAPI(mock_session)
# Act
await api.authenticate(token="")
# Assert
mock_session.get.assert_called_with(
"https://buildbot.python.org/all/auth/login", params={"token": ""}
)
@pytest.mark.asyncio
async def test_buildbotapi_all_builders() -> None:
# Arrange
mock_session = AsyncMock(aiohttp.ClientSession)
mock_session.get.return_value.__aenter__.return_value.status = 200
mock_session.get.return_value.__aenter__.return_value.text.return_value = load(
"tests/buildbotapi/builders.json"
)
api = buildbotapi.BuildBotAPI(mock_session)
# Act
all_builders = await api.all_builders()
# Assert
mock_session.get.assert_called_with(
"https://buildbot.python.org/all/api/v2/builders"
)
assert len(all_builders) == 2
assert all_builders[3].name == "AMD64 RHEL8 LTO 3.13"
assert all_builders[1623].name == "AMD64 Windows PGO NoGIL PR"
@pytest.mark.asyncio
async def test_buildbotapi_all_builders_with_branch() -> None:
# Arrange
mock_session = AsyncMock(aiohttp.ClientSession)
mock_session.get.return_value.__aenter__.return_value.status = 200
mock_session.get.return_value.__aenter__.return_value.text.return_value = load(
"tests/buildbotapi/builders.json"
)
api = buildbotapi.BuildBotAPI(mock_session)
# Act
await api.all_builders(branch="3.13")
# Assert
mock_session.get.assert_called_with(
"https://buildbot.python.org/all/api/v2/builders?tags__contains=3.13"
)
@pytest.mark.asyncio
async def test_buildbotapi_stable_builders() -> None:
# Arrange
mock_session = AsyncMock(aiohttp.ClientSession)
mock_session.get.return_value.__aenter__.return_value.status = 200
mock_session.get.return_value.__aenter__.return_value.text.return_value = load(
"tests/buildbotapi/builders.json"
)
api = buildbotapi.BuildBotAPI(mock_session)
# Act
all_builders = await api.stable_builders()
# Assert
mock_session.get.assert_called_with(
"https://buildbot.python.org/all/api/v2/builders"
)
assert len(all_builders) == 1
assert all_builders[3].name == "AMD64 RHEL8 LTO 3.13"
assert "stable" in all_builders[3].tags
@pytest.mark.asyncio
@pytest.mark.parametrize(
["json_data", "expected"],
[
("tests/buildbotapi/success.json", False),
("tests/buildbotapi/failure.json", True),
("tests/buildbotapi/no-builds.json", False),
],
)
async def test_buildbotapi_is_builder_failing_currently_yes(
json_data: str, expected: bool
) -> None:
# Arrange
mock_session = AsyncMock(aiohttp.ClientSession)
mock_session.get.return_value.__aenter__.return_value.status = 200
mock_session.get.return_value.__aenter__.return_value.text.return_value = load(
json_data
)
api = buildbotapi.BuildBotAPI(mock_session)
builder = buildbotapi.Builder(builderid=3)
# Act
failing = await api.is_builder_failing_currently(builder=builder)
# Assert
mock_session.get.assert_called_with(
"https://buildbot.python.org/all/api/v2/builds?complete__eq=true"
"&&builderid__eq=3&&order=-complete_at&&limit=1"
)
assert failing is expected