Skip to content

Commit

Permalink
refactor(events): reorder events
Browse files Browse the repository at this point in the history
reorder validation events so all get record when transfer finishes
  • Loading branch information
hannahhoward committed May 11, 2022
1 parent e5f267b commit 8b01cdb
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 20 deletions.
35 changes: 18 additions & 17 deletions impl/receiving_requests.go
Original file line number Diff line number Diff line change
Expand Up @@ -250,23 +250,6 @@ func (m *manager) recordRejectedValidationEvents(chid datatransfer.ChannelID, re
func (m *manager) recordAcceptedValidationEvents(chst datatransfer.ChannelState, result datatransfer.ValidationResult) error {
chid := chst.ChannelID()

// pause or resume the request as neccesary
if result.LeaveRequestPaused(chst) {
if !chst.Status().IsResponderPaused() {
err := m.channels.PauseResponder(chid)
if err != nil {
return err
}
}
} else {
if chst.Status().IsResponderPaused() {
err := m.channels.ResumeResponder(chid)
if err != nil {
return err
}
}
}

// record the voucher result if present
if result.VoucherResult != nil {
err := m.channels.NewVoucherResult(chid, result.VoucherResult)
Expand All @@ -290,6 +273,24 @@ func (m *manager) recordAcceptedValidationEvents(chst datatransfer.ChannelState,
return err
}
}

// pause or resume the request as neccesary
if result.LeaveRequestPaused(chst) {
if !chst.Status().IsResponderPaused() {
err := m.channels.PauseResponder(chid)
if err != nil {
return err
}
}
} else {
if chst.Status().IsResponderPaused() {
err := m.channels.ResumeResponder(chid)
if err != nil {
return err
}
}
}

return nil
}

Expand Down
6 changes: 3 additions & 3 deletions impl/responding_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -356,9 +356,9 @@ func TestDataTransferResponding(t *testing.T) {
datatransfer.DataReceived,
datatransfer.DataLimitExceeded,
datatransfer.NewVoucher,
datatransfer.ResumeResponder,
datatransfer.NewVoucherResult,
datatransfer.SetDataLimit,
datatransfer.ResumeResponder,
},
configureValidator: func(sv *testutil.StubbedValidator) {
sv.ExpectSuccessPush()
Expand Down Expand Up @@ -455,9 +455,9 @@ func TestDataTransferResponding(t *testing.T) {
datatransfer.DataQueued,
datatransfer.DataLimitExceeded,
datatransfer.NewVoucher,
datatransfer.ResumeResponder,
datatransfer.NewVoucherResult,
datatransfer.SetDataLimit,
datatransfer.ResumeResponder,
},
configureValidator: func(sv *testutil.StubbedValidator) {
sv.ExpectSuccessPull()
Expand Down Expand Up @@ -505,9 +505,9 @@ func TestDataTransferResponding(t *testing.T) {
datatransfer.SetRequiresFinalization,
datatransfer.BeginFinalizing,
datatransfer.NewVoucher,
datatransfer.ResumeResponder,
datatransfer.NewVoucherResult,
datatransfer.SetRequiresFinalization,
datatransfer.ResumeResponder,
datatransfer.CleanupComplete,
},
configureValidator: func(sv *testutil.StubbedValidator) {
Expand Down

0 comments on commit 8b01cdb

Please sign in to comment.