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_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_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_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)
