class SearchSiteTestCase(TestCase):
def setUp(self):
super(SearchSiteTestCase, self).setUp()
self.site = SearchSite()
def test_register(self):
self.assertRaises(AttributeError, self.site.register, MockNotAModel)
self.site.register(MockModel)
self.assertEqual(len(self.site._registry), 1)
self.assert_(MockModel in self.site._registry)
self.assertRaises(AlreadyRegistered, self.site.register, MockModel)
def test_unregister(self):
self.assertRaises(NotRegistered, self.site.unregister, MockModel)
# Depends on proper function of register.
self.site.register(MockModel)
self.site.unregister(MockModel)
self.assertEqual(len(self.site._registry), 0)
self.assertFalse(MockModel in self.site._registry)
def test_get_index(self):
self.assertRaises(NotRegistered, self.site.get_index, MockModel)
self.site.register(MockModel)
self.assert_(isinstance(self.site.get_index(MockModel), BasicSearchIndex))
def test_get_indexes(self):
self.assertEqual(self.site.get_indexes(), {})
self.site.register(MockModel)
indexes = self.site.get_indexes()
self.assert_(isinstance(indexes, dict))
self.assertEqual(len(indexes), 1)
self.assert_(MockModel in indexes)
def test_get_indexed_models(self):
self.assertEqual(self.site.get_indexed_models(), [])
self.site.register(MockModel)
indexed_models = self.site.get_indexed_models()
self.assertEqual(len(indexed_models), 1)
self.assert_(MockModel in indexed_models)
def test_build_unified_schema(self):
self.site.register(MockModel)
content_field_name, fields = self.site.build_unified_schema()
self.assertEqual(content_field_name, 'text')
self.assertEqual(fields, [{'indexed': 'true', 'type': 'text', 'field_name': 'text', 'multi_valued': 'false'}])
self.site.register(AnotherMockModel)
content_field_name, fields = self.site.build_unified_schema()
self.assertEqual(content_field_name, 'text')
self.assertEqual(fields, [{'indexed': 'true', 'type': 'text', 'field_name': 'text', 'multi_valued': 'false'}])
class SearchSiteTestCase(TestCase):
def setUp(self):
super(SearchSiteTestCase, self).setUp()
self.site = SearchSite()
def test_register(self):
self.assertRaises(AttributeError, self.site.register, MockNotAModel)
self.site.register(MockModel)
self.assertEqual(len(self.site._registry), 1)
self.assert_(MockModel in self.site._registry)
self.assertRaises(AlreadyRegistered, self.site.register, MockModel)
def test_unregister(self):
self.assertRaises(NotRegistered, self.site.unregister, MockModel)
# Depends on proper function of register.
self.site.register(MockModel)
self.site.unregister(MockModel)
self.assertEqual(len(self.site._registry), 0)
self.assertFalse(MockModel in self.site._registry)
def test_get_index(self):
self.assertRaises(NotRegistered, self.site.get_index, MockModel)
self.site.register(MockModel)
self.assert_(isinstance(self.site.get_index(MockModel), indexes.BasicSearchIndex))
def test_get_indexes(self):
self.assertEqual(self.site.get_indexes(), {})
self.site.register(MockModel)
indexes = self.site.get_indexes()
self.assert_(isinstance(indexes, dict))
self.assertEqual(len(indexes), 1)
self.assert_(MockModel in indexes)
def test_get_indexed_models(self):
self.assertEqual(self.site.get_indexed_models(), [])
self.site.register(MockModel)
indexed_models = self.site.get_indexed_models()
self.assertEqual(len(indexed_models), 1)
self.assert_(MockModel in indexed_models)
def test_build_unified_schema(self):
self.site.register(MockModel)
content_field_name, fields = self.site.build_unified_schema()
self.assertEqual(content_field_name, 'text')
self.assertEqual(fields, [{'indexed': 'true', 'type': 'text', 'field_name': 'text', 'multi_valued': 'false'}])
self.site.register(AnotherMockModel)
content_field_name, fields = self.site.build_unified_schema()
self.assertEqual(content_field_name, 'text')
self.assertEqual(fields, [{'indexed': 'true', 'type': 'text', 'field_name': 'text', 'multi_valued': 'false'}])
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, InvalidSeachIndex)
self.assertRaises(SearchFieldError, self.site.build_unified_schema)
def test_update_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
mock.user = '******' % mock.id
mock.pub_date = datetime.datetime(2009, 1, 31, 4, 19, 0)
self.assertEqual(self.site.update_object(mock), True)
def test_remove_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
self.assertEqual(self.site.remove_object(mock), True)
class SearchSiteTestCase(TestCase):
def setUp(self):
super(SearchSiteTestCase, self).setUp()
self.site = SearchSite()
def test_register(self):
self.assertRaises(AttributeError, self.site.register, MockNotAModel)
self.site.register(MockModel)
self.assertEqual(len(self.site._registry), 1)
self.assert_(MockModel in self.site._registry)
self.assertRaises(AlreadyRegistered, self.site.register, MockModel)
def test_unregister(self):
self.assertRaises(NotRegistered, self.site.unregister, MockModel)
# Depends on proper function of register.
self.site.register(MockModel)
self.site.unregister(MockModel)
self.assertEqual(len(self.site._registry), 0)
self.assertFalse(MockModel in self.site._registry)
def test_get_index(self):
self.assertRaises(NotRegistered, self.site.get_index, MockModel)
self.site.register(MockModel)
self.assert_(isinstance(self.site.get_index(MockModel), BasicSearchIndex))
def test_get_indexes(self):
self.assertEqual(self.site.get_indexes(), {})
self.site.register(MockModel)
indexes = self.site.get_indexes()
self.assert_(isinstance(indexes, dict))
self.assertEqual(len(indexes), 1)
self.assert_(MockModel in indexes)
def test_get_indexed_models(self):
self.assertEqual(self.site.get_indexed_models(), [])
self.site.register(MockModel)
indexed_models = self.site.get_indexed_models()
self.assertEqual(len(indexed_models), 1)
self.assert_(MockModel in indexed_models)
def test_all_searchfields(self):
self.site.register(MockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_("text" in fields)
self.assert_(isinstance(fields["text"], CharField))
self.assertEqual(fields["text"].document, True)
self.assertEqual(fields["text"].use_template, True)
self.site.register(AnotherMockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_("text" in fields)
self.assert_(isinstance(fields["text"], CharField))
self.assertEqual(fields["text"].document, True)
self.assertEqual(fields["text"].use_template, True)
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, InvalidSeachIndex)
self.assertRaises(SearchFieldError, self.site.all_searchfields)
def test_update_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
mock.user = "******" % mock.id
mock.pub_date = datetime.datetime(2009, 1, 31, 4, 19, 0)
self.assertEqual(self.site.update_object(mock), True)
def test_remove_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
self.assertEqual(self.site.remove_object(mock), True)
class SearchSiteTestCase(TestCase):
def setUp(self):
super(SearchSiteTestCase, self).setUp()
self.site = SearchSite()
def test_register(self):
self.assertRaises(AttributeError, self.site.register, MockNotAModel)
self.site.register(MockModel)
self.assertEqual(len(self.site._registry), 1)
self.assert_(MockModel in self.site._registry)
self.assertRaises(AlreadyRegistered, self.site.register, MockModel)
def test_unregister(self):
self.assertRaises(NotRegistered, self.site.unregister, MockModel)
# Depends on proper function of register.
self.site.register(MockModel)
self.site.unregister(MockModel)
self.assertEqual(len(self.site._registry), 0)
self.assertFalse(MockModel in self.site._registry)
def test_get_index(self):
self.assertRaises(NotRegistered, self.site.get_index, MockModel)
self.site.register(MockModel)
self.assert_(
isinstance(self.site.get_index(MockModel), BasicSearchIndex))
def test_get_indexes(self):
self.assertEqual(self.site.get_indexes(), {})
self.site.register(MockModel)
indexes = self.site.get_indexes()
self.assert_(isinstance(indexes, dict))
self.assertEqual(len(indexes), 1)
self.assert_(MockModel in indexes)
def test_get_indexed_models(self):
self.assertEqual(self.site.get_indexed_models(), [])
self.site.register(MockModel)
indexed_models = self.site.get_indexed_models()
self.assertEqual(len(indexed_models), 1)
self.assert_(MockModel in indexed_models)
def test_all_searchfields(self):
self.site.register(MockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.site.register(AnotherMockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, AlternateValidSearchIndex)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 3)
self.assertEqual(sorted(fields.keys()), ['author', 'text', 'title'])
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.assert_('title' in fields)
self.assert_(isinstance(fields['title'], CharField))
self.assertEqual(fields['title'].document, False)
self.assertEqual(fields['title'].use_template, False)
self.assertEqual(fields['title'].faceted, True)
self.assertEqual(fields['title'].indexed, True)
self.assert_('author' in fields)
self.assert_(isinstance(fields['author'], CharField))
self.assertEqual(fields['author'].document, False)
self.assertEqual(fields['author'].use_template, False)
self.assertEqual(fields['author'].faceted, True)
self.assertEqual(fields['author'].index_fieldname, 'author')
self.site.unregister(MockModel)
self.site.register(MockModel, ValidSearchIndex)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 4)
self.assertEqual(sorted(fields.keys()),
['author', 'name', 'text', 'title'])
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, False)
self.assert_('title' in fields)
self.assert_(isinstance(fields['title'], CharField))
self.assertEqual(fields['title'].document, False)
self.assertEqual(fields['title'].use_template, False)
self.assertEqual(fields['title'].faceted, True)
self.assertEqual(fields['title'].indexed, True)
self.assert_('author' in fields)
self.assert_(isinstance(fields['author'], CharField))
self.assertEqual(fields['author'].document, False)
self.assertEqual(fields['author'].use_template, False)
self.assertEqual(fields['author'].faceted, True)
self.assertEqual(fields['author'].index_fieldname, 'author')
self.assertEqual(fields['name'].document, False)
self.assertEqual(fields['name'].use_template, False)
self.assertEqual(fields['name'].faceted, False)
self.assertEqual(fields['name'].index_fieldname, 'name')
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, InvalidSearchIndex)
self.assertRaises(SearchFieldError, self.site.all_searchfields)
def test_get_index_fieldname(self):
self.assertEqual(self.site._field_mapping, None)
self.site.register(MockModel, ValidSearchIndex)
self.site.register(AnotherMockModel)
field = self.site.get_index_fieldname('text')
self.assertEqual(self.site._field_mapping, {
'text': 'text',
'title': 'title',
'author': 'name'
})
self.assertEqual(self.site.get_index_fieldname('text'), 'text')
self.assertEqual(self.site.get_index_fieldname('author'), 'name')
self.assertEqual(self.site.get_index_fieldname('title'), 'title')
# Reset the internal state to test the invalid case.
self.site._field_mapping = None
self.assertEqual(self.site._field_mapping, None)
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, AlternateValidSearchIndex)
self.assertRaises(SearchFieldError, self.site.get_index_fieldname,
'text')
def test_update_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
mock.user = '******' % mock.id
mock.pub_date = datetime.datetime(2009, 1, 31, 4, 19, 0)
self.assertEqual(self.site.update_object(mock), True)
def test_remove_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
self.assertEqual(self.site.remove_object(mock), True)
class SearchSiteTestCase(TestCase):
def setUp(self):
super(SearchSiteTestCase, self).setUp()
self.site = SearchSite()
def test_register(self):
self.assertRaises(AttributeError, self.site.register, MockNotAModel)
self.site.register(MockModel)
self.assertEqual(len(self.site._registry), 1)
self.assert_(MockModel in self.site._registry)
self.assertRaises(AlreadyRegistered, self.site.register, MockModel)
def test_unregister(self):
self.assertRaises(NotRegistered, self.site.unregister, MockModel)
# Depends on proper function of register.
self.site.register(MockModel)
self.site.unregister(MockModel)
self.assertEqual(len(self.site._registry), 0)
self.assertFalse(MockModel in self.site._registry)
def test_get_index(self):
self.assertRaises(NotRegistered, self.site.get_index, MockModel)
self.site.register(MockModel)
self.assert_(isinstance(self.site.get_index(MockModel), BasicSearchIndex))
def test_get_indexes(self):
self.assertEqual(self.site.get_indexes(), {})
self.site.register(MockModel)
indexes = self.site.get_indexes()
self.assert_(isinstance(indexes, dict))
self.assertEqual(len(indexes), 1)
self.assert_(MockModel in indexes)
def test_get_indexed_models(self):
self.assertEqual(self.site.get_indexed_models(), [])
self.site.register(MockModel)
indexed_models = self.site.get_indexed_models()
self.assertEqual(len(indexed_models), 1)
self.assert_(MockModel in indexed_models)
def test_all_searchfields(self):
self.site.register(MockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.site.register(AnotherMockModel)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 1)
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, AlternateValidSearchIndex)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 5)
self.assertEqual(sorted(fields.keys()), ['author', 'author_exact', 'text', 'title', 'title_exact'])
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, True)
self.assert_('title' in fields)
self.assert_(isinstance(fields['title'], CharField))
self.assertEqual(fields['title'].document, False)
self.assertEqual(fields['title'].use_template, False)
self.assertEqual(fields['title'].faceted, True)
self.assertEqual(fields['title'].indexed, True)
self.assert_('author' in fields)
self.assert_(isinstance(fields['author'], CharField))
self.assertEqual(fields['author'].document, False)
self.assertEqual(fields['author'].use_template, False)
self.assertEqual(fields['author'].faceted, True)
self.assertEqual(fields['author'].stored, True)
self.assertEqual(fields['author'].index_fieldname, 'author')
self.site.unregister(MockModel)
self.site.register(MockModel, ValidSearchIndex)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 6)
self.assertEqual(sorted(fields.keys()), ['author', 'author_exact', 'name', 'text', 'title', 'title_exact'])
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, False)
self.assert_('title' in fields)
self.assert_(isinstance(fields['title'], CharField))
self.assertEqual(fields['title'].document, False)
self.assertEqual(fields['title'].use_template, False)
self.assertEqual(fields['title'].faceted, True)
self.assertEqual(fields['title'].indexed, True)
self.assert_('author' in fields)
self.assert_(isinstance(fields['author'], CharField))
self.assertEqual(fields['author'].document, False)
self.assertEqual(fields['author'].use_template, False)
self.assertEqual(fields['author'].faceted, True)
self.assertEqual(fields['author'].index_fieldname, 'author')
self.assertEqual(fields['name'].document, False)
self.assertEqual(fields['name'].use_template, False)
self.assertEqual(fields['name'].faceted, False)
self.assertEqual(fields['name'].index_fieldname, 'name')
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, MultiValueValidSearchIndex)
fields = self.site.all_searchfields()
self.assertEqual(len(fields), 4)
self.assertEqual(sorted(fields.keys()), ['author', 'name', 'text', 'title'])
self.assert_('text' in fields)
self.assert_(isinstance(fields['text'], CharField))
self.assertEqual(fields['text'].document, True)
self.assertEqual(fields['text'].use_template, False)
self.assert_('title' in fields)
self.assert_(isinstance(fields['title'], CharField))
self.assertEqual(fields['title'].document, False)
self.assertEqual(fields['title'].use_template, False)
self.assertEqual(fields['title'].faceted, False)
self.assertEqual(fields['title'].indexed, False)
self.assert_('author' in fields)
self.assert_(isinstance(fields['author'], MultiValueField))
self.assertEqual(fields['author'].document, False)
self.assertEqual(fields['author'].use_template, False)
self.assertEqual(fields['author'].stored, False)
self.assertEqual(fields['author'].faceted, False)
self.assertEqual(fields['author'].index_fieldname, 'author')
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, InvalidSearchIndex)
self.assertRaises(SearchFieldError, self.site.all_searchfields)
def test_get_index_fieldname(self):
self.assertEqual(self.site._cached_field_mapping, None)
self.site.register(MockModel, ValidSearchIndex)
self.site.register(AnotherMockModel)
self.site.get_index_fieldname('text')
self.assertEqual(self.site._cached_field_mapping, {
'text': {'index_fieldname': 'text', 'facet_fieldname': None},
'title': {'index_fieldname': 'title', 'facet_fieldname': None},
'author': {'index_fieldname': 'name', 'facet_fieldname': None},
})
self.assertEqual(self.site.get_index_fieldname('text'), 'text')
self.assertEqual(self.site.get_index_fieldname('author'), 'name')
self.assertEqual(self.site.get_index_fieldname('title'), 'title')
# Reset the internal state to test the invalid case.
self.site._cached_field_mapping = None
self.assertEqual(self.site._cached_field_mapping, None)
self.site.unregister(AnotherMockModel)
self.site.register(AnotherMockModel, AlternateValidSearchIndex)
self.assertRaises(SearchFieldError, self.site.get_index_fieldname, 'text')
def test_basic_get_facet_field_name(self):
self.assertEqual(self.site._cached_field_mapping, None)
self.site.register(MockModel, AlternateValidSearchIndex)
self.site.register(AnotherMockModel)
self.site.get_facet_field_name('text')
self.assertEqual(self.site._cached_field_mapping, {
'author': {'facet_fieldname': None, 'index_fieldname': 'author'},
'author_exact': {'facet_fieldname': 'author',
'index_fieldname': 'author_exact'},
'text': {'facet_fieldname': None, 'index_fieldname': 'text'},
'title': {'facet_fieldname': None, 'index_fieldname': 'title'},
'title_exact': {'facet_fieldname': 'title', 'index_fieldname': 'title_exact'},
})
self.assertEqual(self.site.get_index_fieldname('text'), 'text')
self.assertEqual(self.site.get_index_fieldname('author'), 'author')
self.assertEqual(self.site.get_index_fieldname('title'), 'title')
self.assertEqual(self.site.get_facet_field_name('text'), 'text')
self.assertEqual(self.site.get_facet_field_name('author'), 'author')
self.assertEqual(self.site.get_facet_field_name('title'), 'title')
def test_more_advanced_get_facet_field_name(self):
self.assertEqual(self.site._cached_field_mapping, None)
self.site.register(MockModel, ExplicitFacetSearchIndex)
self.site.register(AnotherMockModel)
self.site.get_facet_field_name('text')
self.assertEqual(self.site._cached_field_mapping, {
'author': {'facet_fieldname': None, 'index_fieldname': 'author'},
'author_exact': {'facet_fieldname': 'author', 'index_fieldname': 'author_exact'},
'bare_facet': {'facet_fieldname': 'bare_facet', 'index_fieldname': 'bare_facet'},
'text': {'facet_fieldname': None, 'index_fieldname': 'text'},
'title': {'facet_fieldname': None, 'index_fieldname': 'title'},
'title_facet': {'facet_fieldname': 'title', 'index_fieldname': 'title_facet'},
})
self.assertEqual(self.site.get_facet_field_name('title'), 'title')
self.assertEqual(self.site.get_facet_field_name('title'), 'title')
self.assertEqual(self.site.get_facet_field_name('bare_facet'), 'bare_facet')
def test_update_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
mock.user = '******' % mock.id
mock.pub_date = datetime.datetime(2009, 1, 31, 4, 19, 0)
self.assertEqual(self.site.update_object(mock), True)
def test_remove_object(self):
self.site.register(MockModel, FakeSearchIndex)
mock = MockModel()
mock.pk = 20
self.assertEqual(self.site.remove_object(mock), True)
