def test_add_entity(self): processor = MongoProcessor() assert processor.add_entity("file_text", "tests", "testUser") == None slots = Slots.objects(bot="tests") new_slot = slots.get(name="file_text") enitity = Entities.objects(bot="tests").get(name="file_text") assert slots.__len__() == 3 assert new_slot.name == "file_text" assert new_slot.type == "text" assert enitity.name == "file_text"
def test_add_blank_entity(self): processor = MongoProcessor() with pytest.raises(ValidationError): processor.add_entity(" ", "tests", "testUser")
def test_add_entity_duplicate(self): processor = MongoProcessor() with pytest.raises(Exception): assert processor.add_entity("file_text", "tests", "testUser")
def test_add_empty_entity(self): processor = MongoProcessor() with pytest.raises(AssertionError): processor.add_entity("", "tests", "testUser")