Add tests for six.environ

benjaminp · Sep 14, 2018 · 74ba4a0 · 74ba4a0
1 parent 6f5915a
commit 74ba4a0
Showing 1 changed file with 53 additions and 0 deletions.
diff --git a/test_six.py b/test_six.py
@@ -990,3 +990,56 @@ def test_ensure_text(self):
             assert converted_unicode == self.UNICODE_EMOJI and isinstance(converted_unicode, str)
             # PY3: bytes -> str
             assert converted_binary == self.UNICODE_EMOJI and isinstance(converted_unicode, str)
+
+
+class EnvironTests:
+
+    # grinning face emoji
+    UNICODE_EMOJI = six.u("\U0001F600")
+    BINARY_EMOJI = b"\xf0\x9f\x98\x80"
+
+    def test_set_environ(self):
+        try:
+            del six.environ['_six_test']
+        except KeyError:
+            pass
+
+        six.environ['_six_test'] = UNICODE_EMOJI
+
+        assert six.environ['_six_test'] == UNICODE_EMOJI
+        assert six.environb['_six_test'] == BINARY_EMOJI
+        if PY3:
+            assert os.environ == UNICODE_EMOJI
+        else:
+            assert os.environ == BINARY_EMOJI
+
+        del six.environ['_six_test']
+
+    def test_set_environb(self):
+        try:
+            del six.environ['_six_test']
+        except KeyError:
+            pass
+
+        six.environb['_six_test'] = BINARY_EMOJI
+
+        assert six.environ['_six_test'] == UNICODE_EMOJI
+        assert six.environb['_six_test'] == BINARY_EMOJI
+        if PY3:
+            assert os.environ == UNICODE_EMOJI
+        else:
+            assert os.environ == BINARY_EMOJI
+
+        del six.environ['_six_test']
+
+    def test_del_environ(self):
+        six.environ['_six_test'] = 'testing'
+        assert '_six_test' in os.environ
+        del six.environ['_six_test']
+        assert '_six_test' not in os.environ
+
+    def test_del_environb(self):
+        six.environb['_six_test'] = 'testing'
+        assert '_six_test' in os.environ
+        del six.environb['_six_test']
+        assert '_six_test' not in os.environ