def test_reconstructor_called_when_no_measures_specified(self, db): from frontend.management.commands.import_measures \ import conditional_constraint_and_index_reconstructor with conditional_constraint_and_index_reconstructor({'measure': None}): pass execute = db.connection.cursor.return_value.__enter__.return_value.execute execute.assert_called()
def test_reconstructor_called_when_no_measures_specified(self, conn): from frontend.management.commands.import_measures \ import conditional_constraint_and_index_reconstructor with conditional_constraint_and_index_reconstructor({}): pass calls = conn.cursor.return_value.__enter__\ .return_value.execute.mock_calls self.assertGreater(calls, 0)
def test_reconstructor_called_when_enabled(self, db): from frontend.management.commands.import_measures import ( conditional_constraint_and_index_reconstructor, ) with conditional_constraint_and_index_reconstructor(True): pass execute = db.connection.cursor.return_value.__enter__.return_value.execute execute.assert_called()