blob: 931b43c6a2374511055869098b38e83658317dc0 [file] [log] [blame]
// Copyright 2018 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package builtins
import (
"testing"
"go.starlark.net/starlark"
. "github.com/smartystreets/goconvey/convey"
. "go.chromium.org/luci/common/testing/assertions"
)
const regexpTest = `
def assert(a, b):
if a != b:
fail("%s != %s" % (a, b))
def test():
groups = submatches(r'a (\d+) (\d+)', 'a 123 456 tail')
assert(groups, ("a 123 456", "123", "456"))
groups = submatches(r'a (\d+) (\d+)', 'a huh 456')
assert(groups, ())
groups = submatches('.*', 'a 123 456 tail')
assert(groups, ("a 123 456 tail",))
groups = submatches('', 'zzz')
assert(groups, ('',)) # empty match (not the same as no match at all)
test()
`
func TestRegexp(t *testing.T) {
t.Parallel()
submatches := RegexpMatcher("submatches")
runScript := func(code string) error {
_, err := starlark.ExecFile(&starlark.Thread{}, "main", code, starlark.StringDict{
"submatches": submatches,
"fail": Fail, // for 'assert'
})
return err
}
Convey("Success", t, func() {
So(runScript(regexpTest), ShouldBeNil)
})
Convey("Fails", t, func() {
So(runScript(`submatches('(((', '')`), ShouldErrLike, "error parsing regexp")
})
}