mirror of
https://github.com/restic/restic.git
synced 2024-08-20 19:33:23 +02:00
283 lines
6.5 KiB
Go
283 lines
6.5 KiB
Go
|
// Copyright 2015 The Go Authors. All rights reserved.
|
|||
|
// Use of this source code is governed by a BSD-style
|
|||
|
// license that can be found in the LICENSE file.
|
|||
|
|
|||
|
package runes
|
|||
|
|
|||
|
import (
|
|||
|
"strings"
|
|||
|
"testing"
|
|||
|
"unicode"
|
|||
|
|
|||
|
"golang.org/x/text/cases"
|
|||
|
"golang.org/x/text/language"
|
|||
|
"golang.org/x/text/transform"
|
|||
|
)
|
|||
|
|
|||
|
var (
|
|||
|
toUpper = cases.Upper(language.Und)
|
|||
|
toLower = cases.Lower(language.Und)
|
|||
|
)
|
|||
|
|
|||
|
type spanformer interface {
|
|||
|
transform.SpanningTransformer
|
|||
|
}
|
|||
|
|
|||
|
func TestPredicate(t *testing.T) {
|
|||
|
testConditional(t, func(rt *unicode.RangeTable, t, f spanformer) spanformer {
|
|||
|
return If(Predicate(func(r rune) bool {
|
|||
|
return unicode.Is(rt, r)
|
|||
|
}), t, f)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
func TestIn(t *testing.T) {
|
|||
|
testConditional(t, func(rt *unicode.RangeTable, t, f spanformer) spanformer {
|
|||
|
return If(In(rt), t, f)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
func TestNotIn(t *testing.T) {
|
|||
|
testConditional(t, func(rt *unicode.RangeTable, t, f spanformer) spanformer {
|
|||
|
return If(NotIn(rt), f, t)
|
|||
|
})
|
|||
|
}
|
|||
|
|
|||
|
func testConditional(t *testing.T, f func(rt *unicode.RangeTable, t, f spanformer) spanformer) {
|
|||
|
lower := f(unicode.Latin, toLower, toLower)
|
|||
|
|
|||
|
for i, tt := range []transformTest{{
|
|||
|
desc: "empty",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "",
|
|||
|
out: "",
|
|||
|
outFull: "",
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "small",
|
|||
|
szDst: 1,
|
|||
|
atEOF: true,
|
|||
|
in: "B",
|
|||
|
out: "b",
|
|||
|
outFull: "b",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "short dst",
|
|||
|
szDst: 2,
|
|||
|
atEOF: true,
|
|||
|
in: "AAA",
|
|||
|
out: "aa",
|
|||
|
outFull: "aaa",
|
|||
|
err: transform.ErrShortDst,
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "short dst writing error",
|
|||
|
szDst: 1,
|
|||
|
atEOF: false,
|
|||
|
in: "A\x80",
|
|||
|
out: "a",
|
|||
|
outFull: "a\x80",
|
|||
|
err: transform.ErrShortDst,
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "short dst writing incomplete rune",
|
|||
|
szDst: 2,
|
|||
|
atEOF: true,
|
|||
|
in: "Σ\xc2",
|
|||
|
out: "Σ",
|
|||
|
outFull: "Σ\xc2",
|
|||
|
err: transform.ErrShortDst,
|
|||
|
t: f(unicode.Latin, toLower, nil),
|
|||
|
}, {
|
|||
|
desc: "short dst, longer",
|
|||
|
szDst: 5,
|
|||
|
atEOF: true,
|
|||
|
in: "Hellø",
|
|||
|
out: "Hell",
|
|||
|
outFull: "Hellø",
|
|||
|
err: transform.ErrShortDst,
|
|||
|
// idem is used to test short buffers by forcing processing of full-rune increments.
|
|||
|
t: f(unicode.Latin, Map(idem), nil),
|
|||
|
}, {
|
|||
|
desc: "short dst, longer, writing error",
|
|||
|
szDst: 6,
|
|||
|
atEOF: false,
|
|||
|
in: "\x80Hello\x80",
|
|||
|
out: "\x80Hello",
|
|||
|
outFull: "\x80Hello\x80",
|
|||
|
err: transform.ErrShortDst,
|
|||
|
t: f(unicode.Latin, Map(idem), nil),
|
|||
|
}, {
|
|||
|
desc: "short src",
|
|||
|
szDst: 2,
|
|||
|
atEOF: false,
|
|||
|
in: "A\xc2",
|
|||
|
out: "a",
|
|||
|
outFull: "a\xc2",
|
|||
|
err: transform.ErrShortSrc,
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "short src no change",
|
|||
|
szDst: 2,
|
|||
|
atEOF: false,
|
|||
|
in: "a\xc2",
|
|||
|
out: "a",
|
|||
|
outFull: "a\xc2",
|
|||
|
err: transform.ErrShortSrc,
|
|||
|
errSpan: transform.ErrShortSrc,
|
|||
|
nSpan: 1,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "invalid input, atEOF",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "\x80",
|
|||
|
out: "\x80",
|
|||
|
outFull: "\x80",
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "invalid input, !atEOF",
|
|||
|
szDst: large,
|
|||
|
atEOF: false,
|
|||
|
in: "\x80",
|
|||
|
out: "\x80",
|
|||
|
outFull: "\x80",
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "invalid input, incomplete rune atEOF",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "\xc2",
|
|||
|
out: "\xc2",
|
|||
|
outFull: "\xc2",
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "nop",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "Hello World!",
|
|||
|
out: "Hello World!",
|
|||
|
outFull: "Hello World!",
|
|||
|
t: f(unicode.Latin, nil, nil),
|
|||
|
}, {
|
|||
|
desc: "nop in",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "THIS IS α ΤΕΣΤ",
|
|||
|
out: "this is α ΤΕΣΤ",
|
|||
|
outFull: "this is α ΤΕΣΤ",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: f(unicode.Greek, nil, toLower),
|
|||
|
}, {
|
|||
|
desc: "nop in latin",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "THIS IS α ΤΕΣΤ",
|
|||
|
out: "THIS IS α τεστ",
|
|||
|
outFull: "THIS IS α τεστ",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: f(unicode.Latin, nil, toLower),
|
|||
|
}, {
|
|||
|
desc: "nop not in",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "THIS IS α ΤΕΣΤ",
|
|||
|
out: "this is α ΤΕΣΤ",
|
|||
|
outFull: "this is α ΤΕΣΤ",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: f(unicode.Latin, toLower, nil),
|
|||
|
}, {
|
|||
|
desc: "pass atEOF is true when at end",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "hello",
|
|||
|
out: "HELLO",
|
|||
|
outFull: "HELLO",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: f(unicode.Latin, upperAtEOF{}, nil),
|
|||
|
}, {
|
|||
|
desc: "pass atEOF is true when at end of segment",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "hello ",
|
|||
|
out: "HELLO ",
|
|||
|
outFull: "HELLO ",
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
t: f(unicode.Latin, upperAtEOF{}, nil),
|
|||
|
}, {
|
|||
|
desc: "don't pass atEOF is true when atEOF is false",
|
|||
|
szDst: large,
|
|||
|
atEOF: false,
|
|||
|
in: "hello",
|
|||
|
out: "",
|
|||
|
outFull: "HELLO",
|
|||
|
err: transform.ErrShortSrc,
|
|||
|
errSpan: transform.ErrShortSrc,
|
|||
|
t: f(unicode.Latin, upperAtEOF{}, nil),
|
|||
|
}, {
|
|||
|
desc: "pass atEOF is true when at end, no change",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "HELLO",
|
|||
|
out: "HELLO",
|
|||
|
outFull: "HELLO",
|
|||
|
t: f(unicode.Latin, upperAtEOF{}, nil),
|
|||
|
}, {
|
|||
|
desc: "pass atEOF is true when at end of segment, no change",
|
|||
|
szDst: large,
|
|||
|
atEOF: true,
|
|||
|
in: "HELLO ",
|
|||
|
out: "HELLO ",
|
|||
|
outFull: "HELLO ",
|
|||
|
t: f(unicode.Latin, upperAtEOF{}, nil),
|
|||
|
}, {
|
|||
|
desc: "large input ASCII",
|
|||
|
szDst: 12000,
|
|||
|
atEOF: false,
|
|||
|
in: strings.Repeat("HELLO", 2000),
|
|||
|
out: strings.Repeat("hello", 2000),
|
|||
|
outFull: strings.Repeat("hello", 2000),
|
|||
|
errSpan: transform.ErrEndOfSpan,
|
|||
|
err: nil,
|
|||
|
t: lower,
|
|||
|
}, {
|
|||
|
desc: "large input non-ASCII",
|
|||
|
szDst: 12000,
|
|||
|
atEOF: false,
|
|||
|
in: strings.Repeat("\u3333", 2000),
|
|||
|
out: strings.Repeat("\u3333", 2000),
|
|||
|
outFull: strings.Repeat("\u3333", 2000),
|
|||
|
err: nil,
|
|||
|
t: lower,
|
|||
|
}} {
|
|||
|
tt.check(t, i)
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
// upperAtEOF is a strange Transformer that converts text to uppercase, but only
|
|||
|
// if atEOF is true.
|
|||
|
type upperAtEOF struct{ transform.NopResetter }
|
|||
|
|
|||
|
func (upperAtEOF) Transform(dst, src []byte, atEOF bool) (nDst, nSrc int, err error) {
|
|||
|
if !atEOF {
|
|||
|
return 0, 0, transform.ErrShortSrc
|
|||
|
}
|
|||
|
return toUpper.Transform(dst, src, atEOF)
|
|||
|
}
|
|||
|
|
|||
|
func (upperAtEOF) Span(src []byte, atEOF bool) (n int, err error) {
|
|||
|
if !atEOF {
|
|||
|
return 0, transform.ErrShortSrc
|
|||
|
}
|
|||
|
return toUpper.Span(src, atEOF)
|
|||
|
}
|
|||
|
|
|||
|
func BenchmarkConditional(b *testing.B) {
|
|||
|
doBench(b, If(In(unicode.Hangul), transform.Nop, transform.Nop))
|
|||
|
}
|