变量被访问的Python魔术方法?

时间:2014-04-14 02:57:03

标签: python

我正在寻找一种类似于__get__方法的魔术方法,但在我的情况下,变量不在另一个类中,我需要类似的东西:

class A(object):
  def __similar_to_get__(self):
    print 'called'
    return self

a = A()
b = a
>>> 'called'

有可能吗?

我问这个的原因是,我正在使用python模拟库,让我们说我正在测试的函数使用URI属性,我想模拟它以在后续调用中返回不同的值。例如:

class WebService(obj):
  URI = 'http://works.com'

  def dowork(self):
     call_api(self.URI)

让我模拟失败我正在使用模拟库:

 mock = MagicMock()
 mock.side_effect = ['http://fail.com', 'http://works.com']

 with patch('WebService.URI', mock):
     # do the testing

但问题是我只能通过调用可调用的mock()来模拟返回网址,而不仅仅是访问mock

PS:我是一个模拟菜鸟。

1 个答案:

答案 0 :(得分:1)

没有直接回答我自己的问题,我设法使用一个属性解决这个问题:

    from dps.px.pxpost import PxPost
    mock = MagicMock()
    mock.side_effect = ['http://doesnotexist', PxPost.URI]

    def URI(self):
        return mock()

    with patch('django.conf.settings.CELERY_ALWAYS_EAGER', True, create=True):
        with patch('dps.px.pxpost.PxPost.URI', property(URI)):
            self.transaction.process()
            for sub_transaction in self.transaction.sub_transactions.all():
                self.assertTrue(isinstance(sub_transaction.state, CompletedSubTransaction))
                self.assertTrue(sub_transaction.transaction_logs.count() > 0)
            self.assertTrue(isinstance(self.transaction.state, CompletedTransaction))
相关问题