cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

fix deals test

Browse Source
This commit is contained in:
Aarsh Shah 2021-09-20 10:04:21 +04:00 committed by Dirk McCormick
parent 3c37078a85
commit 189f5f8aae
2 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
itests/deals_anycid_test.go
View File

@ -150,6 +150,7 @@ func TestDealRetrieveByAnyCid(t *testing.T) {
Selector: shared.AllSelector(), Selector: shared.AllSelector(),
}}, }},
).Write(tmp) ).Write(tmp)
require.NoError(t, err)
require.NoError(t, tmp.Close()) require.NoError(t, tmp.Close())
require.NoError(t, rd.Close()) require.NoError(t, rd.Close())

12
itests/kit/deals.go
View File

@ -377,6 +377,7 @@ type RunConcurrentDealsOpts struct {
} }
func (dh *DealHarness) RunConcurrentDeals(opts RunConcurrentDealsOpts) { func (dh *DealHarness) RunConcurrentDeals(opts RunConcurrentDealsOpts) {
ctx := context.Background()
errgrp, _ := errgroup.WithContext(context.Background()) errgrp, _ := errgroup.WithContext(context.Background())
for i := 0; i < opts.N; i++ { for i := 0; i < opts.N; i++ {
i := i i := i
@ -402,7 +403,18 @@ func (dh *DealHarness) RunConcurrentDeals(opts RunConcurrentDealsOpts) {
dh.t.Logf("retrieving deal %d/%d", i, opts.N) dh.t.Logf("retrieving deal %d/%d", i, opts.N)
outPath := dh.PerformRetrieval(context.Background(), deal, res.Root, opts.CarExport) outPath := dh.PerformRetrieval(context.Background(), deal, res.Root, opts.CarExport)
if opts.CarExport {
f, err := os.Open(outPath)
require.NoError(dh.t, err)
actualFile := dh.ExtractFileFromCAR(ctx, f)
require.NoError(dh.t, f.Close())
AssertFilesEqual(dh.t, inPath, actualFile.Name())
} else {
AssertFilesEqual(dh.t, inPath, outPath) AssertFilesEqual(dh.t, inPath, outPath)
}
return nil return nil
}) })
} }