Differential D7994 Diff 24810 contrib/buildbot/test/test_persist_database.py

Changeset View

Standalone View

contrib/buildbot/test/test_persist_database.py

This file was added.

Property	Old Value	New Value
File Mode	null	100755

				#!/usr/bin/env python3
				#
				# Copyright (c) 2020 The Bitcoin 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 os
				import server
				import shelve
				import unittest

				from build import BuildStatus
				from teamcity_wrapper import BuildInfo
				from test.abcbot_fixture import ABCBotFixture
				import test.mocks.teamcity
				from test.mocks.teamcity import DEFAULT_BUILD_ID
				from test.test_endpoint_build import buildRequestQuery
				from test.test_endpoint_status import statusRequestData


				DB_FILE_NO_EXT = "test_database"
				DB_FILE = DB_FILE_NO_EXT + '.db'

				BUILD_NAME = 'build-name'
				BUILD_TYPE_ID = 'build-type-id'
				BUILD_TARGET_PHID = 'build-target-PHID'


				class PersistDataTestCase(ABCBotFixture):
				def setUp(self):
				self.db_file_no_ext = DB_FILE_NO_EXT
				super().setUp()

				self.phab.get_file_content_from_master = mock.Mock()
				self.phab.get_file_content_from_master.return_value = json.dumps({})

				self.phab.set_text_panel_content = mock.Mock()

				self.teamcity.get_coverage_summary = mock.Mock()
				self.teamcity.get_coverage_summary.return_value = None

				self.teamcity.getBuildInfo = mock.Mock()
				self.teamcity.getBuildInfo.return_value = BuildInfo.fromSingleBuildResponse(
				json.loads(test.mocks.teamcity.buildInfo().content)
				)

				self.travis.get_branch_status = mock.Mock()
				self.travis.get_branch_status.return_value = BuildStatus.Success

				def tearDown(self):
				if os.path.exists(DB_FILE):
				os.remove(DB_FILE)

				def test_persist_diff_targets(self):
				queryData = buildRequestQuery()
				queryData.abcBuildName = BUILD_NAME
				queryData.buildTypeId = BUILD_TYPE_ID
				queryData.PHID = BUILD_TARGET_PHID

				triggerBuildResponse = test.mocks.teamcity.buildInfo(
				test.mocks.teamcity.buildInfo_changes(
				['test-change']), buildqueue=True)
				self.teamcity.session.send.return_value = triggerBuildResponse
				response = self.app.post(
				'/build{}'.format(queryData),
				headers=self.headers)
				assert response.status_code == 200
				FabienUnsubmitted Not Done Inline Actions Use `self.assertEqual()`, same for the whole test file Fabien: Use `self.assertEqual()`, same for the whole test file

				# Check the diff target state was persisted
				assert os.path.exists(DB_FILE)
				with shelve.open(DB_FILE_NO_EXT, flag='r') as db:
				assert 'diff_targets' in db
				assert BUILD_TARGET_PHID in db['diff_targets']
				assert DEFAULT_BUILD_ID in db['diff_targets'][BUILD_TARGET_PHID].builds
				assert db['diff_targets'][BUILD_TARGET_PHID].builds[DEFAULT_BUILD_ID].build_id == DEFAULT_BUILD_ID
				assert db['diff_targets'][BUILD_TARGET_PHID].builds[DEFAULT_BUILD_ID].status == BuildStatus.Queued
				assert db['diff_targets'][BUILD_TARGET_PHID].builds[DEFAULT_BUILD_ID].name == BUILD_NAME

				# Restart the server, which we expect to restore the persisted state
				del self.app
				self.app = server.create_server(
				self.teamcity,
				self.phab,
				self.slackbot,
				self.travis,
				db_file_no_ext=DB_FILE_NO_EXT,
				jsonEncoder=test.mocks.fixture.MockJSONEncoder).test_client()

				data = statusRequestData()
				data.buildName = BUILD_NAME
				data.buildId = DEFAULT_BUILD_ID
				data.buildTypeId = BUILD_TYPE_ID
				data.buildTargetPHID = BUILD_TARGET_PHID
				statusResponse = self.app.post(
				'/status', headers=self.headers, json=data)
				assert statusResponse.status_code == 200

				self.phab.harbormaster.createartifact.assert_called_with(
				buildTargetPHID=BUILD_TARGET_PHID,
				artifactKey="{}-{}".format(BUILD_NAME, BUILD_TARGET_PHID),
				artifactType="uri",
				artifactData={
				"uri": self.teamcity.build_url(
				"viewLog.html",
				{
				"buildTypeId": BUILD_TYPE_ID,
				"buildId": DEFAULT_BUILD_ID,
				},
				),
				"name": BUILD_NAME,
				"ui.external": True,
				},
				)

				# Check the diff target was cleared from persisted state
				with shelve.open(DB_FILE_NO_EXT, flag='r') as db:
				assert BUILD_TARGET_PHID not in db['diff_targets']


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