Files
tasks/common/archiver/s3store/query_parser_test.go
Alex ShtinandGitHub 8e8f1b0f7c Remove license header from every file (#7689)
## What changed?
<!-- Describe what has changed in this PR -->
Remove license header from every file. Because it is really hard to
follow in this PR here is the summary:
1. License header is removed from all `.go` and `.proto` files
:fireworks::fireworks:🎆.
2. `LICENSE` file in the root directory has only Temporal and Uber
copyrights.
3. 5 other `LICENSE` files added to the packages which have copyrights
different from Temporal and Uber: Datadog, Xargin, "Mat Ryer, Tyler
Bunnell and contributors".
4. `license_file` flag is removed from all code generation tools.
5. `copyright_file` flag is removed from `go:generate mockgen`
directive.
6. All copyright related targets are removed from `Makefile`.
7. Updated Temporal copyright year to 2025 everywhere.

## Why?
<!-- Tell your future self why have you made these changes -->
I double checked with legal department that it is not needed to have
license header in every file. One file per repo is enough. I put all
copyrights to the root `LICENSE` file and removed header from all other
files. Also updated tools and `Makefile`.
2025-05-01 18:50:21 -07:00

251 lines
5.7 KiB
Go

package s3store
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
"go.temporal.io/server/common/util"
)
type queryParserSuite struct {
*require.Assertions
suite.Suite
parser QueryParser
}
func TestQueryParserSuite(t *testing.T) {
suite.Run(t, new(queryParserSuite))
}
func (s *queryParserSuite) SetupTest() {
s.Assertions = require.New(s.T())
s.parser = NewQueryParser()
}
func (s *queryParserSuite) TestParseWorkflowIDAndWorkflowTypeName() {
testCases := []struct {
query string
expectErr bool
parsedQuery *parsedQuery
}{
{
query: "WorkflowId = \"random workflowID\"",
expectErr: false,
parsedQuery: &parsedQuery{
workflowID: util.Ptr("random workflowID"),
},
},
{
query: "WorkflowTypeName = \"random workflowTypeName\"",
expectErr: false,
parsedQuery: &parsedQuery{
workflowTypeName: util.Ptr("random workflowTypeName"),
},
},
{
query: "WorkflowId = \"random workflowID\" and WorkflowTypeName = \"random workflowTypeName\"",
expectErr: true,
},
{
query: "WorkflowId = \"random workflowID\" and WorkflowId = \"random workflowID\"",
expectErr: true,
},
{
query: "RunId = \"random runID\"",
expectErr: true,
},
{
query: "WorkflowId = 'random workflowID'",
expectErr: false,
parsedQuery: &parsedQuery{
workflowID: util.Ptr("random workflowID"),
},
},
{
query: "(WorkflowId = \"random workflowID\")",
expectErr: false,
parsedQuery: &parsedQuery{
workflowID: util.Ptr("random workflowID"),
},
},
{
query: "runId = random workflowID",
expectErr: true,
},
{
query: "WorkflowId = \"random workflowID\" or WorkflowId = \"another workflowID\"",
expectErr: true,
},
{
query: "WorkflowId = \"random workflowID\" or runId = \"random runID\"",
expectErr: true,
},
{
query: "workflowid = \"random workflowID\"",
expectErr: true,
},
{
query: "runId > \"random workflowID\"",
expectErr: true,
},
}
for _, tc := range testCases {
parsedQuery, err := s.parser.Parse(tc.query)
if tc.expectErr {
s.Error(err)
continue
}
s.NoError(err)
s.Equal(tc.parsedQuery.workflowID, parsedQuery.workflowID)
s.Equal(tc.parsedQuery.workflowTypeName, parsedQuery.workflowTypeName)
}
}
func (s *queryParserSuite) TestParsePrecision() {
commonQueryPart := "WorkflowId = \"random workflowID\" AND "
testCases := []struct {
query string
expectErr bool
parsedQuery *parsedQuery
}{
{
query: commonQueryPart + "CloseTime = 1000 and SearchPrecision = 'Day'",
expectErr: false,
parsedQuery: &parsedQuery{
searchPrecision: util.Ptr(PrecisionDay),
},
},
{
query: commonQueryPart + "CloseTime = 1000 and SearchPrecision = 'Hour'",
expectErr: false,
parsedQuery: &parsedQuery{
searchPrecision: util.Ptr(PrecisionHour),
},
},
{
query: commonQueryPart + "CloseTime = 1000 and SearchPrecision = 'Minute'",
expectErr: false,
parsedQuery: &parsedQuery{
searchPrecision: util.Ptr(PrecisionMinute),
},
},
{
query: commonQueryPart + "StartTime = 1000 and SearchPrecision = 'Second'",
expectErr: false,
parsedQuery: &parsedQuery{
searchPrecision: util.Ptr(PrecisionSecond),
},
},
{
query: commonQueryPart + "SearchPrecision = 'Second'",
expectErr: true,
},
{
query: commonQueryPart + "SearchPrecision = 'Invalid string'",
expectErr: true,
},
}
for _, tc := range testCases {
parsedQuery, err := s.parser.Parse(tc.query)
if tc.expectErr {
s.Error(err)
continue
}
s.NoError(err)
s.Equal(tc.parsedQuery.searchPrecision, parsedQuery.searchPrecision)
}
}
func (s *queryParserSuite) TestParseCloseTime() {
commonQueryPart := "WorkflowId = \"random workflowID\" AND SearchPrecision = 'Day' AND "
testCases := []struct {
query string
expectErr bool
parsedQuery *parsedQuery
}{
{
query: commonQueryPart + "CloseTime = 1000",
expectErr: false,
parsedQuery: &parsedQuery{
closeTime: util.Ptr(time.Unix(0, 1000).UTC()),
},
},
{
query: commonQueryPart + "CloseTime = \"2019-01-01T11:11:11Z\"",
expectErr: false,
parsedQuery: &parsedQuery{
closeTime: util.Ptr(time.Date(2019, 1, 1, 11, 11, 11, 0, time.UTC)),
},
},
{
query: commonQueryPart + "closeTime = 2000",
expectErr: true,
},
{
query: commonQueryPart + "CloseTime > \"2019-01-01 00:00:00\"",
expectErr: true,
},
}
for _, tc := range testCases {
parsedQuery, err := s.parser.Parse(tc.query)
if tc.expectErr {
s.Error(err)
continue
}
s.NoError(err)
s.Equal(tc.parsedQuery.closeTime, parsedQuery.closeTime)
}
}
func (s *queryParserSuite) TestParseStartTime() {
commonQueryPart := "WorkflowId = \"random workflowID\" AND SearchPrecision = 'Day' AND "
testCases := []struct {
query string
expectErr bool
parsedQuery *parsedQuery
}{
{
query: commonQueryPart + "StartTime = 1000",
expectErr: false,
parsedQuery: &parsedQuery{
startTime: util.Ptr(time.Unix(0, 1000)),
},
},
{
query: commonQueryPart + "StartTime = \"2019-01-01T11:11:11Z\"",
expectErr: false,
parsedQuery: &parsedQuery{
startTime: util.Ptr(time.Date(2019, 1, 1, 11, 11, 11, 0, time.UTC)),
},
},
{
query: commonQueryPart + "startTime = 2000",
expectErr: true,
},
{
query: commonQueryPart + "StartTime > \"2019-01-01 00:00:00\"",
expectErr: true,
},
}
for _, tc := range testCases {
parsedQuery, err := s.parser.Parse(tc.query)
if tc.expectErr {
s.Error(err)
continue
}
s.NoError(err)
s.Equal(tc.parsedQuery.closeTime, parsedQuery.closeTime)
}
}