archives/snowflake
2
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git synced 2025-10-13 20:11:19 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for proxy poll with Relay URL

Browse source
This commit is contained in:
Shelikhoo 2022-05-02 13:24:39 +01:00
parent 6e8fbe54ee
commit 1b48ee14f4
No known key found for this signature in database
GPG key ID: C4D5E79D22B25316
1 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

28
common/messages/messages_test.go
View file

@ -194,6 +194,34 @@ func TestEncodeProxyPollResponse(t *testing.T) {
So(err, ShouldEqual, nil) So(err, ShouldEqual, nil)
}) })
} }
func TestEncodeProxyPollResponseWithProxyURL(t *testing.T) {
Convey("Context", t, func() {
b, err := EncodePollResponseWithRelayURL("fake offer", true, "restricted", "wss://test/", "")
So(err, ShouldBeNil)
offer, natType, err := DecodePollResponse(b)
So(err, ShouldNotBeNil)
offer, natType, relay, err := DecodePollResponseWithRelayURL(b)
So(offer, ShouldEqual, "fake offer")
So(natType, ShouldEqual, "restricted")
So(relay, ShouldEqual, "wss://test/")
So(err, ShouldBeNil)
b, err = EncodePollResponse("", false, "unknown")
So(err, ShouldBeNil)
offer, natType, relay, err = DecodePollResponseWithRelayURL(b)
So(offer, ShouldEqual, "")
So(natType, ShouldEqual, "unknown")
So(err, ShouldBeNil)
b, err = EncodePollResponseWithRelayURL("fake offer", false, "restricted", "wss://test/", "test error reason")
So(err, ShouldBeNil)
offer, natType, relay, err = DecodePollResponseWithRelayURL(b)
So(err, ShouldNotBeNil)
So(err.Error(), ShouldContainSubstring, "test error reason")
})
}
func TestDecodeProxyAnswerRequest(t *testing.T) { func TestDecodeProxyAnswerRequest(t *testing.T) {
Convey("Context", t, func() { Convey("Context", t, func() {
for _, test := range []struct { for _, test := range []struct {