def test_remove_tag(self): node = Node(name='JeffTest') node.add_tag(metadata='Franklin', category='PTC-K') tags = node.tags.all() self.assertEqual(tags[0].metadata.name, 'Franklin') node.remove_tag(tags[0]) node = Node.objects.get(name='JeffTest') self.assertEqual(len(node.tags.all()), 0)
def test_add_tag(self): node = Node(name='WKCLSBIZTALBT03') meta = Metadata.objects.filter(name='eStore') category = Category.objects.filter(name='Applications') tag = Tag(metadata=meta[0], category=category[0]) node.add_tag(tag) db_tag = Tag.objects.get(pk=1) self.assertEqual(tag, db_tag)
def test_add_tag_params(self): node = Node(name='WKCLSBIZTALBT04') meta = Metadata.objects.get(name='TAA') category = Category.objects.get(name='Business Units') node.add_tag(metadata='TAA', category='Business Units') db_tag = Tag.objects.get(metadata__name=meta.name, category__name=category.name) self.assertEqual(db_tag.metadata.name, meta.name) self.assertEqual(db_tag.category.name, category.name)
def save(self): tags = process_tags_in_form(self) node = Node() node.name = self.cleaned_data['name'] metadata = Metadata.objects.get_or_create(name=node.name) category = Category.objects.get(pk=1) device_tag = Tag.objects.get_or_create(category=category, metadata=metadata[0]) node.description = self.cleaned_data['description'] node.save() node.tags.add(*tags) node.tags.add(device_tag[0])