如何模拟 google.cloud.storage.blob.Blob 对象列表？

Question

我有一个 function，它接收一堆 google.cloud.storage.blob.Blob 对象列表，按创建日期解析，我想对其进行单元测试。 我怎么把go转成mocking呢？ 我考虑过打补丁，但我意识到这更适用于正在测试的 function 内的呼叫。 我如何模拟这些 blob 的输入？ 这是我要测试的 function：

def bucket_parser(objects,start_time,end_time):
    file_count = 0
    latest_timestamp = datetime.datetime.strptime('0000','%H%M')
    for file in objects:
        if file.time_created.time().replace(tzinfo=None) <= end_time and file.time_created.time().replace(tzinfo=None) >= start_time:
            file_count += 1
        if file.time_created.replace(tzinfo=None) > latest_timestamp.replace(tzinfo=None):
            latest_timestamp = file.time_created
    return file_count , latest_timestamp.time()

其中对象属于 <class 'google.cloud.storage.blob.Blob'>，开始和结束时间属于 <class 'datetime.time'>。 我 go 如何使用 mocking 对这个 function 进行单元测试？ 我感到卡住了，因为我不确定如何 go 了解它。 谢谢。

编辑：这是从桶中获取 blob 的代码，并将它们传递到上面的 function 中：

def bucket_scraper(bucket_name,bucket_prefix):
    '''
    Input: Bucket , Bucket Prefix
    Output: All objects/files in specified bucket prefix
    '''
    client = storage.Client()
    prefix_files = list(client.list_blobs(bucket_name,max_results = 700,page_token=None,prefix=bucket_prefix,delimiter='/'))
    return prefix_files

返回的 prefix_files 是进入 bucket_parser function 的内容。

Answer 1

这是 java 中关于如何创建模拟 blob 列表的方法，可能会有所帮助（这是在测试环境中）

private List<Blob> getMockBlobList(int range){
        List<Blob> blobs = new ArrayList<>();
        for(int m = 0; m < range; m++){
            Blob blob = mock(Blob.class);
            when(mockReturnBlob.getName()).thenReturn("test-" + m +".pdf");
            //Add more returns to this blob
            blobs.add(blob);
        }

        return blobs;
    }

一种使用方法是

PageImpl<Blob> expectedBlobPage = new PageImpl<>(null, "c", List.of(getMockBlobList()));