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)