def test_publish_analysis(self):
analysis_service = AnalysisService()
with analysis_service.readonly_scope() as session:
analysis = session.query(Analysis).filter(Analysis.viewable_by == self._user_id).first()
analysis_service.publish_analysis(analysis.id)
with analysis_service.readonly_scope() as another_session:
updated_analysis = another_session.query(Analysis).filter(Analysis.id == analysis.id).one()
self.assertEqual(updated_analysis.viewable_by, None, "Expected viewable by field to be cleared out")
def test_publish_analysis(self):
mock_query_result = MagicMock()
mock_query_result.one = MagicMock(return_value=self.sample_analysis)
mock_query = MagicMock()
mock_query.filter = MagicMock()
mock_query.filter.return_value = mock_query_result
self._mock_session.query = MagicMock()
self._mock_session.query.return_value = mock_query
analysis_service = AnalysisService(self._mock_session)
analysis_service.publish_analysis(self.sample_analysis.id)
self.assertEqual(self.sample_analysis.viewable_by, None)
self.assertEqual(self.sample_analysis.result_dataset.viewable_by_user_id, None)
