vitess.io/vitess@v0.16.2/go/vt/sqlparser/truncate_query_test.go (about)

     1  package sqlparser
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  
     7  	"github.com/stretchr/testify/assert"
     8  )
     9  
    10  func TestTruncateQuery(t *testing.T) {
    11  	tests := []struct {
    12  		query string
    13  		max   int
    14  		want  string
    15  	}{
    16  		{
    17  			query: "select * from test where name = 'abc'",
    18  			max:   30,
    19  			want:  "select * from test [TRUNCATED]",
    20  		},
    21  		{
    22  			query: "select * from test where name = 'abc'",
    23  			max:   1005,
    24  			want:  "select * from test where name = 'abc'",
    25  		},
    26  	}
    27  	for _, tt := range tests {
    28  		t.Run(fmt.Sprintf("%s-%d", tt.query, tt.max), func(t *testing.T) {
    29  			assert.Equalf(t, tt.want, truncateQuery(tt.query, tt.max), "truncateQuery(%v, %v)", tt.query, tt.max)
    30  		})
    31  	}
    32  }