def test_context_event(self):
db_event = Event.get_or_create(db.session, "msg1", "current_event",
ErrataAdvisoryRPMsSignedEvent)
db.session.commit()
handler = MyHandler()
handler.set_context(db_event)
self.assertEqual(handler.current_db_event_id, db_event.id)
self.assertEqual(handler.current_db_artifact_build_id, None)
def test_context_artifact_build(self):
db_event = Event.get_or_create(db.session, "msg1", "current_event",
ErrataAdvisoryRPMsSignedEvent)
build = ArtifactBuild.create(db.session, db_event, "parent1-1-4",
"image")
db.session.commit()
handler = MyHandler()
handler.set_context(build)
self.assertEqual(handler.current_db_event_id, db_event.id)
self.assertEqual(handler.current_db_artifact_build_id, build.id)
def setUp(self):
super(TestCheckUnfinishedKojiTasks, self).setUp()
self.koji_read_config_patcher = patch(
'koji.read_config', return_value={'server': 'http://localhost/'})
self.koji_read_config_patcher.start()
db_event = Event.get_or_create(
db.session, "msg1", "current_event", ErrataAdvisoryRPMsSignedEvent)
db_event.state = EventState.BUILDING
self.build = ArtifactBuild.create(db.session, db_event, "parent1-1-4",
"image")
self.build.state = ArtifactBuildState.BUILD
self.build.build_id = 10
db.session.commit()
def test_start_to_build_images(self, build_artifact):
build_artifact.side_effect = [HTTPError('500 Server Error'), 1]
db_event = Event.get_or_create(db.session, 'msg1', 'current_event',
ErrataAdvisoryRPMsSignedEvent)
build = ArtifactBuild.create(db.session, db_event, 'parent1-1-4',
'image')
build2 = ArtifactBuild.create(db.session, db_event, 'parent1-1-5',
'image')
db.session.commit()
handler = MyHandler()
with self.assertLogs('freshmaker', 'ERROR'):
handler.start_to_build_images([build, build2])
self.assertEqual(build.state, ArtifactBuildState.FAILED.value)
self.assertEqual(build2.state, ArtifactBuildState.BUILD.value)
self.assertEqual(len(db.session.query(ArtifactBuild).all()), 2)
def mocked_handle(cls, msg):
event = Event.get_or_create(db.session, "msg_id", "msg_id", 0)
ArtifactBuild.create(db.session, event, "foo", 0)
db.session.commit()
cls.set_context(event)
raise ValueError("Expected exception")
