def sqlite_store(): test_violations_file = "test_violations.sqlite" mock_options = mock.Mock() mock_options.violation_dburl = "sqlite:///%s" % test_violations_file mock_options.violation_dbconfig = None mock_options.build_info = None yield ViolationStore(mock_options) os.unlink(test_violations_file)
def sqlite_store(): test_violations_filename = tempfile.mkstemp( suffix='test_violations.sqlite')[1] try: mock_options = mock.Mock() mock_options.violation_dburl = "sqlite:///%s" % test_violations_filename mock_options.violation_dbconfig = None mock_options.build_info = None yield ViolationStore(mock_options) finally: os.remove(test_violations_filename)
def mocked_store(): def mock_init_database(obj): obj.metadata = mock.MagicMock() obj.Violations = mock.MagicMock() obj.Methods = mock.MagicMock() with mock.patch('testify.plugins.violation_collector.SA'): mock_options = mock.Mock() mock_options.violation_dburl = "fake db url" mock_options.violation_dbconfig = None mock_options.build_info = None # we're doing our own method paching here because # mock.patch.object's side_effect functions are not passed in # the object. original_init_database = ViolationStore.init_database ViolationStore.init_database = mock_init_database yield ViolationStore(mock_options) ViolationStore.init_database = original_init_database