github.com/bhojpur/cache@v0.0.4/pkg/engine/cache_test.go (about)

     1  package engine
     2  
     3  // Copyright (c) 2018 Bhojpur Consulting Private Limited, India. All rights reserved.
     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
    13  // all 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
    21  // THE SOFTWARE.
    22  
    23  import (
    24  	"fmt"
    25  	"testing"
    26  
    27  	"github.com/stretchr/testify/require"
    28  
    29  	"github.com/bhojpur/cache/pkg/engine/ristretto"
    30  )
    31  
    32  func TestNewDefaultCacheImpl(t *testing.T) {
    33  	assertNullCache := func(t *testing.T, cache Cache) {
    34  		_, ok := cache.(*nullCache)
    35  		require.True(t, ok)
    36  	}
    37  
    38  	assertLFUCache := func(t *testing.T, cache Cache) {
    39  		_, ok := cache.(*ristretto.Cache)
    40  		require.True(t, ok)
    41  	}
    42  
    43  	assertLRUCache := func(t *testing.T, cache Cache) {
    44  		_, ok := cache.(*LRUCache)
    45  		require.True(t, ok)
    46  	}
    47  
    48  	tests := []struct {
    49  		cfg    *Config
    50  		verify func(t *testing.T, cache Cache)
    51  	}{
    52  		{&Config{MaxEntries: 0, MaxMemoryUsage: 0, LFU: false}, assertNullCache},
    53  		{&Config{MaxEntries: 0, MaxMemoryUsage: 0, LFU: true}, assertNullCache},
    54  		{&Config{MaxEntries: 100, MaxMemoryUsage: 0, LFU: false}, assertLRUCache},
    55  		{&Config{MaxEntries: 0, MaxMemoryUsage: 1000, LFU: false}, assertNullCache},
    56  		{&Config{MaxEntries: 100, MaxMemoryUsage: 1000, LFU: false}, assertLRUCache},
    57  		{&Config{MaxEntries: 100, MaxMemoryUsage: 0, LFU: true}, assertNullCache},
    58  		{&Config{MaxEntries: 100, MaxMemoryUsage: 1000, LFU: true}, assertLFUCache},
    59  		{&Config{MaxEntries: 0, MaxMemoryUsage: 1000, LFU: true}, assertNullCache},
    60  	}
    61  	for _, tt := range tests {
    62  		t.Run(fmt.Sprintf("%d.%d.%v", tt.cfg.MaxEntries, tt.cfg.MaxMemoryUsage, tt.cfg.LFU), func(t *testing.T) {
    63  			cache := NewDefaultCacheImpl(tt.cfg)
    64  			tt.verify(t, cache)
    65  		})
    66  	}
    67  }