GoCollections/lang/atomic/Int32_test.go

/*
 * Copyright (c) 2022 tursom. All rights reserved.
 * Use of this source code is governed by a GPL-3
 * license that can be found in the LICENSE file.
 */

package atomic

import (
	"fmt"
	"sync"
	"testing"

	"github.com/tursom/GoCollections/util/time"
)

func TestInt32_SetBit(t *testing.T) {
	var i Int32
	for j := 0; j < 32; j++ {
		if i.SetBit(j, true) {
			t.Fail()
		}
		if i.Load() != 1<<j {
			t.Fail()
		}
		fmt.Println(i.Load())
		if !i.SetBit(j, false) {
			t.Fail()
		}
		if i.Load() != 0 {
			t.Fail()
		}
	}
}

func TestInt32_CompareAndSwapBit(t *testing.T) {
	var i Int32
	cond := sync.WaitGroup{}
	cond.Add(1)
	for k := 0; k < 128; k++ {
		id := k
		go func() {
			cond.Wait()
			for j := 0; j < 32; j++ {
				for !i.CompareAndSwapBit(j, false, true) {
				}
				for !i.CompareAndSwapBit(j, true, false) {
				}
			}
			fmt.Printf("%d finished\n", id)
		}()
	}
	cond.Done()
	time.Sleep(1 * time.Second)
	fmt.Println(i.Load())
}