gocryptotrader/exchanges/nonce/nonce_test.go

package nonce

import (
	"testing"
	"time"
)

func TestInc(t *testing.T) {
	var nonce Nonce
	nonce.Set(1)
	nonce.Inc()
	expected := int64(2)
	result := nonce.Get()
	if result != expected {
		t.Errorf("Test failed. Expected %d got %d", expected, result)
	}
}

func TestGet(t *testing.T) {
	var nonce Nonce
	nonce.Set(112321313)
	expected := int64(112321313)
	result := nonce.Get()
	if expected != result {
		t.Errorf("Test failed. Expected %d got %d", expected, result)
	}
}

func TestGetInc(t *testing.T) {
	var nonce Nonce
	nonce.Set(1)
	expected := int64(2)
	result := nonce.GetInc()
	if expected != result {
		t.Errorf("Test failed. Expected %d got %d", expected, result)
	}
}

func TestSet(t *testing.T) {
	var nonce Nonce
	nonce.Set(1)
	expected := int64(1)
	result := nonce.Get()
	if expected != result {
		t.Errorf("Test failed. Expected %d got %d", expected, result)
	}
}

func TestString(t *testing.T) {
	var nonce Nonce
	nonce.Set(12312313131)
	expected := "12312313131"
	result := nonce.String()
	if expected != result {
		t.Errorf("Test failed. Expected %s got %s", expected, result)
	}
}

func TestNonceConcurrency(t *testing.T) {
	var nonce Nonce
	nonce.Set(12312)

	for i := 0; i < 1000; i++ {
		go nonce.Inc()
	}

	// Allow sufficient time for all routines to finish
	time.Sleep(time.Second)

	result := nonce.Get()
	expected := int64(12312 + 1000)
	if expected != result {
		t.Errorf("Test failed. Expected %d got %d", expected, result)
	}
}