common_test.go

// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
// 🤖 Github Repository: https://github.com/gofiber/fiber
// 📌 API Documentation: https://docs.gofiber.io

package utils

import (
	"crypto/rand"
	"fmt"
	"testing"
)

func Test_FunctionName(t *testing.T) {
	t.Parallel()
	AssertEqual(t, "github.com/gofiber/fiber/v2/utils.Test_UUID", FunctionName(Test_UUID))

	AssertEqual(t, "github.com/gofiber/fiber/v2/utils.Test_FunctionName.func1", FunctionName(func() {}))

	dummyint := 20
	AssertEqual(t, "int", FunctionName(dummyint))
}

func Test_UUID(t *testing.T) {
	t.Parallel()
	res := UUID()
	AssertEqual(t, 36, len(res))
	AssertEqual(t, true, res != emptyUUID)
}

func Test_UUID_Concurrency(t *testing.T) {
	t.Parallel()
	iterations := 1000
	var res string
	ch := make(chan string, iterations)
	results := make(map[string]string)
	for i := 0; i < iterations; i++ {
		go func() {
			ch <- UUID()
		}()
	}
	for i := 0; i < iterations; i++ {
		res = <-ch
		results[res] = res
	}
	AssertEqual(t, iterations, len(results))
}

func Test_UUIDv4(t *testing.T) {
	t.Parallel()
	res := UUIDv4()
	AssertEqual(t, 36, len(res))
	AssertEqual(t, true, res != emptyUUID)
}

func Test_UUIDv4_Concurrency(t *testing.T) {
	t.Parallel()
	iterations := 1000
	var res string
	ch := make(chan string, iterations)
	results := make(map[string]string)
	for i := 0; i < iterations; i++ {
		go func() {
			ch <- UUIDv4()
		}()
	}
	for i := 0; i < iterations; i++ {
		res = <-ch
		results[res] = res
	}
	AssertEqual(t, iterations, len(results))
}

// go test -v -run=^$ -bench=Benchmark_UUID -benchmem -count=2

func Benchmark_UUID(b *testing.B) {
	var res string
	b.Run("fiber", func(b *testing.B) {
		for n := 0; n < b.N; n++ {
			res = UUID()
		}
		AssertEqual(b, 36, len(res))
	})
	b.Run("default", func(b *testing.B) {
		rnd := make([]byte, 16)
		_, err := rand.Read(rnd)
		AssertEqual(b, nil, err)
		for n := 0; n < b.N; n++ {
			res = fmt.Sprintf("%x-%x-%x-%x-%x", rnd[0:4], rnd[4:6], rnd[6:8], rnd[8:10], rnd[10:])
		}
		AssertEqual(b, 36, len(res))
	})
}

func Test_ConvertToBytes(t *testing.T) {
	t.Parallel()
	AssertEqual(t, 0, ConvertToBytes(""))
	AssertEqual(t, 42, ConvertToBytes("42"))
	AssertEqual(t, 42, ConvertToBytes("42b"))
	AssertEqual(t, 42, ConvertToBytes("42B"))
	AssertEqual(t, 42, ConvertToBytes("42 b"))
	AssertEqual(t, 42, ConvertToBytes("42 B"))

	AssertEqual(t, 42*1000, ConvertToBytes("42k"))
	AssertEqual(t, 42*1000, ConvertToBytes("42K"))
	AssertEqual(t, 42*1000, ConvertToBytes("42kb"))
	AssertEqual(t, 42*1000, ConvertToBytes("42KB"))
	AssertEqual(t, 42*1000, ConvertToBytes("42 kb"))
	AssertEqual(t, 42*1000, ConvertToBytes("42 KB"))

	AssertEqual(t, 42*1000000, ConvertToBytes("42M"))
	AssertEqual(t, int(42.5*1000000), ConvertToBytes("42.5MB"))
	AssertEqual(t, 42*1000000000, ConvertToBytes("42G"))

	AssertEqual(t, 0, ConvertToBytes("string"))
	AssertEqual(t, 0, ConvertToBytes("MB"))
}

// go test -v -run=^$ -bench=Benchmark_ConvertToBytes -benchmem -count=2
func Benchmark_ConvertToBytes(b *testing.B) {
	var res int
	b.Run("fiber", func(b *testing.B) {
		for n := 0; n < b.N; n++ {
			res = ConvertToBytes("42B")
		}
		AssertEqual(b, 42, res)
	})
}