github.com/vicanso/pike@v1.0.1-0.20210630235453-9099e041f6ec/cache/cache_test.go (about) 1 // MIT License 2 3 // Copyright (c) 2020 Tree Xie 4 5 // Permission is hereby granted, free of charge, to any person obtaining a copy 6 // of this software and associated documentation files (the "Software"), to deal 7 // in the Software without restriction, including without limitation the rights 8 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 // copies of the Software, and to permit persons to whom the Software is 10 // furnished to do so, subject to the following conditions: 11 12 // The above copyright notice and this permission notice shall be included in all 13 // copies or substantial portions of the Software. 14 15 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 // SOFTWARE. 22 23 package cache 24 25 import ( 26 "bytes" 27 "testing" 28 29 "github.com/stretchr/testify/assert" 30 "github.com/vicanso/pike/config" 31 ) 32 33 func TestConvertConfig(t *testing.T) { 34 assert := assert.New(t) 35 36 name := "cache-test" 37 size := 100 38 hitForPass := 60 39 configs := []config.CacheConfig{ 40 { 41 Name: name, 42 Size: size, 43 HitForPass: "1m", 44 }, 45 } 46 opts := convertConfigs(configs) 47 assert.Equal(1, len(opts)) 48 assert.Equal(name, opts[0].Name) 49 assert.Equal(size, opts[0].Size) 50 assert.Equal(hitForPass, opts[0].HitForPass) 51 } 52 53 func TestDefaultDispatcher(t *testing.T) { 54 assert := assert.New(t) 55 name := "test" 56 assert.Nil(GetDispatcher(name)) 57 ResetDispatchers([]config.CacheConfig{ 58 { 59 Name: name, 60 Size: 100, 61 HitForPass: "1m", 62 }, 63 }) 64 assert.NotNil(GetDispatcher(name)) 65 } 66 67 func TestUint32ToBytes(t *testing.T) { 68 assert := assert.New(t) 69 buf := uint32ToBytes(1) 70 value, err := readUint32ToInt(bytes.NewBuffer(buf)) 71 assert.Nil(err) 72 assert.Equal(1, value) 73 } 74 75 func TestUint64ToBytes(t *testing.T) { 76 assert := assert.New(t) 77 buf := uint64ToBytes(1) 78 value, err := readUint64ToInt64(bytes.NewBuffer(buf)) 79 assert.Nil(err) 80 assert.Equal(int64(1), value) 81 }