单元Django TestCase上的断言错误

时间:2019-09-11 19:55:50

标签: python django-tests

我在自己的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)

2 个答案:

答案 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,
            }
        )