1 files changed, 98 insertions, 0 deletions
diff --git a/encoding_test.cc b/encoding_test.cc
new file mode 100644
index 0000000..9dad6b5
--- /dev/null
+++ b/encoding_test.cc
@@ -0,0 +1,98 @@
+// Copyright 2020 Benjamin Barenblat
+//
+// Licensed under the Apache License, Version 2.0 (the "License"); you may not
+// use this file except in compliance with the License.  You may obtain a copy
+// of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+// WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
+// License for the specific language governing permissions and limitations under
+// the License.
+
+#include "encoding.h"
+
+#include <absl/strings/str_format.h>
+#include <gtest/gtest.h>
+
+namespace scoville {
+namespace {
+
+const char kAllGoodCharacters[] =
+    " !#$&'()+,-.0123456789;=@ABCDEFGHIJKLMNOPQRSTUVWXYZ[]^_`abcdefghijklmnopqr"
+    "stuvwxyz{}~\177";
+
+TEST(ScovilleEncodingTest, EncodesEmptyToEmpty) { EXPECT_EQ(Encode(""), ""); }
+
+TEST(ScovilleEncodingTest, EncodesBadCharacters) {
+  for (int i = 1; i < 0x20; ++i) {
+    EXPECT_EQ(Encode(absl::StrFormat("foo%cbar", i)),
+              absl::StrFormat("foo%%%02xbar", i));
+  }
+  EXPECT_EQ(Encode("foo*bar"), "foo%2abar");
+  EXPECT_EQ(Encode("foo?bar"), "foo%3fbar");
+  EXPECT_EQ(Encode("foo<bar"), "foo%3cbar");
+  EXPECT_EQ(Encode("foo>bar"), "foo%3ebar");
+  EXPECT_EQ(Encode("foo|bar"), "foo%7cbar");
+  EXPECT_EQ(Encode("foo\"bar"), "foo%22bar");
+  EXPECT_EQ(Encode("foo:bar"), "foo%3abar");
+  EXPECT_EQ(Encode("foo\\bar"), "foo%5cbar");
+}
+
+TEST(ScovilleEncodingTest, EncodesPercent) {
+  EXPECT_EQ(Encode("foo%bar"), "foo%%bar");
+}
+
+TEST(ScovilleEncodingTest, EncodesGoodCharacters) {
+  EXPECT_EQ(Encode(kAllGoodCharacters), kAllGoodCharacters);
+}
+
+TEST(ScovilleEncodingTest, EncodesTrailingBadCharacters) {
+  EXPECT_EQ(Encode("foo."), "foo%2e");
+  EXPECT_EQ(Encode("foo "), "foo%20");
+}
+
+TEST(ScovilleEncodingTest, EncodesDirectoryTrailingBadCharacters) {
+  EXPECT_EQ(Encode("foo./bar"), "foo%2e/bar");
+  EXPECT_EQ(Encode("foo /bar"), "foo%20/bar");
+}
+
+TEST(ScovilleDecodingTest, DecodesEmptyToEmpty) { EXPECT_EQ(Decode(""), ""); }
+
+TEST(ScovilleDecodingTest, DecodesBadCharacters) {
+  for (int i = 1; i < 0x20; ++i) {
+    EXPECT_EQ(Decode(absl::StrFormat("foo%%%02xbar", i)),
+              absl::StrFormat("foo%cbar", i));
+  }
+  EXPECT_EQ(Decode("foo%2abar"), "foo*bar");
+  EXPECT_EQ(Decode("foo%3fbar"), "foo?bar");
+  EXPECT_EQ(Decode("foo%3cbar"), "foo<bar");
+  EXPECT_EQ(Decode("foo%3ebar"), "foo>bar");
+  EXPECT_EQ(Decode("foo%7cbar"), "foo|bar");
+  EXPECT_EQ(Decode("foo%22bar"), "foo\"bar");
+  EXPECT_EQ(Decode("foo%3abar"), "foo:bar");
+  EXPECT_EQ(Decode("foo%5cbar"), "foo\\bar");
+}
+
+TEST(ScovilleDecodingTest, DecodesPercent) {
+  EXPECT_EQ(Decode("foo%%bar"), "foo%bar");
+}
+
+TEST(ScovilleDecodingTest, DecodesGoodCharacters) {
+  EXPECT_EQ(Decode(kAllGoodCharacters), kAllGoodCharacters);
+}
+
+TEST(ScovilleDecodingTest, DecodesTrailingBadCharacters) {
+  EXPECT_EQ(Decode("foo%2e"), "foo.");
+  EXPECT_EQ(Decode("foo%20"), "foo ");
+}
+
+TEST(ScovilleDecodingTest, DecodesDirectoryTrailingBadCharacters) {
+  EXPECT_EQ(Decode("foo%2e/bar"), "foo./bar");
+  EXPECT_EQ(Decode("foo%20/bar"), "foo /bar");
+}
+
+}  // namespace
+}  // namespace scoville