mock: Make it possible to unfail sectors

This commit is contained in:
Łukasz Magiera 2020-08-03 20:49:04 +02:00
parent ed251d9f82
commit 3cab915fd2

View File

@ -236,7 +236,7 @@ func (mgr *SectorMgr) SealCommit2(ctx context.Context, sid abi.SectorID, phase1O
// Test Instrumentation Methods // Test Instrumentation Methods
func (mgr *SectorMgr) FailSector(sid abi.SectorID) error { func (mgr *SectorMgr) MarkFailed(sid abi.SectorID, failed bool) error {
mgr.lk.Lock() mgr.lk.Lock()
defer mgr.lk.Unlock() defer mgr.lk.Unlock()
ss, ok := mgr.sectors[sid] ss, ok := mgr.sectors[sid]
@ -244,7 +244,7 @@ func (mgr *SectorMgr) FailSector(sid abi.SectorID) error {
return fmt.Errorf("no such sector in storage") return fmt.Errorf("no such sector in storage")
} }
ss.failed = true ss.failed = failed
return nil return nil
} }