github.com/gohugoio/hugo@v0.88.1/navigation/menu_cache_test.go (about)

     1  // Copyright 2021 The Hugo Authors. All rights reserved.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  // http://www.apache.org/licenses/LICENSE-2.0
     7  //
     8  // Unless required by applicable law or agreed to in writing, software
     9  // distributed under the License is distributed on an "AS IS" BASIS,
    10  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package navigation
    15  
    16  import (
    17  	"sync"
    18  	"sync/atomic"
    19  	"testing"
    20  
    21  	qt "github.com/frankban/quicktest"
    22  )
    23  
    24  func createSortTestMenu(num int) Menu {
    25  	menu := make(Menu, num)
    26  	for i := 0; i < num; i++ {
    27  		m := &MenuEntry{}
    28  		menu[i] = m
    29  	}
    30  	return menu
    31  }
    32  
    33  func TestMenuCache(t *testing.T) {
    34  	t.Parallel()
    35  	c := qt.New(t)
    36  	c1 := newMenuCache()
    37  
    38  	changeFirst := func(m Menu) {
    39  		m[0].title = "changed"
    40  	}
    41  
    42  	var o1 uint64
    43  	var o2 uint64
    44  
    45  	var wg sync.WaitGroup
    46  
    47  	var l1 sync.Mutex
    48  	var l2 sync.Mutex
    49  
    50  	var testMenuSets []Menu
    51  
    52  	for i := 0; i < 50; i++ {
    53  		testMenuSets = append(testMenuSets, createSortTestMenu(i+1))
    54  	}
    55  
    56  	for j := 0; j < 100; j++ {
    57  		wg.Add(1)
    58  		go func() {
    59  			defer wg.Done()
    60  			for k, menu := range testMenuSets {
    61  				l1.Lock()
    62  				m, ca := c1.get("k1", nil, menu)
    63  				c.Assert(ca, qt.Equals, !atomic.CompareAndSwapUint64(&o1, uint64(k), uint64(k+1)))
    64  				l1.Unlock()
    65  				m2, c2 := c1.get("k1", nil, m)
    66  				c.Assert(c2, qt.Equals, true)
    67  				c.Assert(menuEqual(m, m2), qt.Equals, true)
    68  				c.Assert(menuEqual(m, menu), qt.Equals, true)
    69  				c.Assert(m, qt.Not(qt.IsNil))
    70  
    71  				l2.Lock()
    72  				m3, c3 := c1.get("k2", changeFirst, menu)
    73  				c.Assert(c3, qt.Equals, !atomic.CompareAndSwapUint64(&o2, uint64(k), uint64(k+1)))
    74  				l2.Unlock()
    75  				c.Assert(m3, qt.Not(qt.IsNil))
    76  				c.Assert("changed", qt.Equals, m3[0].title)
    77  			}
    78  		}()
    79  	}
    80  	wg.Wait()
    81  }