improve testing

Signed-off-by: Tim Vaillancourt <[email protected]>
timvaillancourt · May 11, 2024 · 977efa4 · 977efa4
1 parent e8aa18f
commit 977efa4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/go/streamlog/streamlog_test.go b/go/streamlog/streamlog_test.go
@@ -30,6 +30,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
 	"vitess.io/vitess/go/vt/servenv"
@@ -264,6 +265,20 @@ func TestFile(t *testing.T) {
 	}
 }
 
+func TestShouldSampleQuery(t *testing.T) {
+	queryLogSampleRate = -1
+	assert.False(t, shouldSampleQuery())
+
+	queryLogSampleRate = 0
+	assert.False(t, shouldSampleQuery())
+
+	queryLogSampleRate = 1.0
+	assert.True(t, shouldSampleQuery())
+
+	queryLogSampleRate = 100.0
+	assert.True(t, shouldSampleQuery())
+}
+
 func TestShouldEmitLog(t *testing.T) {
 	origQueryLogFilterTag := queryLogFilterTag
 	origQueryLogRowThreshold := queryLogRowThreshold