mirror of
https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git
synced 2025-10-14 05:11:19 -04:00
95 lines
2.6 KiB
CoffeeScript
95 lines
2.6 KiB
CoffeeScript
###
|
|
jasmine tests for Snowflake
|
|
###
|
|
|
|
# Stubs to fake browser functionality.
|
|
class PeerConnection
|
|
class WebSocket
|
|
OPEN: 1
|
|
CLOSED: 0
|
|
ui =
|
|
log: ->
|
|
setActive: ->
|
|
log = ->
|
|
|
|
describe 'ProxyPair', ->
|
|
fakeRelay = Parse.address '0.0.0.0:12345'
|
|
rateLimit = new DummyRateLimit()
|
|
destination = []
|
|
fakeClient = send: (d) -> destination.push d
|
|
# Fake snowflake to interact with
|
|
snowflake = {
|
|
broker:
|
|
sendAnswer: ->
|
|
}
|
|
pp = new ProxyPair(fakeClient, fakeRelay, rateLimit)
|
|
|
|
it 'begins webrtc connection', ->
|
|
pp.begin()
|
|
expect(pp.pc).not.toBeNull()
|
|
|
|
it 'accepts WebRTC offer from some client', ->
|
|
it 'rejects invalid offers', ->
|
|
expect(pp.receiveWebRTCOffer {}).toBe false
|
|
expect pp.receiveWebRTCOffer {
|
|
type: 'answer'
|
|
}.toBeFalse()
|
|
it 'accepts valid offers', ->
|
|
goodOffer = {
|
|
type: 'offer'
|
|
sdp: 'foo'
|
|
}
|
|
expect(pp.receiveWebRTCOffer goodOffer).toBe true
|
|
|
|
it 'responds with a WebRTC answer correctly', ->
|
|
spyOn snowflake.broker, 'sendAnswer'
|
|
pp.pc.onicecandidate {
|
|
candidate: null
|
|
}
|
|
expect(snowflake.broker.sendAnswer).toHaveBeenCalled()
|
|
|
|
it 'handles a new data channel correctly', ->
|
|
expect(pp.client).toBeNull()
|
|
pp.pc.ondatachannel {
|
|
channel: {}
|
|
}
|
|
expect(pp.client).not.toBeNull()
|
|
expect(pp.client.onopen).not.toBeNull()
|
|
expect(pp.client.onclose).not.toBeNull()
|
|
expect(pp.client.onerror).not.toBeNull()
|
|
expect(pp.client.onmessage).not.toBeNull()
|
|
|
|
it 'connects to the relay once datachannel opens', ->
|
|
spyOn pp, 'connectRelay'
|
|
pp.client.onopen()
|
|
expect(pp.connectRelay).toHaveBeenCalled()
|
|
|
|
it 'connects to a relay', ->
|
|
pp.connectRelay()
|
|
expect(pp.relay.onopen).not.toBeNull()
|
|
expect(pp.relay.onclose).not.toBeNull()
|
|
expect(pp.relay.onerror).not.toBeNull()
|
|
expect(pp.relay.onmessage).not.toBeNull()
|
|
|
|
it 'flushes data between client and relay', ->
|
|
|
|
it 'proxies data from client to relay', ->
|
|
spyOn pp.relay, 'send'
|
|
pp.c2rSchedule.push { data: 'foo' }
|
|
pp.flush()
|
|
expect(pp.client.send).not.toHaveBeenCalled()
|
|
expect(pp.relay.send).toHaveBeenCalledWith 'foo'
|
|
|
|
it 'proxies data from relay to client', ->
|
|
spyOn pp.client, 'send'
|
|
pp.r2cSchedule.push { data: 'bar' }
|
|
pp.flush()
|
|
expect(pp.client.send).toHaveBeenCalledWith 'bar'
|
|
expect(pp.relay.send).not.toHaveBeenCalled()
|
|
|
|
it 'sends nothing with nothing to flush', ->
|
|
pp.flush()
|
|
expect(pp.client.send).not.toHaveBeenCalled()
|
|
expect(pp.relay.send).not.toHaveBeenCalled()
|
|
|
|
# TODO: rate limit tests
|