/
ColumnFilterBuilderTest.scala
93 lines (72 loc) · 2.99 KB
/
ColumnFilterBuilderTest.scala
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
/**
* Copyright (c) 2002-2015 "Neo Technology,"
* Network Engine for Objects in Lund AB [http://neotechnology.com]
*
* This file is part of Neo4j.
*
* Neo4j is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.neo4j.cypher.internal.executionplan.builders
import org.junit.Test
import org.junit.Assert._
import org.neo4j.cypher.internal.executionplan.PartiallySolvedQuery
import org.neo4j.cypher.internal.commands.{Slice, ReturnItem, SortItem}
import org.neo4j.cypher.internal.commands.expressions.{Identifier, Literal}
class ColumnFilterBuilderTest extends BuilderTest {
val builder = new ColumnFilterBuilder
@Test def should_accept_if_all_work_is_done_and_sorting_not_yet() {
val q = PartiallySolvedQuery().copy(
extracted = true,
returns = Seq(Solved(ReturnItem(Literal("foo"), "foo")))
)
val p = createPipe(nodes = Seq("x"))
assertTrue("Builder should accept this", builder.canWorkWith(plan(p, q)))
val resultQ = builder(plan(p, q)).query
assert(resultQ.returns === q.returns.map(_.solve))
}
@Test def should_not_accept_if_not_yet_extracted() {
val q = PartiallySolvedQuery().copy(
extracted = false,
returns = Seq(Unsolved(ReturnItem(Literal("foo"), "foo")))
)
val p = createPipe(nodes = Seq("x"))
assertFalse("Builder should not accept this", builder.canWorkWith(plan(p, q)))
}
@Test def should_not_accept_if_not_sorted() {
val q = PartiallySolvedQuery().copy(
extracted = true,
sort = Seq(Unsolved(SortItem(Literal(1), true))),
returns = Seq(Unsolved(ReturnItem(Literal("foo"), "foo")))
)
val p = createPipe(nodes = Seq("x"))
assertFalse("Builder should not accept this", builder.canWorkWith(plan(p, q)))
}
@Test def should_not_accept_if_not_sliced() {
val q = PartiallySolvedQuery().copy(
extracted = true,
slice = Some(Unsolved(Slice(Some(Literal(19)), None))),
returns = Seq(Unsolved(ReturnItem(Literal("foo"), "foo")))
)
val p = createPipe(nodes = Seq("x"))
assertFalse("Builder should not accept this", builder.canWorkWith(plan(p, q)))
}
@Test def should_not_introduce_column_filter_pipe_unless_needed() {
val q = PartiallySolvedQuery().copy(
extracted = true,
returns = Seq(Solved(ReturnItem(Identifier("foo"), "foo")))
)
val p = createPipe(nodes = Seq("foo"))
assertFalse("Builder should not accept this", builder.canWorkWith(plan(p, q)))
}
}