add test

giampaolo · Jul 8, 2024 · 6621db8 · 6621db8
1 parent 730e062
commit 6621db8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/pyftpdlib/test/__init__.py b/pyftpdlib/test/__init__.py
@@ -379,6 +379,7 @@ def reset_server_opts():
         klass.use_sendfile = hasattr(os, "sendfile")
         klass.ac_in_buffer_size = 4096
         klass.ac_out_buffer_size = 4096
+        klass.encoding = "utf8"
         if klass.__name__ == 'TLS_FTPHandler':
             klass.tls_control_required = False
             klass.tls_data_required = False

diff --git a/pyftpdlib/test/test_functional.py b/pyftpdlib/test/test_functional.py
@@ -1693,6 +1693,7 @@ def tearDown(self):
             self.server.handler.passive_ports = None
             self.server.handler.use_gmt_times = True
             self.server.handler.tcp_no_delay = hasattr(socket, 'TCP_NODELAY')
+            self.server.handler.encoding = "utf8"
             self.server.stop()
         super().tearDown()
 
@@ -1907,6 +1908,16 @@ def test_use_gmt_times(self):
             assert gmt2 == loc2
             assert gmt3 == loc3
 
+    def test_encoding(self):
+        # Make sure that if encoding != UTF-8, FEAT command does not
+        # list UTF-8.
+        self.server = self.server_class()
+        self.server.handler.encoding = "latin-1"
+        self.server.start()
+        self.connect()
+        resp = self.client.sendcmd('feat')
+        assert 'UTF8' not in resp
+
 
 @pytest.mark.xdist_group(name="serial")
 class TestCallbacks(PyftpdlibTestCase):