def test_parse_message_empty(self, dll):
dll.ldap_first_entry.return_value = None
mock_l = mock.Mock()
mock_m = mock.Mock()
message = Message(mock_l, mock_m)
self.assertEqual(parse_message(message), [])
def test_parse_message(self, dll):
dll.ldap_first_entry.return_value = 'entry_1'
dll.ldap_next_entry.side_effect = ['entry_2', None]
dll.ldap_first_attributeW.side_effect = [1, 3]
dll.ldap_next_attributeW.side_effect = [2, None, 4, None]
dll.ldap_get_valuesW.side_effect = [['1.1'], ['1.2', '1.3'], None, ['4']]
expects = [{1: ['1.1'], 2: ['1.2', '1.3']}, {3: [], 4: ['4']}]
mock_l = mock.Mock()
mock_m = mock.Mock()
message = Message(mock_l, mock_m)
self.assertEqual(parse_message(message), expects)
