1 // Copyright 2016, 2017 Thales e-Security, Inc 2 // 3 // Permission is hereby granted, free of charge, to any person obtaining 4 // a copy of this software and associated documentation files (the 5 // "Software"), to deal in the Software without restriction, including 6 // without limitation the rights to use, copy, modify, merge, publish, 7 // distribute, sublicense, and/or sell copies of the Software, and to 8 // permit persons to whom the Software is furnished to do so, subject to 9 // the following conditions: 10 // 11 // The above copyright notice and this permission notice shall be 12 // included in all copies or substantial portions of the Software. 13 // 14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, 15 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF 16 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND 17 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE 18 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION 19 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION 20 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. 21 22 package crypto11 23 24 import ( 25 "testing" 26 27 "github.com/stretchr/testify/require" 28 ) 29 30 func TestRandomReader(t *testing.T) { 31 ctx, err := ConfigureFromFile("config") 32 require.NoError(t, err) 33 34 defer func() { 35 err = ctx.Close() 36 require.NoError(t, err) 37 }() 38 39 reader, err := ctx.NewRandomReader() 40 require.NoError(t, err) 41 42 var a [8192]byte 43 for _, size := range []int{1, 16, 32, 256, 347, 4096, 8192} { 44 n, err := reader.Read(a[:size]) 45 require.NoError(t, err) 46 require.Equal(t, size, n) 47 } 48 } 49