forked from celluloid/reel
-
Notifications
You must be signed in to change notification settings - Fork 0
/
websocket_spec.rb
77 lines (60 loc) · 1.87 KB
/
websocket_spec.rb
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
require 'spec_helper'
describe Reel::WebSocket do
include WebSocketHelpers
let(:example_message) { "Hello, World!" }
let(:another_message) { "What's going on?" }
it "performs websocket handshakes" do
with_socket_pair do |client, connection|
client << handshake.to_data
websocket = connection.request
websocket.should be_a Reel::WebSocket
handshake.errors.should be_empty
end
end
it "knows its URL" do
with_websocket_pair do |_, websocket|
websocket.url.should == example_path
end
end
it "knows its headers" do
with_websocket_pair do |_, websocket|
websocket['Host'].should == example_host
end
end
it "reads frames" do
with_websocket_pair do |client, websocket|
client << WebSocket::Message.new(example_message).to_data
client << WebSocket::Message.new(another_message).to_data
websocket.read.should == example_message
websocket.read.should == another_message
end
end
it "writes messages" do
with_websocket_pair do |client, websocket|
websocket.write example_message
websocket.write another_message
parser = WebSocket::Parser.new
parser.append client.readpartial(4096) until first_message = parser.next_message
first_message.should == example_message
parser.append client.readpartial(4096) until next_message = parser.next_message
next_message.should == another_message
end
end
it "closes" do
with_websocket_pair do |_, websocket|
websocket.should_not be_closed
websocket.close
websocket.should be_closed
end
end
def with_websocket_pair
with_socket_pair do |client, connection|
client << handshake.to_data
websocket = connection.request
websocket.should be_a Reel::WebSocket
# Discard handshake
client.readpartial(4096)
yield client, websocket
end
end
end