htlcswitch/test: extract failure code assertion to function

This commit is contained in:
Joost Jager committed 2019-08-08 11:04:15 +02:00
1 parent c3aa46e08b
commit 8681729561
2 files changed
+19 -24

No files matched your search

+1 -7
View File
@@ -1824,14 +1824,8 @@ func TestSwitchSendPayment(t *testing.T) {
select {
case err := <-errChan:
fErr, ok := err.(*ForwardingError)
if !ok {
t.Fatal("expected ForwardingError")
}
assertFailureCode(t, err, lnwire.CodeUnknownPaymentHash)
if _, ok := fErr.FailureMessage.(*lnwire.FailUnknownPaymentHash); !ok {
t.Fatalf("expected UnknownPaymentHash got %v", fErr)
}
case <-time.After(time.Second):
t.Fatal("err wasn't received")
}