def test_get_related(self): entry1 = Entry(owner=self.user2, title='foo1', content='bar', draft=False, tags='python, django') entry1.save() entry1.sites = Site.objects.all() entry1.save() entry2 = Entry(owner=self.user2, title='foo2', content='bar', featured=True, draft=False, tags='python, django') entry2.save() entry2.sites = Site.objects.all() entry2.save() entry3 = Entry(owner=self.user2, title='foo3', content='bar', draft=False, tags='python, django') entry3.save() entry3.sites = Site.objects.all() entry3.save() self.assertEqual(len(entry3.get_related()), 2) entry4 = Entry(owner=self.user2, title='foo3', content='bar', draft=False) entry4.save() entry4.sites = Site.objects.all() entry4.related_content.add(entry1) entry4.related_content.add(entry2) entry4.save() self.assertEqual(len(entry4.get_related(max=3)), 2) self.assertEqual(len(entry4.get_related(max=1)), 1) # clean up the objecs so haystack doesn't have indexes laying around entry1.delete() entry2.delete() entry3.delete() entry4.delete()
val = getNodeValue(node) if node.tagName == 'author': owner = User.objects.filter(username=val)[0] if owner: e.owner = owner else: e.owner = User.objects.filter(pk=1)[0] elif hasattr(e, node.tagName): setattr(e, node.tagName, val) else: if node.tagName == 'lastModified': e.modified_on = val elif node.tagName == 'ispublished': e.draft = not val elif node.tagName == 'pubDate': e.published_on = val elif node.tagName == 'iscommentsenabled': e.allow_comments = val elif node.tagName == 'tags': for node2 in node.childNodes: if getValidNode(node2): if node2.tagName == 'tag': val = getNodeValue(node2) if val: tags.append(val) e.save() e.sites = Site.objects.filter(pk=1) e.tags = ",".join(tags) e.save()