def METHOD_NAME(bucket):
  if bucket.name == 'bucket1':
    return [
        storage.Blob(
            'foo/file1.csv', storage.Bucket(mock_client, name='bucket1')),
        storage.Blob(
            'foo/file2.csv', storage.Bucket(mock_client, name='bucket1')),
        storage.Blob(
            'bar/file3.csv', storage.Bucket(mock_client, name='bucket1')),
    ]
  elif bucket.name == 'bucket2':
    return [
        storage.Blob(
            'foo/file1.csv', storage.Bucket(mock_client, name='bucket2')),
    ]
  else:
    raise ValueError(f'Unknown bucket: {bucket.name}')