github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/x/unsafe/bytes_test.go (about) 1 // Copyright (c) 2019 Uber Technologies, Inc. 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining a copy 4 // of this software and associated documentation files (the "Software"), to deal 5 // in the Software without restriction, including without limitation the rights 6 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 7 // copies of the Software, and to permit persons to whom the Software is 8 // furnished to do so, subject to the following conditions: 9 // 10 // The above copyright notice and this permission notice shall be included in 11 // all copies or substantial portions of the Software. 12 // 13 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 14 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 15 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 16 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 17 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 18 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 19 // THE SOFTWARE. 20 21 package unsafe 22 23 import ( 24 "bytes" 25 "testing" 26 27 "github.com/stretchr/testify/require" 28 ) 29 30 func TestWithStringSmallString(t *testing.T) { 31 str := []byte("foobarbaz") 32 validateWithString(t, str) 33 } 34 35 func TestWithStringLargeString(t *testing.T) { 36 var buf bytes.Buffer 37 for i := 0; i < 65536; i++ { 38 buf.WriteByte(byte(i % 256)) 39 } 40 str := buf.Bytes() 41 validateWithString(t, str) 42 } 43 44 func TestWithStringAndArgSmallString(t *testing.T) { 45 str := []byte("foobarbaz") 46 validateWithStringAndArg(t, str) 47 } 48 49 func TestWithStringAndArgLargeString(t *testing.T) { 50 var buf bytes.Buffer 51 for i := 0; i < 65536; i++ { 52 buf.WriteByte(byte(i % 256)) 53 } 54 str := buf.Bytes() 55 validateWithStringAndArg(t, str) 56 } 57 58 var withStringBenchSink string 59 60 func BenchmarkWithString(b *testing.B) { 61 str := []byte("foobarbaz") 62 WithString(str, func(s string) { 63 withStringBenchSink = s 64 }) 65 } 66 67 func validateWithString(t *testing.T, b []byte) { 68 WithString(b, func(str string) { 69 require.Equal(t, []byte(str), []byte(b)) 70 require.Equal(t, len(str), len(b)) 71 }) 72 } 73 74 func validateWithStringAndArg(t *testing.T, b []byte) { 75 WithStringAndArg(b, "cat", func(str string, arg interface{}) { 76 var buf bytes.Buffer 77 buf.WriteString(str) 78 buf.WriteString(arg.(string)) 79 require.Equal(t, string(b)+"cat", buf.String()) 80 }) 81 }