패치 - 패치 된 클래스의 인스턴스가 클래스 함수를 호출했는지 확인하는 방법은 무엇입니까?

Question

클래스를 패치하려고합니다. 테스트중인 함수에 패치 된 클래스의 인스턴스를 만들고, 해당 인스턴스에서 클래스 함수를 호출 한 다음 해당 클래스 함수가 ​​호출되었는지 테스트하려고합니다. 어떻게해야합니까?

테스트 파일 :

class TestChannel(unittest.TestCase): 
    @patch("notification.models.Channel") 
    def testAddChannelWithNamePutsChannel(self, *args): 
     addChannelWithName("channel1") 
     self.assertTrue(Channel.put.called) 

코드를 테스트 할 :

def addChannelWithName(name): 
    channel = Channel(name = name) 
    channel.put() 

는 기본적으로, 그 채널을 테스트 할

다음은 내 코드의 기본 변형입니다. put()이 호출되었습니다. 물론이 코드는 클래스 함수 put()이 인스턴스가 아니라 일반적으로 호출되는지 검사하기 때문에 작동하지 않습니다. 그것을 통과시키기 위해 인스턴스에서 호출해야합니다. Channel 클래스의 다른 부분을 조롱하는 몇 가지 변형을 시도했지만 반환 값이지만 작동하도록 설정할 수는 없습니다. 어떻게해야합니까? 많이 고마워!

Answer 1

addChannelWithName 함수를 만들면 Channel 인스턴스가 반환됩니다.

def addChannelWithName(name): 
    channel = Channel(name = name) 
    channel.put() 
    return channel 

class TestChannel(unittest.TestCase): 
    @patch("notification.models.Channel") 
    def testAddChannelWithNamePutsChannel(self, *args): 
     channel = addChannelWithName("channel1") 
     self.assertTrue(channel.put.called)