def test_delete_text_response(self): processor = MongoProcessor() responses = Responses.objects(bot="tests", name="utter_happy") expected_length = responses.__len__() - 1 response = responses[0] expected_text = response.text.text processor.remove_document(Responses, response.id, "tests", "testUser") actual = list(processor.get_response("utter_happy", "tests")) assert actual.__len__() == expected_length assert all(expected_text != item["value"]["text"] for item in actual if "text" in item["value"])
def test_delete_training_example(self): processor = MongoProcessor() training_examples = TrainingExamples.objects(bot="tests", intent="get_priority", status=True) expected_length = training_examples.__len__() - 1 training_example = training_examples[0] expected_text = training_example.text processor.remove_document(TrainingExamples, training_example.id, "tests", "testUser") new_training_examples = list( processor.get_training_examples(intent="get_priority", bot="tests")) assert new_training_examples.__len__() == expected_length assert any(expected_text != example["text"] for example in new_training_examples)