Differential D5123 Diff 15932 abcbot/teamcity_test.py

Changeset View

Standalone View

abcbot/teamcity_test.py

This file was added.

Property	Old Value	New Value
File Mode	null	100755

				#!/usr/bin/env python3
				Lint: Code style violation Inline Actions '/home/jasonbcox/projects/bitcoin-abc/abcbot/teamcity_test.py' has code style errors. Lint: Code style violation: '/home/jasonbcox/projects/bitcoin-abc/abcbot/teamcity_test.py' has code style errors.
				#
				# Copyright (c) 2019 The Bitcoin ABC developers
				# Distributed under the MIT software license, see the accompanying
				# file COPYING or http://www.opensource.org/licenses/mit-license.php.

				import json
				import mock
				import requests
				import unittest

				from teamcity import TeamCity
				Lint: flake8 F401 Inline Actions 'teamcity.TeamCity' imported but unused Lint: flake8 F401: 'teamcity.TeamCity' imported but unused
				from testutil import AnyWith

				import mocks

				class TeamcityTests(unittest.TestCase):
				def setUp(self):
				self.teamcity = mocks.mockTeamcity()

				def tearDown(self):
				pass

				def test_getBuildProblems_requestFailure(self):
				self.teamcity.session.send.return_value.status_code = requests.codes.bad_request
				output = self.teamcity.getBuildProblems('1234')
				assert output == []
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/problemOccurrences?locator=build:(id:1234)&fields=problemOccurrence(id,details)',
				}))

				def test_getBuildProblems_noProblems(self):
				self.teamcity.session.send.return_value.content = json.dumps({})
				output = self.teamcity.getBuildProblems('1234')
				assert output == []
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/problemOccurrences?locator=build:(id:1234)&fields=problemOccurrence(id,details)',
				}))

				def test_getBuildProblems_hasProblems(self):
				problems = [{
				'id': 'id:2500,build:(id:12345)',
				'details': 'test-details',
				}]
				self.teamcity.session.send.return_value.content = json.dumps({
				'problemOccurrence': problems,
				})
				output = self.teamcity.getBuildProblems('1234')
				assert output[0]['id'] == problems[0]['id']
				assert output[0]['details'] == problems[0]['details']
				assert output[0]['logUrl'] == 'https://teamcity.test/viewLog.html?tab=buildLog&logTab=tree&filter=debug&expand=all&buildId=1234&guest=1#footer'
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/problemOccurrences?locator=build:(id:1234)&fields=problemOccurrence(id,details)',
				}))

				def test_getFailedTests_requestFailure(self):
				self.teamcity.session.send.return_value.status_code = requests.codes.bad_request
				output = self.teamcity.getFailedTests('1234')
				assert output == []
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/testOccurrences?locator=build:(id:1234),status:FAILURE&fields=testOccurrence(id,details,name)',
				}))

				def test_getFailedTests_noTestFailures(self):
				self.teamcity.session.send.return_value.content = json.dumps({})
				output = self.teamcity.getFailedTests('1234')
				assert output == []
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/testOccurrences?locator=build:(id:1234),status:FAILURE&fields=testOccurrence(id,details,name)',
				}))

				def test_getFailedTests_hasTestFailures(self):
				failures = [{
				'id': 'id:2500,build:(id:12345)',
				'details': 'stacktrace',
				'name': 'test name',
				}]
				self.teamcity.session.send.return_value.content = json.dumps({
				'testOccurrence': failures,
				})
				output = self.teamcity.getFailedTests('1234')
				assert output[0]['id'] == failures[0]['id']
				assert output[0]['details'] == failures[0]['details']
				assert output[0]['name'] == failures[0]['name']
				assert output[0]['logUrl'] == 'https://teamcity.test/viewLog.html?tab=buildLog&logTab=tree&filter=debug&expand=all&buildId=1234&guest=1&_focus=2500'
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/testOccurrences?locator=build:(id:1234),status:FAILURE&fields=testOccurrence(id,details,name)',
				}))

				def test_triggerBuild(self):
				output = self.teamcity.trigger_build('1234', 'branch-name', 'test-phid')
				assert output is True
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/buildQueue',
				'body': json.dumps({
				'branchName': 'branch-name',
				'buildType': {
				'id': '1234',
				},
				'properties': {
				'property': [{
				'name': 'env.harborMasterTargetPHID',
				'value': 'test-phid',
				}],
				},
				}),
				}))

				def test_triggerBuild_requestFailure(self):
				self.teamcity.session.send.return_value.status_code = requests.codes.bad_request
				output = self.teamcity.trigger_build('1234', 'branch-name', 'test-phid')
				assert output is False

				def test_getBuildChangeDetails(self):
				expectedOutput = {
				'username': 'email@bitcoinabc.org',
				'user': {
				'name': 'Author Name',
				},
				}
				self.teamcity.session.send.return_value.content = json.dumps(expectedOutput)
				output = self.teamcity.getBuildChangeDetails('1234')
				assert output == expectedOutput
				self.teamcity.session.send.assert_called_with(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/changes/1234',
				}))

				def test_getBuildChanges(self):
				self.teamcity.session.send.side_effect = [
				mocks.mockTeamcityResponse(json.dumps({
				'change': [{
				'id': '1234',
				}],
				})),
				mocks.mockTeamcityResponse(json.dumps({
				'username': 'email@bitcoinabc.org',
				'user': {
				'name': 'Author Name',
				},
				})),
				]
				output = self.teamcity.getBuildChanges('2345')
				assert output[0]['username'] == 'email@bitcoinabc.org'
				assert output[0]['user']['name'] == 'Author Name'
				calls = [mock.call(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/changes?locator=build:(id:2345)&fields=change(id)',
				})), mock.call(AnyWith(requests.PreparedRequest, {
				'url': 'https://teamcity.test/app/rest/changes/1234',
				}))]
				self.teamcity.session.send.assert_has_calls(calls, any_order=False)


				if __name__ == '__main__':
				unittest.main()