我在自己的Web应用程序中为runReaderT (load "myFile.txt") "./" :: IO String
方法创建了一个单元测试,而当我运行Question.save()
时,出现如下所述的断言错误?
这是我python manage.py test
的一部分:
test.py
models.py的一部分:
def test_elasticsearch_upsert_on_save(self, ElasticsearchMock):
user = get_user_model().objects.create_user(
username = 'unittest',
password='unittest',
)
question_title = 'Unit test'
question_body = 'some long text'
q = Question(
title=question_title,
question=question_body,
user=user,
)
q.save()
self.assertIsNotNone(q.id)
self.assertTrue(ElasticsearchMock.called)
mock_client = ElasticsearchMock.return_value
mock_client.update.assert_called_once_with(
settings.ES_INDEX,
id=q.id,
body={
'doc': {
'_type': 'doc',
'text':'{}\n{}'.format(question_title, question_body),
'question_body':question_body,
'title': question_title,
'id': q.id,
'created': q.created,
},'doc_as_upsert': True,
}
)
我希望测试通过,但是出现以下错误:
def as_elasticsearch_dict(self):
return {
'_id': self.id,
'_type': 'doc',
'text': '{}\n{}'.format(self.title, self.question),
'question_body': self.question,
'title': self.title,
'id': self.id,
'created': self.created,
}
def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
super().save(force_insert=force_insert,
force_update=force_update,
using=using,
update_fields=update_fields)
elasticsearch.upsert(self)
答案 0 :(得分:0)
将doc
参数添加到您的模拟中:
mock_client.update.assert_called_once_with(
settings.ES_INDEX,
id=q.id,
doc='doc',
body={
'doc': {
'_type': 'doc',
'text':'{}\n{}'.format(question_title, question_body),
'question_body':question_body,
'title': question_title,
'id': q.id,
'created': q.created,
},'doc_as_upsert': True,
}
)
答案 1 :(得分:0)
如果将期望的呼叫和实际的呼叫放在下面,您会发现它们实际上是不同的,因此您的测试失败是有道理的。
expected=update('answerly', body={'doc': {'_type': 'doc', 'text': 'Unit test\nsome long text', 'question_body': 'some long text', 'title': 'Unit test', 'id': 1, 'created': datetime.datetime(2019, 9, 11, 19, 38, 37, 498420, tzinfo=)}, 'doc_as_upsert': True}, id=1)
actual= update('answerly', 'doc', body={'doc': {'_type': 'doc', 'text': 'Unit test\nsome long text', 'question_body': 'some long text', 'title': 'Unit test', 'id': 1, 'created': datetime.datetime(2019, 9, 11, 19, 38, 37, 498420, tzinfo=)}, 'doc_as_upsert': True}, id=1)
将doc
作为第二个参数应该可以解决您的问题:
mock_client.update.assert_called_once_with(
settings.ES_INDEX,
'doc', # Added
id=q.id,
body={
'doc': {
'_type': 'doc',
'text':'{}\n{}'.format(question_title, question_body),
'question_body':question_body,
'title': question_title,
'id': q.id,
'created': q.created,
},'doc_as_upsert': True,
}
)