def setUp(self): self.portal = self.layer['portal'] self.experiments = self.portal[defaults.EXPERIMENTS_FOLDER_ID] # create and run sdm experiment sdmform = SDMExperimentHelper(self.portal).get_form() sdmform.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request sdmform.update() transaction.commit() # We should have only one SDM sdmexp = self.experiments.values()[0] self.form = EnsembleExperimentHelper(self.portal, sdmexp)
def setUp(self, mock_run_script): self.portal = self.layer['portal'] self.experiments = self.portal[defaults.EXPERIMENTS_FOLDER_ID] # create and run sdm experiment formhelper = SDMExperimentHelper(self.portal) sdmform = formhelper.get_form() sdmform.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request sdmform.update() # setup mock_run_script mock_run_script.side_effect = formhelper.mock_run_script transaction.commit() # We should have only one SDM sdmexp = self.experiments.values()[0] self.form = EnsembleExperimentHelper(self.portal, sdmexp)
class ExperimentEnsembleAddTest(unittest.TestCase): # Use functional layer for now to get a new demostorage layer for each test layer = BCCVL_FUNCTIONAL_TESTING # integration testing gives only a new transaction for each test # (rolled back at end) # layer = BCCVL_INTEGRATION_TESTING @mock.patch('pwd.getpwuid', mock.MagicMock(return_value=mock.MagicMock(pw_name='bccvl'))) @mock.patch('org.bccvl.tasks.compute.run_script') def setUp(self, mock_run_script): self.portal = self.layer['portal'] self.experiments = self.portal[defaults.EXPERIMENTS_FOLDER_ID] # create and run sdm experiment formhelper = SDMExperimentHelper(self.portal) sdmform = formhelper.get_form() sdmform.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request sdmform.update() # setup mock_run_script mock_run_script.side_effect = formhelper.mock_run_script transaction.commit() # We should have only one SDM sdmexp = self.experiments.values()[0] self.form = EnsembleExperimentHelper(self.portal, sdmexp) def test_add_experiment_missing_input(self): form = self.form.get_form() # remove required field del form.request.form['form.widgets.IDublinCore.title'] form.request.form.update({ 'form.buttons.create': 'Create', }) # submit form form.update() errors = [e for e in form.widgets.errors] self.assertEqual(len(errors), 1) # check self.experiments still empty # IStatusMessage @mock.patch('org.bccvl.tasks.compute.run_script') def test_add_experiment(self, mock_run_script): form = self.form.get_form() form.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request form.update() self.assertEqual(form.status, u'') self.assertEqual(len(form.widgets.errors), 0) self.assertIn('my-en-experiment', self.experiments) exp = self.experiments['my-en-experiment'] self.assertEqual(exp.datasets, {unicode(self.form.sdmexp.UID()): [unicode(self.form.sdmproj.UID())]}) # get result container: (there is only one) self.assertEqual(len(exp.objectIds()), 1) result = exp.objectValues()[0] # FIXME: test result.job_params self.assertEqual(result.job_params['datasets'], [unicode(self.form.sdmproj.UID())]) self.assertEqual(result.job_params['resolution'], u'Resolution30s') # no result files yet self.assertEqual(len(result.keys()), 0) # test job state jt = IExperimentJobTracker(exp) self.assertEqual(jt.state, u'QUEUED') # setup mock_run_script mock_run_script.side_effect = self.form.mock_run_script # after transaction commit the job should finish transaction.commit() self.assertEqual(jt.state, u'COMPLETED') # and we should have a result as well self.assertGreaterEqual(len(result.keys()), 1)
class ExperimentEnsembleAddTest(unittest.TestCase): # Use functional layer for now to get a new demostorage layer for each test layer = BCCVL_FUNCTIONAL_TESTING # integration testing gives only a new transaction for each test # (rolled back at end) # layer = BCCVL_INTEGRATION_TESTING @mock.patch('pwd.getpwuid', mock.MagicMock(return_value=mock.MagicMock(pw_name='bccvl'))) @mock.patch('org.bccvl.tasks.compute.run_script') def setUp(self, mock_run_script): self.portal = self.layer['portal'] self.experiments = self.portal[defaults.EXPERIMENTS_FOLDER_ID] # create and run sdm experiment formhelper = SDMExperimentHelper(self.portal) sdmform = formhelper.get_form() sdmform.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request sdmform.update() # setup mock_run_script mock_run_script.side_effect = formhelper.mock_run_script transaction.commit() # We should have only one SDM sdmexp = self.experiments.values()[0] self.form = EnsembleExperimentHelper(self.portal, sdmexp) def test_add_experiment_missing_input(self): form = self.form.get_form() # remove required field del form.request.form['form.widgets.IDublinCore.title'] form.request.form.update({ 'form.buttons.create': 'Create', }) # submit form form.update() errors = [e for e in form.widgets.errors] self.assertEqual(len(errors), 1) # check self.experiments still empty # IStatusMessage @mock.patch('org.bccvl.tasks.compute.run_script') def test_add_experiment(self, mock_run_script): form = self.form.get_form() form.request.form.update({ 'form.buttons.save': 'Create and start', }) # update form with updated request form.update() self.assertEqual(form.status, u'') self.assertEqual(len(form.widgets.errors), 0) self.assertIn('my-en-experiment', self.experiments) exp = self.experiments['my-en-experiment'] self.assertEqual(exp.datasets, { unicode(self.form.sdmexp.UID()): [unicode(self.form.sdmproj.UID())] }) # get result container: (there is only one) self.assertEqual(len(exp.objectIds()), 1) result = exp.objectValues()[0] # FIXME: test result.job_params self.assertEqual(result.job_params['datasets'], [unicode(self.form.sdmproj.UID())]) self.assertEqual(result.job_params['resolution'], u'Resolution30s') # no result files yet self.assertEqual(len(result.keys()), 0) # test job state jt = IExperimentJobTracker(exp) self.assertEqual(jt.state, u'QUEUED') # setup mock_run_script mock_run_script.side_effect = self.form.mock_run_script # after transaction commit the job should finish transaction.commit() self.assertEqual(jt.state, u'COMPLETED') # and we should have a result as well self.assertGreaterEqual(len(result.keys()), 1)