forked from phiSgr/gatling-grpc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
LensExprSpec.scala
149 lines (129 loc) · 4.13 KB
/
LensExprSpec.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
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
146
147
148
149
package com.github.phisgr.gatling.pb
import com.github.phisgr.gatling.grpc.Predef.$
import com.github.phisgr.pb.complex.complex.{Bar, Foo}
import com.typesafe.scalalogging.StrictLogging
import io.gatling.commons.validation.{Failure, Success}
import io.gatling.core.session.{Expression, Session}
import org.scalatest.{FlatSpec, Matchers}
class LensExprSpec extends FlatSpec with Matchers with StrictLogging {
val session = Session(
scenario = "Scenario",
userId = 1L,
attributes = Map(
"bar" -> Bar(baz = 1),
"baz" -> 2,
"bars" -> List(Bar(baz = 5), Bar(baz = 6)),
"stringBar" -> ("bar", Bar(baz = 10)),
"stringBarMap" -> Map("barr" -> Bar(baz = 11)),
"generics" -> List("erasure", "BOOM!")
),
startDate = System.currentTimeMillis()
)
"Setting with lens" should "work" in {
val barE = Bar.defaultInstance.updateExpr(_.baz :~ $("baz"))
barE(session) shouldBe Success(Bar(2))
val fooE = Foo(bar = List(Bar(100))).updateExpr(
_.bar(0).baz :~ $("baz")
)
fooE(session) shouldBe Success(Foo(bar = List(Bar(2))))
}
"Modifying with lens" should "work" in {
val barE = $[Bar]("bar")
.updateExpr(_.baz.modifyExpr(
$[Int]("baz").map(sessionValue => _ - sessionValue)
))
barE(session) shouldBe Success(Bar(-1))
}
"Seq lens" should "work" in {
val fooE = Foo.defaultInstance.updateExpr(
_.bar :+~ Bar.defaultInstance.updateExpr(_.baz :~ $("baz")),
_.bar :+~ $("bar"),
_.bar :++~ $("bars")
)
val foo = Foo(
bar = List(
Bar(2), Bar(1), Bar(5), Bar(6)
)
)
fooE(session) shouldBe Success(foo)
fooE.updateExpr(
_.bar.foreachExpr(_.baz :~ $("baz"))
)(session) shouldBe Success(
Foo(
bar = List(
Bar(2), Bar(2), Bar(2), Bar(2)
)
)
)
}
"Set lens" should "work" in {
val fooE = Foo.defaultInstance.updateExpr(
_.barSet :+~ Bar.defaultInstance.updateExpr(_.baz :~ $("baz")),
_.barSet :+~ $[Bar]("bar"),
_.barSet :++~ $[List[Bar]]("bars")
)
val foo = Foo(
barSet = Set(
Bar(1), Bar(2), Bar(5), Bar(6)
)
)
fooE(session) shouldBe Success(foo)
fooE.updateExpr(
_.barSet.foreachExpr(_.baz :~ $("baz"))
)(session) shouldBe Success(
Foo(barSet = Set(Bar(2)))
)
}
"Map lens" should "work" in {
val fooE = Foo.defaultInstance.updateExpr(
_.barMap :+~ $("stringBar"),
_.barMap :++~ $[List[String]]("generics").map(_.map { s => s -> Bar(s.length) }),
_.barMap :++~ $("stringBarMap")
)
val foo = Foo(
barMap = Map(
"bar" -> 10,
"barr" -> 11,
"erasure" -> 7,
"BOOM!" -> 5
).mapValues(Bar(_))
)
fooE(session) shouldBe Success(foo)
fooE.updateExpr(
_.barMap.foreachExpr(_ :~ $("stringBar"))
)(session) shouldBe Success(
Foo(barMap = Map("bar" -> Bar(baz = 10)))
)
fooE.updateExpr(
_.barMap.foreachValueExpr(_.baz :~ $("baz"))
)(session) shouldBe Success(Foo(
barMap = Map(
"bar" -> 2,
"barr" -> 2,
"erasure" -> 2,
"BOOM!" -> 2
).mapValues(Bar(_))
))
}
"Wrong session variables" should "fail" in {
Bar.defaultInstance.updateExpr(_.baz :~ $("bar"))(session) shouldBe
Failure("Value Bar(1,UnknownFieldSet(Map())) is of type com.github.phisgr.pb.complex.complex.Bar, expected int")
Bar.defaultInstance.updateExpr(_.baz :~ $("nonExisting"))(session) shouldBe
Failure("No attribute named 'nonExisting' is defined")
}
"Erased generics" should "be unsafe" in {
def assertClassCastException(fooE: Expression[Foo]): Unit = {
val Success(foo) = fooE(session)
logger.warn("We have a malformed object: {}", foo)
// Notice that the error is thrown after the expression is evaluated
an[ClassCastException] should be thrownBy foo.toByteArray
}
assertClassCastException(
Foo(bar = List(Bar(1))).updateExpr(_.bar :++~ $("generics"))
)
import io.gatling.core.Predef.stringToExpression
assertClassCastException(
Foo(bar = List(Bar(1))).updateExpr(_.bar :++~ "${generics}")
)
}
}