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)
