Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,29 @@ | ||
| name: Go | ||
|
|
||
| on: | ||
| push: | ||
| pull_request: | ||
|
|
||
| jobs: | ||
|
|
||
| build: | ||
| runs-on: ubuntu-latest | ||
| steps: | ||
| - uses: actions/checkout@v2 | ||
| - name: Set up Go | ||
| uses: actions/setup-go@v2 | ||
| with: | ||
| go-version: 1.21 | ||
| - name: Build | ||
| run: go build -v ./... | ||
| - name: Test | ||
| run: go test -v -race $(go list ./... | grep -v /examples/) -coverprofile=coverage.txt -covermode=atomic | ||
| - name: CodeCov | ||
| uses: codecov/codecov-action@v1 | ||
| with: | ||
| # token: ${{ secrets.CODECOV_TOKEN }} # not required for public repos | ||
| files: ./coverage.txt | ||
| flags: unittests # optional | ||
| name: codecov-umbrella # optional | ||
| fail_ci_if_error: true # optional (default = false) | ||
| verbose: true # optional (default = false) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| package esnowflake | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "testing" | ||
| "time" | ||
|
|
||
| "github.com/stretchr/testify/assert" | ||
| ) | ||
|
|
||
| func TestConfigWithValidIP_ReturnsConfig(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| if config == nil { | ||
| t.Error("Expected non-nil config") | ||
| } | ||
| } | ||
|
|
||
| func TestConfigWithInvalidIP_Panics(t *testing.T) { | ||
| defer func() { | ||
| if r := recover(); r == nil { | ||
| t.Error("Expected panic for invalid IP") | ||
| } | ||
| }() | ||
| New("invalid_ip", 1, 2, 3) | ||
| } | ||
|
|
||
| func TestConfigWithValidIP_SetsMaskedIP(t *testing.T) { | ||
| config := New("192.168.1.2", 1, 2, 3) | ||
| expectedIP := []byte{168 ^ 1, 1 ^ 2, 2 ^ 3} | ||
| for i, b := range expectedIP { | ||
| if config.ip[i] != b { | ||
| t.Errorf("Expected masked IP byte %d to be %d, got %d", i, b, config.ip[i]) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func TestConfigWithValidIP_GetIp(t *testing.T) { | ||
| config := New("192.168.1.2", 1, 2, 3) | ||
| encode := config.GenerateByRandom() | ||
| ip := config.GetIP(encode) | ||
| assert.Equal(t, "xxx.168.1.2", ip) | ||
|
|
||
| } | ||
|
|
||
| func TestGenerateByRandom_ReturnsUniqueIDs(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| id1 := config.GenerateByRandom() | ||
| id2 := config.GenerateByRandom() | ||
| if id1 == id2 { | ||
| t.Errorf("Expected unique IDs, but got %s and %s", id1, id2) | ||
| } | ||
| } | ||
|
|
||
| func TestGenerateByRandom_HandlesPoolRefill(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| for i := 0; i < 100; i++ { | ||
| config.GenerateByRandom() | ||
| } | ||
| } | ||
|
|
||
| func TestGenerateByRandom_GetTime(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| encode := config.GenerateByRandom() | ||
| encodeTime := config.GetTime(encode) | ||
| fmt.Printf("time--------------->"+"%+v\n", encodeTime) | ||
| } | ||
|
|
||
| func TestGenerateByRandom_HandlesLocking(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| done := make(chan bool) | ||
| go func() { | ||
| config.GenerateByRandom() | ||
| done <- true | ||
| }() | ||
| select { | ||
| case <-done: | ||
| case <-time.After(1 * time.Second): | ||
| t.Error("GenerateByRandom did not return within 1 second") | ||
| } | ||
| } | ||
|
|
||
| func TestGenerateBySequence_ReturnsUniqueIDs(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| id1 := config.GenerateBySequence() | ||
| id2 := config.GenerateBySequence() | ||
| if id1 == id2 { | ||
| t.Errorf("Expected unique IDs, but got %s and %s", id1, id2) | ||
| } | ||
| } | ||
|
|
||
| func TestGenerateBySequence_HandlesSequenceOverflow(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| config.timestamp = time.Now().UnixNano() / 1000000 | ||
| config.sequence = sequenceMask | ||
| id := config.GenerateBySequence() | ||
| if id == "" { | ||
| t.Error("Expected non-empty ID on sequence overflow") | ||
| } | ||
| } | ||
|
|
||
| func TestGenerateBySequence_HandlesLocking(t *testing.T) { | ||
| config := New("192.168.1.1", 1, 2, 3) | ||
| done := make(chan bool) | ||
| go func() { | ||
| config.GenerateBySequence() | ||
| done <- true | ||
| }() | ||
| select { | ||
| case <-done: | ||
| case <-time.After(1 * time.Second): | ||
| t.Error("GenerateBySequence did not return within 1 second") | ||
| } | ||
| } | ||
|
|
||
| func BenchmarkRandomAndSequence(b *testing.B) { | ||
| obj := New("192.168.1.1", 1, 2, 3) | ||
| b.Run("Random", func(b *testing.B) { | ||
| for i := 0; i < b.N; i++ { | ||
| obj.GenerateByRandom() | ||
| } | ||
| }) | ||
| b.Run("Sequence", func(b *testing.B) { | ||
| for i := 0; i < b.N; i++ { | ||
| obj.GenerateBySequence() | ||
| } | ||
| }) | ||
| } | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,11 @@ | ||
| module github.com/ego-component/esnowflake | ||
|
|
||
| go 1.21.1 | ||
|
|
||
| require github.com/stretchr/testify v1.9.0 | ||
|
|
||
| require ( | ||
| github.com/davecgh/go-spew v1.1.1 // indirect | ||
| github.com/pmezard/go-difflib v1.0.0 // indirect | ||
| gopkg.in/yaml.v3 v3.0.1 // indirect | ||
| ) |