From dbc62aac93542dfc6571776677be7a10d476ff0e Mon Sep 17 00:00:00 2001 From: MaysWind Date: Mon, 25 Sep 2023 23:37:16 +0800 Subject: [PATCH] fix unit test failure --- pkg/uuid/internal_generator_test.go | 26 ++++++++++++++++++++------ 1 file changed, 20 insertions(+), 6 deletions(-) diff --git a/pkg/uuid/internal_generator_test.go b/pkg/uuid/internal_generator_test.go index 95f4be90..c499bddb 100644 --- a/pkg/uuid/internal_generator_test.go +++ b/pkg/uuid/internal_generator_test.go @@ -61,6 +61,7 @@ func TestGenerateUuid_MultiType(t *testing.T) { func TestGenerateUuid_2000TimesIn2Seconds(t *testing.T) { generator, _ := NewInternalUuidGenerator(&settings.Config{UuidServerId: 2}) + firstGeneratedTime := int64(0) for i := 0; i < 1000; i++ { generationStartUnixTime := time.Now().Unix() @@ -70,13 +71,20 @@ func TestGenerateUuid_2000TimesIn2Seconds(t *testing.T) { uuidInfo := generator.parseInternalUuidInfo(uuid) if generationStartUnixTime == generationEndUnixTime { - assert.Equal(t, uint32(generationStartUnixTime), uuidInfo.UnixTime) - } + if i == 0 { + firstGeneratedTime = time.Now().Unix() + } - assert.Equal(t, uint32(i), uuidInfo.SequentialId) + assert.Equal(t, uint32(generationStartUnixTime), uuidInfo.UnixTime) + + if generationStartUnixTime == firstGeneratedTime { + assert.Equal(t, uint32(i), uuidInfo.SequentialId) + } + } } time.Sleep(1 * time.Second) + firstGeneratedTime = int64(0) for i := 0; i < 1000; i++ { generationStartUnixTime := time.Now().Unix() @@ -86,10 +94,16 @@ func TestGenerateUuid_2000TimesIn2Seconds(t *testing.T) { uuidInfo := generator.parseInternalUuidInfo(uuid) if generationStartUnixTime == generationEndUnixTime { - assert.Equal(t, uint32(generationStartUnixTime), uuidInfo.UnixTime) - } + if i == 0 { + firstGeneratedTime = time.Now().Unix() + } - assert.Equal(t, uint32(i), uuidInfo.SequentialId) + assert.Equal(t, uint32(generationStartUnixTime), uuidInfo.UnixTime) + + if generationStartUnixTime == firstGeneratedTime { + assert.Equal(t, uint32(i), uuidInfo.SequentialId) + } + } } }