def test_get_normalized_translated_data__typical(self, mock_norm): """ verify the runtime behavior of get_normalized_translated_data() """ unit = Unit({}) with mock.patch.object(unit, "get_translated_data") as mock_tr: retval = unit.get_normalized_translated_data('foo') # get_translated_data('foo') was called mock_tr.assert_called_with("foo") # normalize_rfc822_value(x) was called mock_norm.assert_called_with(mock_tr()) # return value was returned self.assertEqual(retval, mock_norm())
def test_get_normalized_translated_data__no_translation(self, mock_norm): """ verify the runtime behavior of get_normalized_translated_data() """ unit = Unit({}) with mock.patch.object(unit, "get_translated_data") as mock_tr: mock_tr.return_value = None retval = unit.get_normalized_translated_data('foo') # get_translated_data('foo') was called mock_tr.assert_called_with("foo") # normalize_rfc822_value(x) was NOT called self.assertEqual(mock_norm.call_count, 0) # return value was returned self.assertEqual(retval, 'foo')