add tests for 64bit id probabilistic sampling

jaegertracing · Mar 2, 2020 · d117971 · d117971
1 parent 23de8b9
commit d117971
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sampler_test.go b/sampler_test.go
@@ -85,6 +85,15 @@ func TestProbabilisticSampler(t *testing.T) {
 	assert.Equal(t, testProbabilisticExpectedTags, tags)
 	sampled, tags = sampler.IsSampled(TraceID{Low: testMaxID - 20}, testOperationName)
 	assert.True(t, sampled)
+
+	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))