forked from opendistro-for-elasticsearch/sql
-
Notifications
You must be signed in to change notification settings - Fork 0
/
IntervalClauseTest.java
145 lines (126 loc) · 5.34 KB
/
IntervalClauseTest.java
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
/*
* Copyright 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License").
* You may not use this file except in compliance with the License.
* A copy of the License is located at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* or in the "license" file accompanying this file. This file 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 com.amazon.opendistroforelasticsearch.sql.expression.datetime;
import static com.amazon.opendistroforelasticsearch.sql.data.model.ExprValueUtils.intervalValue;
import static com.amazon.opendistroforelasticsearch.sql.data.model.ExprValueUtils.missingValue;
import static com.amazon.opendistroforelasticsearch.sql.data.model.ExprValueUtils.nullValue;
import static com.amazon.opendistroforelasticsearch.sql.data.type.ExprCoreType.INTEGER;
import static com.amazon.opendistroforelasticsearch.sql.data.type.ExprCoreType.INTERVAL;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.Mockito.when;
import com.amazon.opendistroforelasticsearch.sql.data.model.ExprValue;
import com.amazon.opendistroforelasticsearch.sql.exception.ExpressionEvaluationException;
import com.amazon.opendistroforelasticsearch.sql.expression.DSL;
import com.amazon.opendistroforelasticsearch.sql.expression.Expression;
import com.amazon.opendistroforelasticsearch.sql.expression.ExpressionTestBase;
import com.amazon.opendistroforelasticsearch.sql.expression.FunctionExpression;
import com.amazon.opendistroforelasticsearch.sql.expression.env.Environment;
import java.time.Duration;
import java.time.Period;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
@ExtendWith(MockitoExtension.class)
public class IntervalClauseTest extends ExpressionTestBase {
@Mock
Environment<Expression, ExprValue> env;
@Mock
Expression nullRef;
@Mock
Expression missingRef;
@Test
public void microsecond() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("microsecond"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Duration.ofNanos(1000)), expr.valueOf(env));
}
@Test
public void second() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("second"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Duration.ofSeconds(1)), expr.valueOf(env));
}
@Test
public void minute() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("minute"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Duration.ofMinutes(1)), expr.valueOf(env));
}
@Test
public void hour() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("HOUR"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Duration.ofHours(1)), expr.valueOf(env));
}
@Test
public void day() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("day"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Duration.ofDays(1)), expr.valueOf(env));
}
@Test
public void week() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("week"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Period.ofWeeks(1)), expr.valueOf(env));
}
@Test
public void month() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("month"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Period.ofMonths(1)), expr.valueOf(env));
}
@Test
public void quarter() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("quarter"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Period.ofMonths(3)), expr.valueOf(env));
}
@Test
public void year() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("year"));
assertEquals(INTERVAL, expr.type());
assertEquals(intervalValue(Period.ofYears(1)), expr.valueOf(env));
}
@Test
public void unsupported_unit() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("year_month"));
assertThrows(ExpressionEvaluationException.class, () -> expr.valueOf(env),
"interval unit year_month is not supported");
}
@Test
public void to_string() {
FunctionExpression expr = dsl.interval(DSL.literal(1), DSL.literal("day"));
assertEquals("interval(1, \"day\")", expr.toString());
}
@Test
public void null_value() {
when(nullRef.type()).thenReturn(INTEGER);
when(nullRef.valueOf(env)).thenReturn(nullValue());
FunctionExpression expr = dsl.interval(nullRef, DSL.literal("day"));
assertEquals(INTERVAL, expr.type());
assertEquals(nullValue(), expr.valueOf(env));
}
@Test
public void missing_value() {
when(missingRef.type()).thenReturn(INTEGER);
when(missingRef.valueOf(env)).thenReturn(missingValue());
FunctionExpression expr = dsl.interval(missingRef, DSL.literal("day"));
assertEquals(INTERVAL, expr.type());
assertEquals(missingValue(), expr.valueOf(env));
}
}