vendor/github.com/hashicorp/yamux/const_test.go

   1 package yamux
   2
   3 import (
   4         "testing"
   5 )
   6
   7 func TestConst(t *testing.T) {
   8         if protoVersion != 0 {
   9                 t.Fatalf("bad: %v", protoVersion)
  10         }
  11
  12         if typeData != 0 {
  13                 t.Fatalf("bad: %v", typeData)
  14         }
  15         if typeWindowUpdate != 1 {
  16                 t.Fatalf("bad: %v", typeWindowUpdate)
  17         }
  18         if typePing != 2 {
  19                 t.Fatalf("bad: %v", typePing)
  20         }
  21         if typeGoAway != 3 {
  22                 t.Fatalf("bad: %v", typeGoAway)
  23         }
  24
  25         if flagSYN != 1 {
  26                 t.Fatalf("bad: %v", flagSYN)
  27         }
  28         if flagACK != 2 {
  29                 t.Fatalf("bad: %v", flagACK)
  30         }
  31         if flagFIN != 4 {
  32                 t.Fatalf("bad: %v", flagFIN)
  33         }
  34         if flagRST != 8 {
  35                 t.Fatalf("bad: %v", flagRST)
  36         }
  37
  38         if goAwayNormal != 0 {
  39                 t.Fatalf("bad: %v", goAwayNormal)
  40         }
  41         if goAwayProtoErr != 1 {
  42                 t.Fatalf("bad: %v", goAwayProtoErr)
  43         }
  44         if goAwayInternalErr != 2 {
  45                 t.Fatalf("bad: %v", goAwayInternalErr)
  46         }
  47
  48         if headerSize != 12 {
  49                 t.Fatalf("bad header size")
  50         }
  51 }
  52
  53 func TestEncodeDecode(t *testing.T) {
  54         hdr := header(make([]byte, headerSize))
  55         hdr.encode(typeWindowUpdate, flagACK|flagRST, 1234, 4321)
  56
  57         if hdr.Version() != protoVersion {
  58                 t.Fatalf("bad: %v", hdr)
  59         }
  60         if hdr.MsgType() != typeWindowUpdate {
  61                 t.Fatalf("bad: %v", hdr)
  62         }
  63         if hdr.Flags() != flagACK|flagRST {
  64                 t.Fatalf("bad: %v", hdr)
  65         }
  66         if hdr.StreamID() != 1234 {
  67                 t.Fatalf("bad: %v", hdr)
  68         }
  69         if hdr.Length() != 4321 {
  70                 t.Fatalf("bad: %v", hdr)
  71         }
  72 }