Hulk did something

[bytom/vapor.git] / vendor / github.com / gorilla / websocket / compression_test.go

diff --git a/vendor/github.com/gorilla/websocket/compression_test.go b/vendor/github.com/gorilla/websocket/compression_test.go
new file mode 100644 (file)
index 0000000..8a26b30
--- /dev/null
+++ b/vendor/github.com/gorilla/websocket/compression_test.go
@@ -0,0 +1,80 @@
+package websocket
+
+import (
+       "bytes"
+       "fmt"
+       "io"
+       "io/ioutil"
+       "testing"
+)
+
+type nopCloser struct{ io.Writer }
+
+func (nopCloser) Close() error { return nil }
+
+func TestTruncWriter(t *testing.T) {
+       const data = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijlkmnopqrstuvwxyz987654321"
+       for n := 1; n <= 10; n++ {
+               var b bytes.Buffer
+               w := &truncWriter{w: nopCloser{&b}}
+               p := []byte(data)
+               for len(p) > 0 {
+                       m := len(p)
+                       if m > n {
+                               m = n
+                       }
+                       w.Write(p[:m])
+                       p = p[m:]
+               }
+               if b.String() != data[:len(data)-len(w.p)] {
+                       t.Errorf("%d: %q", n, b.String())
+               }
+       }
+}
+
+func textMessages(num int) [][]byte {
+       messages := make([][]byte, num)
+       for i := 0; i < num; i++ {
+               msg := fmt.Sprintf("planet: %d, country: %d, city: %d, street: %d", i, i, i, i)
+               messages[i] = []byte(msg)
+       }
+       return messages
+}
+
+func BenchmarkWriteNoCompression(b *testing.B) {
+       w := ioutil.Discard
+       c := newTestConn(nil, w, false)
+       messages := textMessages(100)
+       b.ResetTimer()
+       for i := 0; i < b.N; i++ {
+               c.WriteMessage(TextMessage, messages[i%len(messages)])
+       }
+       b.ReportAllocs()
+}
+
+func BenchmarkWriteWithCompression(b *testing.B) {
+       w := ioutil.Discard
+       c := newTestConn(nil, w, false)
+       messages := textMessages(100)
+       c.enableWriteCompression = true
+       c.newCompressionWriter = compressNoContextTakeover
+       b.ResetTimer()
+       for i := 0; i < b.N; i++ {
+               c.WriteMessage(TextMessage, messages[i%len(messages)])
+       }
+       b.ReportAllocs()
+}
+
+func TestValidCompressionLevel(t *testing.T) {
+       c := newTestConn(nil, nil, false)
+       for _, level := range []int{minCompressionLevel - 1, maxCompressionLevel + 1} {
+               if err := c.SetCompressionLevel(level); err == nil {
+                       t.Errorf("no error for level %d", level)
+               }
+       }
+       for _, level := range []int{minCompressionLevel, maxCompressionLevel} {
+               if err := c.SetCompressionLevel(level); err != nil {
+                       t.Errorf("error for level %d", level)
+               }
+       }
+}