@@ -2,7 +2,6 @@ package upnp
2
2
3
3
import (
4
4
"context"
5
- "errors"
6
5
"sync"
7
6
"testing"
8
7
"time"
@@ -55,26 +54,18 @@ func TestIGD(t *testing.T) {
55
54
}
56
55
t .Log ("Your external IP is:" , ip )
57
56
58
- // check that port 9001 is not forwarded
59
- res , err := d .IsForwardedTCP (9001 )
60
- if err != nil {
61
- t .Fatal (err )
62
- } else if res == true {
63
- t .Fatal (errors .New ("port 9001 is already forwarded" ))
64
- }
65
-
66
57
// forward a port
67
58
err = d .Forward (9001 , "upnp test" )
68
59
if err != nil {
69
60
t .Fatal (err )
70
61
}
71
62
72
63
// check that port 9001 is now forwarded
73
- res , err = d .IsForwardedTCP (9001 )
64
+ forwarded , err : = d .IsForwardedTCP (9001 )
74
65
if err != nil {
75
66
t .Fatal (err )
76
- } else if res == false {
77
- t .Fatal (errors . New ( "port 9001 was not reported as forwarded" ) )
67
+ } else if ! forwarded {
68
+ t .Fatal ("port 9001 was not reported as forwarded" )
78
69
}
79
70
80
71
// un-forward a port
@@ -84,11 +75,11 @@ func TestIGD(t *testing.T) {
84
75
}
85
76
86
77
// check that port 9001 is no longer forwarded
87
- res , err = d .IsForwardedTCP (9001 )
78
+ forwarded , err = d .IsForwardedTCP (9001 )
88
79
if err != nil {
89
80
t .Fatal (err )
90
- } else if res == true {
91
- t .Fatal (errors . New ( "port 9001 is still forwarded" ) )
81
+ } else if forwarded {
82
+ t .Fatal ("port 9001 should no longer be forwarded" )
92
83
}
93
84
94
85
// record router's location
0 commit comments