make sha256 test in crypto/sha256 package

golang · Feb 13, 2025 · 68ee1cb · 68ee1cb
1 parent d3d5a3b
commit 68ee1cb
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 18 deletions.
diff --git a/src/crypto/sha256/sha256_test.go b/src/crypto/sha256/sha256_test.go
@@ -391,3 +391,17 @@ func BenchmarkHash1K(b *testing.B) {
 func BenchmarkHash8K(b *testing.B) {
 	benchmarkSize(b, 8192)
 }
+
+func TestAllocatonsWithTypeAsserts(t *testing.T) {
+	cryptotest.SkipTestAllocations(t)
+	allocs := testing.AllocsPerRun(100, func() {
+		h := New()
+		h.Write([]byte{1, 2, 3})
+		marshaled, _ := h.(encoding.BinaryMarshaler).MarshalBinary()
+		marshaled, _ = h.(encoding.BinaryAppender).AppendBinary(marshaled[:0])
+		h.(encoding.BinaryUnmarshaler).UnmarshalBinary(marshaled)
+	})
+	if allocs != 0 {
+		t.Fatalf("allocs = %v; want = 0", allocs)
+	}
+}
diff --git a/test/escape_iface_with_devirt_type_assertions.go b/test/escape_iface_with_devirt_type_assertions.go
@@ -6,13 +6,6 @@
 
 package escape
 
-import (
-	"crypto/sha256"
-	"encoding"
-	"hash"
-	"io"
-)
-
 type M interface{ M() }
 
 type A interface{ A() }
@@ -138,14 +131,3 @@ func testInvalidAsserts() {
 		a.(any).(M).(*Impl).M() // ERROR "inlining"
 	}
 }
-
-func testSha256() {
-	h := sha256.New()                                   // ERROR "inlining call" "does not escape"
-	h.Write(nil)                                        // ERROR "devirtualizing"
-	h.(io.Writer).Write(nil)                            // ERROR "devirtualizing"
-	h.(hash.Hash).Write(nil)                            // ERROR "devirtualizing"
-	h.(encoding.BinaryUnmarshaler).UnmarshalBinary(nil) // ERROR "devirtualizing"
-
-	h2 := sha256.New() // ERROR "escapes" "inlining call"
-	h2.(M).M()         // this will panic
-}