def mock_source(name='', **kwargs): if not kwargs: kwargs = {'value': 0.5} obj = DataSource(name=name) obj.open = mock.Mock(wraps=obj.open) obj.close = mock.Mock(wraps=obj.close) obj.read = mock.Mock() obj.read.return_value = kwargs return obj
def mock_source(**kwargs): obj = DataSource(**kwargs) obj._read = mock.Mock(return_value={'field1': 5, 'field2': -7}) obj._check_conditions = mock.Mock(return_value=(True, None)) obj._update_conditions = mock.Mock() return obj