From: Josh Bleecher Snyder <josharian@gmail.com>
Date: Fri, 5 Nov 2021 22:34:20 +0000 (-0700)
Subject: net/netip: add tests for Addr.AsSlice
X-Git-Tag: go1.18beta1~475
X-Git-Url: http://www.git.cypherpunks.su/?a=commitdiff_plain;h=3b7e376df87fa5255c7aa58d5719593b314338fd;p=gostls13.git

net/netip: add tests for Addr.AsSlice

Change-Id: Ib88dd101b3bbdf4d2bfd79838994cfadef1b604d
Reviewed-on: https://go-review.googlesource.com/c/go/+/361915
Trust: Josh Bleecher Snyder <josharian@gmail.com>
Run-TryBot: Josh Bleecher Snyder <josharian@gmail.com>
Reviewed-by: Brad Fitzpatrick <bradfitz@golang.org>
TryBot-Result: Go Bot <gobot@golang.org>
---

diff --git a/src/net/netip/netip_test.go b/src/net/netip/netip_test.go
index 63af853cb3..a6327f0dea 100644
--- a/src/net/netip/netip_test.go
+++ b/src/net/netip/netip_test.go
@@ -1889,6 +1889,24 @@ func TestInvalidAddrPortString(t *testing.T) {
 	}
 }
 
+func TestAsSlice(t *testing.T) {
+	tests := []struct {
+		in   Addr
+		want []byte
+	}{
+		{in: Addr{}, want: nil},
+		{in: mustIP("1.2.3.4"), want: []byte{1, 2, 3, 4}},
+		{in: mustIP("ffff::1"), want: []byte{0xff, 0xff, 15: 1}},
+	}
+
+	for _, test := range tests {
+		got := test.in.AsSlice()
+		if !bytes.Equal(got, test.want) {
+			t.Errorf("%v.AsSlice() = %v want %v", test.in, got, test.want)
+		}
+	}
+}
+
 var sink16 [16]byte
 
 func BenchmarkAs16(b *testing.B) {