단위 테스트를 위해 입력 dir을 조롱하는 방법? 광산의 이전 질문에 관련

Question

: 나는

import threedi_utils 

@mock.patch('threedi_utils.files.os.walk') 
def test_get_dir_size_can_get_dir_size(self, mock_walk): 
    mock_walk.return_value(5000) 
    size = threedi_utils.files.get_dir_size(self.test_path) 
    self.assertEqual(size, 5000) 

그러나 mock_walk.return_value(5000)없이가는 os.walk 기능을 조롱해야 이해 것과 그래서

def get_dir_size(dir_path): 
    """Determine the size of a dir. 

    This function also takes into account the allocated size of 
    directories (4096 bytes). 

    Note: The function may crash on symlinks with something like: 
    OSError: [Errno 40] Too many levels of symbolic links 

    :param dir_path (str): path to the directory 
    :return: size in bytes. 
    """ 
    tot_size = 0 
    for (root_path, dirnames, filenames) in os.walk(dir_path): 
     for f in filenames: 
      fpath = os.path.join(root_path, f) 
      tot_size += os.path.getsize(fpath) 
     tot_size += os.path.getsize(root_path) 
    return tot_size 

: How to unit test a method that calculates the size of a dir? 내가 단위 테스트에이 기능을 원하는 내 검사 실패로 인한 영향

Traceback (most recent call last): 
    File "/home/vagrant/.buildout/eggs/mock-1.3.0-py2.7.egg/mock/mock.py", line 1305, in patched 
    return func(*args, **keywargs) 
    File "/srv/lib/threedi_utils/tests/test_files.py", line 55, in test_get_dir_size_can_get_dir_size 
    self.assertEqual(size, 5000) 
AssertionError: 0 != 5000 

나는 무엇입니까? ssing?

Answer 1

좋아, 나는 잘못된 길로 향하고 있었다. os.path.getsize() 메서드를 조롱 했어야합니다. 또한 반환 값은 다음과 같이 제공되어야합니다. .return_value = 5000

@mock.patch('threedi_utils.files.os.path.getsize') 
def test_get_dir_size_can_get_dir_size(self, mock_size): 
    mock_size.return_value = 50 
    size = threedi_utils.files.get_dir_size(self.test_path) 
    self.assertEqual(size, 750) 
    self.tearDown()