def test_normalize_true_true(self): with patch('jnpr.junos.Device.transform', new_callable=PropertyMock) as mock_transform: self.dev._normalize = True def function(x): return x decorator = normalizeDecorator(function) decorator(self.dev, normalize=True) self.assertFalse(mock_transform.called)
def test_normalize_true_true_except(self): with patch('jnpr.junos.Device.transform', new_callable=PropertyMock) as mock_transform: self.dev._normalize = True def function(*args, **kwargs): raise Exception() decorator = normalizeDecorator(function) with self.assertRaises(Exception): decorator(self.dev, normalize=True) self.assertFalse(mock_transform.called)
def test_normalize_false_false(self): with patch('jnpr.junos.Device.transform', new_callable=PropertyMock) as mock_transform: self.dev._normalize = False def function(x): return x decorator = normalizeDecorator(function) decorator(self.dev, normalize=False) self.assertFalse(mock_transform.called)
def test_normalize_true_false(self): with patch('jnpr.junos.Device.transform', new_callable=PropertyMock) as mock_transform: mock_transform.return_value = 'o.g.' self.dev._normalize = True def function(x): return x decorator = normalizeDecorator(function) decorator(self.dev, normalize=False) calls = [call(), call(self.dev._nc_transform), call('o.g.')] mock_transform.assert_has_calls(calls)
def test_normalize_true_false_except(self): with patch('jnpr.junos.Device.transform', new_callable=PropertyMock) as mock_transform: mock_transform.return_value = 'o.g.' self.dev._normalize = True def function(*args, **kwargs): raise Exception() decorator = normalizeDecorator(function) with self.assertRaises(Exception): decorator(self.dev, normalize=False) calls = [call(), call(self.dev._nc_transform), call('o.g.')] mock_transform.assert_has_calls(calls)
def test_normalize_false_true(self): with patch("jnpr.junos.Device.transform", new_callable=PropertyMock) as mock_transform: mock_transform.return_value = "o.g." self.dev._normalize = False def function(x): return x decorator = normalizeDecorator(function) decorator(self.dev, normalize=True) calls = [call(), call(self.dev._norm_transform), call("o.g.")] # print mock_transform.call_args_list mock_transform.assert_has_calls(calls)