test(portfwd): assert ipv6 removals

Signed-off-by: Casey Quinn <casey.quinn@agyn.io>

Author: Casey Quinn

pkg/portfwd/forward_test.go

@@ -156,11 +156,11 @@ func TestForwarderIgnoreSkipsRemoval(t *testing.T) {
 			listenerKey := key(tc.protocol, local, remote)
 			tc.prepopulate(fw, listenerKey)
 			fw.OnEvent(t.Context(), dial, &api.Event{RemovedLocalPorts: []*api.IPPort{port}})
-			if tc.protocol == "tcp" {
+			if tc.protocol == "tcp" || tc.protocol == "tcp6" {
 				_, ok := fw.closableListeners.listeners[listenerKey]
 				assert.Assert(t, ok, "tcp listener %s should not be removed", listenerKey)
 			}
-			if tc.protocol == "udp" {
+			if tc.protocol == "udp" || tc.protocol == "udp6" {
 				_, ok := fw.closableListeners.udpListeners[listenerKey]
 				assert.Assert(t, ok, "udp listener %s should not be removed", listenerKey)
 			}