1 files changed, 36 insertions, 0 deletions
diff --git a/test/filename_test.rb b/test/filename_test.rb
new file mode 100644
index 0000000000..c42ae8ca54
--- /dev/null
+++ b/test/filename_test.rb
@@ -0,0 +1,36 @@
+require "test_helper"
+
+class ActiveFile::FilenameTest < ActiveSupport::TestCase
+  test "sanitize" do
+    "%$|:;/\t\r\n\\".each_char do |character|
+      filename = ActiveFile::Filename.new("foo#{character}bar.pdf")
+      assert_equal 'foo-bar.pdf', filename.sanitized
+      assert_equal 'foo-bar.pdf', filename.to_s
+    end
+  end
+
+  test "sanitize transcodes to valid UTF-8" do
+    { "\xF6".force_encoding(Encoding::ISO8859_1) => "ö",
+      "\xC3".force_encoding(Encoding::ISO8859_1) => "Ã",
+      "\xAD" => "�",
+      "\xCF" => "�",
+      "\x00" => "",
+    }.each do |actual, expected|
+      assert_equal expected, ActiveFile::Filename.new(actual).sanitized
+    end
+  end
+
+  test "strips RTL override chars used to spoof unsafe executables as docs" do
+    # Would be displayed in Windows as "evilexe.pdf" due to the right-to-left
+    # (RTL) override char!
+    assert_equal 'evil-fdp.exe', ActiveFile::Filename.new("evil\u{202E}fdp.exe").sanitized
+  end
+
+  test "compare case-insensitively" do
+    assert_operator ActiveFile::Filename.new('foobar.pdf'), :==, ActiveFile::Filename.new('FooBar.PDF')
+  end
+
+  test "compare sanitized" do
+    assert_operator ActiveFile::Filename.new('foo-bar.pdf'), :==, ActiveFile::Filename.new("foo\tbar.pdf")
+  end
+end