1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
package examples
import (
"testing"
"time"
"github.com/antonmedv/expr"
"github.com/stretchr/testify/require"
)
func TestExamples_dates(t *testing.T) {
code := `
Now() > Date("2020-01-01") &&
Now() - CreatedAt > Duration("24h")
`
options := []expr.Option{
expr.Env(Env{}),
// Operators override for date comprising.
expr.Operator("==", "Equal"),
expr.Operator("<", "Before"),
expr.Operator("<=", "BeforeOrEqual"),
expr.Operator(">", "After"),
expr.Operator(">=", "AfterOrEqual"),
// Time and duration manipulation.
expr.Operator("+", "Add"),
expr.Operator("-", "Sub"),
// Operators override for duration comprising.
expr.Operator("==", "EqualDuration"),
expr.Operator("<", "BeforeDuration"),
expr.Operator("<=", "BeforeOrEqualDuration"),
expr.Operator(">", "AfterDuration"),
expr.Operator(">=", "AfterOrEqualDuration"),
}
program, err := expr.Compile(code, options...)
require.NoError(t, err)
env := Env{
CreatedAt: Env{}.Date("2018-07-14"), // first commit date
}
output, err := expr.Run(program, env)
require.NoError(t, err)
require.Equal(t, true, output)
}
type Env struct {
datetime
CreatedAt time.Time
}
type datetime struct{}
func (datetime) Date(s string) time.Time {
t, err := time.Parse("2006-01-02", s)
if err != nil {
panic(err)
}
return t
}
func (datetime) Duration(s string) time.Duration {
d, err := time.ParseDuration(s)
if err != nil {
panic(err)
}
return d
}
func (datetime) Now() time.Time { return time.Now() }
func (datetime) Equal(a, b time.Time) bool { return a.Equal(b) }
func (datetime) Before(a, b time.Time) bool { return a.Before(b) }
func (datetime) BeforeOrEqual(a, b time.Time) bool { return a.Before(b) || a.Equal(b) }
func (datetime) After(a, b time.Time) bool { return a.After(b) }
func (datetime) AfterOrEqual(a, b time.Time) bool { return a.After(b) || a.Equal(b) }
func (datetime) Add(a time.Time, b time.Duration) time.Time { return a.Add(b) }
func (datetime) Sub(a, b time.Time) time.Duration { return a.Sub(b) }
func (datetime) EqualDuration(a, b time.Duration) bool { return a == b }
func (datetime) BeforeDuration(a, b time.Duration) bool { return a < b }
func (datetime) BeforeOrEqualDuration(a, b time.Duration) bool { return a <= b }
func (datetime) AfterDuration(a, b time.Duration) bool { return a > b }
func (datetime) AfterOrEqualDuration(a, b time.Duration) bool { return a >= b }
|