-
Notifications
You must be signed in to change notification settings - Fork 10
/
UserSocketSpec.scala
74 lines (59 loc) · 2.33 KB
/
UserSocketSpec.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
package actors
import actors.UserSocket.Message
import actors.ChatMessage
import akka.actor._
import akka.cluster.ddata.DistributedData
import akka.cluster.ddata.LWWRegister
import akka.cluster.ddata.Replicator
import akka.cluster.ddata.Replicator._
import akka.testkit.TestProbe
import org.specs2.mutable._
import play.api.libs.json._
import scala.concurrent.duration._
class UserSocketSpec extends Specification {
val UserId = "user1"
implicit val messageWrites = new Writes[Message] {
def writes(message: Message) = Json.obj(
"type" -> "message",
"topic" -> message.topic,
"msg" -> message.msg
)
}
case class MsgSubscribe(topic: String)
implicit val msgSubscribeWrites = new Writes[MsgSubscribe] {
def writes(message: MsgSubscribe) = Json.obj(
"type" -> "subscribe",
"topic" -> message.topic
)
}
"A user socket" should {
val topic = "chat"
"send chat message to all subscribers" in new AkkaTestkitSpecs2Support {
//val mediator = DistributedPubSub(system).mediator
val replicator = DistributedData(system).replicator
val browser = TestProbe()
val chatMember1 = TestProbe()
val chatMember2 = TestProbe()
replicator ! Subscribe(UserSocket.topicMsgKey(topic), chatMember1.ref)
replicator ! Subscribe(UserSocket.topicMsgKey(topic), chatMember2.ref)
val socket = system.actorOf(UserSocket.props("user1")(browser.ref), "userSocket")
browser.expectNoMsg(6 seconds)
val message = "hello"
socket ! Json.toJson(Message(topic, message))
chatMember1.expectMsgPF(3 seconds){case c : Changed[LWWRegister[ChatMessage]] => true}
chatMember2.expectMsgPF(3 seconds){case c : Changed[LWWRegister[ChatMessage]] => true}
}
"forward chat message to browser" in new AkkaTestkitSpecs2Support {
val replicator = DistributedData(system).replicator
val browser = TestProbe()
val socket = system.actorOf(UserSocket.props(UserId)(browser.ref), "userSocket")
val text = "There is important thing to do!"
val chatMessage = ChatMessage(topic, UserId, text, new java.util.Date())
browser.expectNoMsg(6 seconds)
socket ! Json.toJson(MsgSubscribe(topic))
browser.expectNoMsg(2 seconds)
socket ! Json.toJson(Message(topic, text))
browser.expectMsg(Json.toJson(chatMessage))
}
}
}