BOTO 예외로 모의 side_effect를 할당하고 전화를 주장

Question

I S3에서 파일을 다운로드 할 수 BOTO를 사용하는 클래스, 메소드가 연결을 initiliase하는 S3Connection 클래스를 사용하고 나중에 얻을 수있는 get_key에게 방법을 사용하여이 파일'. 여기 는 코드 블록

import sys 

from boto.exception import S3ResponseError, S3DataError 
from boto.s3.connection import S3Connection 

class MyClass(object): 
    def __init__(self): 
     self.s3conn = S3Connection(
       AWS_ACCESS_KEY_ID, 
       AWS_SECRET_ACCESS_KEY, 
       host=HOST, 
       is_secure=True 
     ) 
     self.bucket = self.s3conn.get_bucket(BUCKET_NAME) 
    def get_file(self, key): 
     try: 
      return self.bucket.get_key(key) 
     except (S3ResponseError, S3DataError): 
      sys.exit(3) 

내가 무조건 sys.exit 방법을 조롱하고 주장 할 수 있도록이 불렀다는 get_bucket 방법을 조롱하고 그것을 S3ResponseError의 side_effect을 제공 할 것입니다 .

내가 어떻게하는지 알려드립니다.

import unittest 
import mock 
from boto.exception import S3ResponseError, S3DataError 
from mymodule import MyClass 

class TestS3(unittest.TestCase): 
    def setUp(self): 
     self.key = 'sample/bubket/key.txt' 
     self.myclass = MyClass() 

    def test_my_method(self): 
     exit_method = (
      'mymodule.' 
      'sys.' 
      'exit' 
     ) 
     s3_get_bucket = (
      'mymodule.' 
      'S3Connection.' 
      'get_bucket' 
     ) 


     with mock.patch(s3_get_bucket) as mocked_bucket, \ 
      mock.patch(exit_method) as mocked_exit: 
       mocked_bucket.side_effect = S3ResponseError 
       self.myclass.get_file(self.key) 
       mocked_exit.assert_called_once_with(3) 

그러나 어설 션이 실패한 경우, 도움이나 지침을 주시면 감사하겠습니다. MyClass.get_file()가 get_bucket()를 호출하지 않는 그것은 문제처럼 나에게 보이는

AssertionError: Expected to be called once. Called 0 times. 

Answer 1

이다. 그렇기 때문에 sys.exit()에 전화가 표시되지 않습니다. 모조가 get_key()이거나 get_bucket()으로 전화하십시오.