def test_ingest_nightly_with_params(self): workflow_name = 'GC-fast_workflow' workflow_parameters = 'trimHold:true;' repo = Repository('/tmp/repo_night') dispatcher = Dispatcher() logger = Logger(None) w = worker.Worker(dispatcher, repo, logger, self.client) mp = mediapackage.Mediapackage() repo.add(mp) w.enqueue_nightly_job_by_name( worker.INGEST_CODE, mp, { 'workflow': workflow_name, 'workflow_parameters': workflow_parameters }) dispatcher.emit('timer-nightly') time.sleep(1) # Need time to create zip self.assertEqual(len(self.client.calls), 1) self.assertEqual(self.client.calls[0]['workflow'], workflow_name) self.assertEqual(self.client.calls[0]['workflow_parameters'], workflow_parameters) self.assertEqual(mp.getOpStatus(worker.INGEST_CODE), mediapackage.OP_DONE) rmtree('/tmp/repo_night')
def test_connect_and_emit(self): dispatcher = Dispatcher() dispatcher.connect('pr', self.callback) obj = {'called': False} dispatcher.emit('pr', obj) self.assertTrue(obj['called'])
def test_add_new_signal(self): dispatcher = Dispatcher() dispatcher.add_new_signal('test-signal', True) dispatcher.connect('test-signal', self.callback) obj = {'called': False} dispatcher.emit('test-signal', obj) self.assertTrue(obj['called'])
def test_ingest_nightly(self): repo = Repository('/tmp/repo_night') dispatcher = Dispatcher() logger = Logger(None) w = worker.Worker(dispatcher, repo, logger, self.client) mp = mediapackage.Mediapackage() mp.setOpStatus(worker.INGEST_CODE, mediapackage.OP_NIGHTLY) repo.add(mp) dispatcher.emit('timer-nightly') time.sleep(1) # Need time to create zip self.assertEqual(len(self.client.calls), 1) self.assertEqual(mp.getOpStatus(worker.INGEST_CODE), mediapackage.OP_DONE) rmtree('/tmp/repo_night')
def test_exec_nightly(self): repo = Repository('/tmp/repo_night') client = self.MHHTTPClientMock() dispatcher = Dispatcher() w = worker.Worker(dispatcher, repo, client) mp = mediapackage.Mediapackage() mp.setOpStatus(worker.INGEST_CODE, mediapackage.OP_NIGHTLY) repo.add(mp) dispatcher.emit('galicaster-notify-nightly') time.sleep(1) # Need time to create zip self.assertEqual(len(client.calls), 1) self.assertEqual(mp.getOpStatus(worker.INGEST_CODE), mediapackage.OP_DONE) rmtree('/tmp/repo_night')
def test_ingest_nightly_with_params(self): workflow_name = 'GC-fast_workflow' workflow_parameters = 'trimHold:true;' repo = Repository('/tmp/repo_night') dispatcher = Dispatcher() logger = Logger(None) w = worker.Worker(dispatcher, repo, logger, self.client) mp = mediapackage.Mediapackage() repo.add(mp) w.enqueue_nightly_job_by_name(worker.INGEST_CODE, mp, {'workflow': workflow_name, 'workflow_parameters' : workflow_parameters}) dispatcher.emit('timer-nightly') time.sleep(1) # Need time to create zip self.assertEqual(len(self.client.calls), 1) self.assertEqual(self.client.calls[0]['workflow'], workflow_name) self.assertEqual(self.client.calls[0]['workflow_parameters'], workflow_parameters) self.assertEqual(mp.getOpStatus(worker.INGEST_CODE), mediapackage.OP_DONE) rmtree('/tmp/repo_night')