def test_add_highlighting_should_do_nothing_for_empty_fields(self): search_body = {'highlight' : { 'fields' : {} }} add_highlighting([], search_body) self.assertEqual({'highlight': { 'fields': {} }}, search_body)
def test_add_highlighting_should_include_every_field_passed(self): search_body = {'highlight' : { 'fields' : {} }} add_highlighting(['field_1', 'field_2'], search_body) self.assertEqual(search_body, { 'highlight': { 'fields': { 'field_1': {}, 'field_2': {} } } })