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
|
package org.zwobble.couscous.tests.transforms;
import org.junit.Test;
import org.zwobble.couscous.ast.Operations;
import org.zwobble.couscous.ast.VariableDeclaration;
import org.zwobble.couscous.ast.identifiers.Identifier;
import org.zwobble.couscous.ast.sugar.SwitchNode;
import org.zwobble.couscous.frontends.java.Scope;
import org.zwobble.couscous.transforms.DesugarSwitchToIfElse;
import org.zwobble.couscous.types.Types;
import static org.junit.Assert.assertEquals;
import static org.zwobble.couscous.ast.ExpressionStatementNode.expressionStatement;
import static org.zwobble.couscous.ast.IfStatementNode.ifStatement;
import static org.zwobble.couscous.ast.LiteralNode.literal;
import static org.zwobble.couscous.ast.LocalVariableDeclarationNode.localVariableDeclaration;
import static org.zwobble.couscous.ast.ReturnNode.returns;
import static org.zwobble.couscous.ast.VariableDeclaration.var;
import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
import static org.zwobble.couscous.ast.sugar.SwitchCaseNode.switchCase;
import static org.zwobble.couscous.util.ExtraLists.list;
public class DesugarSwitchToIfElseTests {
@Test
public void switchStatementWithoutFallThroughIsReadAsIfElseStatement() {
VariableDeclaration tmp = var(
Identifier.TOP.variable("_couscous_tmp_0"),
"_couscous_tmp_0",
Types.STRING
);
assertEquals(
list(
localVariableDeclaration(tmp, literal("one")),
ifStatement(
Operations.methodEquals(reference(tmp), literal("one")),
list(returns(literal(1))),
list(
ifStatement(
Operations.methodEquals(reference(tmp), literal("two")),
list(returns(literal(2))),
list(
returns(literal(0))
)
)
)
)
),
DesugarSwitchToIfElse.desugar(
Scope.create(),
new SwitchNode(
literal("one"),
list(
switchCase(literal("one"), list(returns(literal(1)))),
switchCase(literal("two"), list(returns(literal(2)))),
switchCase(list(returns(literal(0))))
)
)
)
);
}
@Test
public void switchStatementWithoutDefaultIsReadAsIfStatementWithoutElseStatement() {
VariableDeclaration tmp = var(
Identifier.TOP.variable("_couscous_tmp_0"),
"_couscous_tmp_0",
Types.STRING
);
assertEquals(
list(
localVariableDeclaration(tmp, literal("one")),
ifStatement(
Operations.methodEquals(reference(tmp), literal("two")),
list(returns(literal(0))),
list()
)
),
DesugarSwitchToIfElse.desugar(
Scope.create(),
new SwitchNode(
literal("one"),
list(
switchCase(literal("two"), list(returns(literal(0))))
)
)
)
);
}
@Test
public void casesCanFallthrough() {
VariableDeclaration tmp = var(
Identifier.TOP.variable("_couscous_tmp_0"),
"_couscous_tmp_0",
Types.STRING
);
assertEquals(
list(
localVariableDeclaration(tmp, literal("one")),
ifStatement(
Operations.methodEquals(reference(tmp), literal("one")),
list(
expressionStatement(literal(1)),
returns(literal(2))
),
list(
ifStatement(
Operations.methodEquals(reference(tmp), literal("two")),
list(
returns(literal(2))
),
list()
)
)
)
),
DesugarSwitchToIfElse.desugar(
Scope.create(),
new SwitchNode(
literal("one"),
list(
switchCase(literal("one"), list(expressionStatement(literal(1)))),
switchCase(literal("two"), list(returns(literal(2))))
)
)
)
);
}
}
|
