net/websockets/websocket_extension_parser_test.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/websockets/websocket_extension_parser.h"
   6
   7 #include <string>
   8
   9 #include "net/websockets/websocket_extension.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace net {
  13
  14 namespace {
  15
  16 TEST(WebSocketExtensionParserTest, ParseEmpty) {
  17   WebSocketExtensionParser parser;
  18   parser.Parse("", 0);
  19
  20   EXPECT_TRUE(parser.has_error());
  21 }
  22
  23 TEST(WebSocketExtensionParserTest, ParseSimple) {
  24   WebSocketExtensionParser parser;
  25   WebSocketExtension expected("foo");
  26
  27   parser.Parse("foo");
  28
  29   ASSERT_FALSE(parser.has_error());
  30   EXPECT_TRUE(expected.Equals(parser.extension()));
  31 }
  32
  33 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithoutValue) {
  34   WebSocketExtensionParser parser;
  35   WebSocketExtension expected("foo");
  36   expected.Add(WebSocketExtension::Parameter("bar"));
  37
  38   parser.Parse("\tfoo ; bar");
  39
  40   ASSERT_FALSE(parser.has_error());
  41   EXPECT_TRUE(expected.Equals(parser.extension()));
  42 }
  43
  44 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithOneParamWithValue) {
  45   WebSocketExtensionParser parser;
  46   WebSocketExtension expected("foo");
  47   expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  48
  49   parser.Parse("foo ; bar= baz\t");
  50
  51   ASSERT_FALSE(parser.has_error());
  52   EXPECT_TRUE(expected.Equals(parser.extension()));
  53 }
  54
  55 TEST(WebSocketExtensionParserTest, ParseOneExtensionWithParams) {
  56   WebSocketExtensionParser parser;
  57   WebSocketExtension expected("foo");
  58   expected.Add(WebSocketExtension::Parameter("bar", "baz"));
  59   expected.Add(WebSocketExtension::Parameter("hoge", "fuga"));
  60
  61   parser.Parse("foo ; bar= baz;\t \thoge\t\t=fuga");
  62
  63   ASSERT_FALSE(parser.has_error());
  64   EXPECT_TRUE(expected.Equals(parser.extension()));
  65 }
  66
  67 TEST(WebSocketExtensionParserTest, InvalidPatterns) {
  68   const char* const patterns[] = {
  69     "fo\ao",  // control in extension name
  70     "fo\x01o",  // control in extension name
  71     "fo<o",  // separator in extension name
  72     "foo/",  // separator in extension name
  73     ";bar",  // empty extension name
  74     "foo bar",  // missing ';'
  75     "foo;",  // extension parameter without name and value
  76     "foo; b\ar",  // control in parameter name
  77     "foo; b\x7fr",  // control in parameter name
  78     "foo; b[r",  // separator in parameter name
  79     "foo; ba:",  // separator in parameter name
  80     "foo; =baz",  // empty parameter name
  81     "foo; bar=",  // empty parameter value
  82     "foo; =",  // empty parameter name and value
  83     "foo; bar=b\x02z",  // control in parameter value
  84     "foo; bar=b@z",  // separator in parameter value
  85     "foo; bar=b\\z",  // separator in parameter value
  86     "foo; bar=b?z",  // separator in parameter value
  87     "\"foo\"",  // quoted extension name
  88     "foo; \"bar\"",  // quoted parameter name
  89     "foo; bar=\"\a2\"",  // control in quoted parameter value
  90     "foo; bar=\"b@z\"",  // separator in quoted parameter value
  91     "foo; bar=\"b\\\\z\"",  // separator in quoted parameter value
  92     "foo; bar=\"\"",  // quoted empty parameter value
  93     "foo; bar=\"baz",  // unterminated quoted string
  94     "foo; bar=\"baz \"",  // space in quoted string
  95     "foo; bar baz",  // mising '='
  96     "foo; bar - baz",  // '-' instead of '=' (note: "foo; bar-baz" is valid).
  97     "foo; bar=\r\nbaz",  // CRNL not followed by a space
  98     "foo; bar=\r\n baz",  // CRNL followed by a space
  99     "foo, bar"  // multiple extensions
 100   };
 101
 102   for (size_t i = 0; i < arraysize(patterns); ++i) {
 103     WebSocketExtensionParser parser;
 104     parser.Parse(patterns[i]);
 105     EXPECT_TRUE(parser.has_error());
 106   }
 107 }
 108
 109 TEST(WebSocketExtensionParserTest, QuotedParameterValue) {
 110   WebSocketExtensionParser parser;
 111   WebSocketExtension expected("foo");
 112   expected.Add(WebSocketExtension::Parameter("bar", "baz"));
 113
 114   parser.Parse("foo; bar = \"ba\\z\" ");
 115
 116   ASSERT_FALSE(parser.has_error());
 117   EXPECT_TRUE(expected.Equals(parser.extension()));
 118 }
 119
 120 }  // namespace
 121
 122 }  // namespace net