def test_cleanup_sub_folders_with_directories(self, mock_path, mock_os):
mock_os.listdir.return_value = ['1', '2', '3', '4', '5', '6', '7', '8']
mock_path.isfile.return_value = True
backup = Backup(config)
backup.cleanup_sub_folders()
self.assertTrue(mock_os.listdir.called)
self.assertTrue(mock_os.remove.called)
def test_cleanup_sub_folders_with_directories_less_than_required(self, mock_path, mock_os):
mock_os.listdir.return_value = ['123123123', '234234234']
mock_path.isfile.return_value = True
backup = Backup(config)
backup.cleanup_sub_folders()
self.assertTrue(mock_os.listdir.called)
self.assertFalse(mock_os.remove.called)
