はのboto例外にモックside_effectを割り当て、呼び出し

Question

私は、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) 

私はALO 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) 

ただし、アサーションは失敗しています。ヘルプやガイダンスは高く評価されます。

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

Answer 1

問題はMyClass.get_file()がget_bucket()を呼び出さないということであるようにそれは私には見えます。だからあなたはsys.exit()への呼び出しを見ていないのです。モックget_key()またはget_bucket()に電話してください。