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")