def test_flat_directory_with_one_file(self): FLAT_DIR_WITH_ONE_FILE_PATH = "::flat_dir_with_one_path::" RB_FILE_PATH = "::rb_file_path::" when(os).listdir(FLAT_DIR_WITH_ONE_FILE_PATH).thenReturn([RB_FILE_PATH]) when(os.path).isdir(FLAT_DIR_WITH_ONE_FILE_PATH).thenReturn(True) when(os.path).getsize(join(FLAT_DIR_WITH_ONE_FILE_PATH, RB_FILE_PATH)).thenReturn(789) assert_that(FilePathSizeCalculator.calculate_size(FLAT_DIR_WITH_ONE_FILE_PATH), equal_to(789))
import os from com.senpaidevs.katas.file_size_calculator import FilePathSizeCalculator if __name__ == '__main__': print FilePathSizeCalculator.calculate_size("/Users/agualis/PycharmProjects/Katas/example_dir")
def test_one_file_and_one_subdirectory_with_one_file(self): when(os).listdir(ROOT_DIR_PATH).thenReturn([SUBDIR_WITH_ONE_FILE_PATH, SUBDIR_WITH_TWO_FILES_PATH, TXT_FILE_PATH]) expected_size = HTML_FILE_SIZE + TXT_FILE_SIZE + DOC_FILE_SIZE + TXT_FILE_SIZE assert_that(FilePathSizeCalculator.calculate_size(ROOT_DIR_PATH), equal_to(expected_size))
def test_flat_directory_with_several_files(self): assert_that(FilePathSizeCalculator.calculate_size(join(ROOT_DIR_PATH, SUBDIR_WITH_TWO_FILES_PATH)), equal_to(TXT_FILE_SIZE + DOC_FILE_SIZE))
def test_empty_directory(self): EMPTY_PATH = "::empty_path::" when(os).listdir(EMPTY_PATH).thenReturn([]) when(os.path).isdir(EMPTY_PATH).thenReturn(True) assert_that(FilePathSizeCalculator.calculate_size(EMPTY_PATH), equal_to(0))