add tests for 64bit id probabilistic sampling

Signed-off-by: Sokolov Yura <funny.falcon@gmail.com>
jaegertracing · Mar 2, 2020 · 00ac23a · 00ac23a
1 parent 23de8b9
commit 00ac23a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/sampler_test.go b/sampler_test.go
@@ -86,6 +86,16 @@ func TestProbabilisticSampler(t *testing.T) {
 	sampled, tags = sampler.IsSampled(TraceID{Low: testMaxID - 20}, testOperationName)
 	assert.True(t, sampled)
 	assert.Equal(t, testProbabilisticExpectedTags, tags)
+
+	t.Run("test_64bit_id", func(t *testing.T) {
+		sampled, tags := sampler.IsSampled(TraceID{Low: (testMaxID + 10) | 1<<63}, testOperationName)
+		assert.False(t, sampled)
+		assert.Equal(t, testProbabilisticExpectedTags, tags)
+		sampled, tags = sampler.IsSampled(TraceID{Low: (testMaxID - 20) | 1<<63}, testOperationName)
+		assert.True(t, sampled)
+		assert.Equal(t, testProbabilisticExpectedTags, tags)
+	})
+
 	sampler2, _ := NewProbabilisticSampler(0.5)
 	assert.True(t, sampler.Equal(sampler2))
 	assert.False(t, sampler.Equal(NewConstSampler(true)))