Refactored tests into smaller modules.

2026-04-26 20:02:11 +02:00 · 2014-04-24 14:07:31 +02:00
parent 746a1899f3
commit 6071fff4af
18 changed files with 1803 additions and 1762 deletions
@@ -0,0 +1,93 @@
+"""Tests for dealing with binary request and response data."""
+from httpie.compat import urlopen
+from httpie.output import BINARY_SUPPRESSED_NOTICE
+from tests import (
+    BaseTestCase, TestEnvironment, http, httpbin,
+    BIN_FILE_PATH, BIN_FILE_CONTENT, BIN_FILE_PATH_ARG,
+
+)
+
+
+class BinaryRequestDataTest(BaseTestCase):
+
+    def test_binary_stdin(self):
+        with open(BIN_FILE_PATH, 'rb') as stdin:
+            env = TestEnvironment(
+                stdin=stdin,
+                stdin_isatty=False,
+                stdout_isatty=False
+            )
+            r = http(
+                '--print=B',
+                'POST',
+                httpbin('/post'),
+                env=env,
+            )
+            self.assertEqual(r, BIN_FILE_CONTENT)
+
+    def test_binary_file_path(self):
+        env = TestEnvironment(
+            stdin_isatty=True,
+            stdout_isatty=False
+        )
+        r = http(
+            '--print=B',
+            'POST',
+            httpbin('/post'),
+            '@' + BIN_FILE_PATH_ARG,
+            env=env,
+        )
+
+        self.assertEqual(r, BIN_FILE_CONTENT)
+
+    def test_binary_file_form(self):
+        env = TestEnvironment(
+            stdin_isatty=True,
+            stdout_isatty=False
+        )
+        r = http(
+            '--print=B',
+            '--form',
+            'POST',
+            httpbin('/post'),
+            'test@' + BIN_FILE_PATH_ARG,
+            env=env,
+        )
+        self.assertIn(bytes(BIN_FILE_CONTENT), bytes(r))
+
+
+class BinaryResponseDataTest(BaseTestCase):
+
+    url = 'http://www.google.com/favicon.ico'
+
+    @property
+    def bindata(self):
+        if not hasattr(self, '_bindata'):
+            self._bindata = urlopen(self.url).read()
+        return self._bindata
+
+    def test_binary_suppresses_when_terminal(self):
+        r = http(
+            'GET',
+            self.url
+        )
+        self.assertIn(BINARY_SUPPRESSED_NOTICE.decode(), r)
+
+    def test_binary_suppresses_when_not_terminal_but_pretty(self):
+        r = http(
+            '--pretty=all',
+            'GET',
+            self.url,
+            env=TestEnvironment(stdin_isatty=True,
+                            stdout_isatty=False)
+        )
+        self.assertIn(BINARY_SUPPRESSED_NOTICE.decode(), r)
+
+    def test_binary_included_and_correct_when_suitable(self):
+        r = http(
+            'GET',
+            self.url,
+            env=TestEnvironment(stdin_isatty=True,
+                            stdout_isatty=False)
+        )
+        self.assertEqual(r, self.bindata)