From 7afd34b4cbe4e4e49ffc3ead47b3ffbcc29af913 Mon Sep 17 00:00:00 2001
From: Samuel Colvin <s@muelcolvin.com>
Date: Sat, 6 May 2017 00:11:57 +0100
Subject: [PATCH] add test_complex.py

---
 tests/test_complex.py | 85 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 tests/test_complex.py

diff --git a/tests/test_complex.py b/tests/test_complex.py
new file mode 100644
index 0000000..311c220
--- /dev/null
+++ b/tests/test_complex.py
@@ -0,0 +1,85 @@
+from collections import OrderedDict
+from typing import Union
+
+import pytest
+
+from pydantic import BaseModel, NoneStrBytes, StrBytes, ValidationError
+
+
+def test_str_bytes():
+    class StrBytesModel(BaseModel):
+        v: StrBytes = ...
+
+    m = StrBytesModel(v='s')
+    assert m.v == 's'
+    assert repr(m.fields['v']) == (
+        "<Field v: type='typing.Union[str, bytes]', required=True, "
+        "validators={'str': ['not_none_validator', 'str_validator', 'anystr_length_validator'], "
+        "'bytes': ['not_none_validator', 'bytes_validator', 'anystr_length_validator']}>"
+    )
+
+    m = StrBytesModel(v=b'b')
+    assert m.v == b'b'
+
+    with pytest.raises(ValidationError) as exc_info:
+        StrBytesModel(v=None)
+    assert exc_info.value.message == '1 error validating input'
+    assert exc_info.value.errors == OrderedDict(
+        [
+            ('v', [{'type': 'TypeError', 'route': 'str', 'msg': 'None is not an allow value',
+                    'validator': 'not_none_validator'},
+                   {'type': 'TypeError', 'route': 'bytes', 'msg': 'None is not an allow value',
+                    'validator': 'not_none_validator'}])
+        ])
+
+
+def test_str_bytes_none():
+    class StrBytesModel(BaseModel):
+        v: NoneStrBytes = ...
+
+    m = StrBytesModel(v='s')
+    assert m.v == 's'
+
+    m = StrBytesModel(v=b'b')
+    assert m.v == b'b'
+
+    m = StrBytesModel(v=None)
+    assert m.v is None
+
+    assert m.fields['v'].info == {
+        'required': True,
+        'type': 'typing.Union[str, bytes, NoneType]',
+        'validators': {
+            'bytes': ['bytes_validator', 'anystr_length_validator'],
+            'str': ['str_validator', 'anystr_length_validator']
+        }
+    }
+
+
+def test_union_int_str():
+    class Model(BaseModel):
+        v: Union[int, str] = ...
+
+    m = Model(v=123)
+    assert m.v == 123
+
+    m = Model(v='123')
+    assert m.v == '123'
+
+    m = Model(v=b'foobar')
+    assert m.v == 'foobar'
+
+    # here both validators work and it's impossible to work out which value "closer"
+    m = Model(v=12.2)
+    assert m.v == '12.2'
+
+    with pytest.raises(ValidationError) as exc_info:
+        Model(v=None)
+    assert exc_info.value.message == '1 error validating input'
+    assert exc_info.value.errors == OrderedDict(
+        [
+            ('v', [{'type': 'TypeError', 'route': 'int', 'validator': 'int',
+                    'msg': "int() argument must be a string, a bytes-like object or a number, not 'NoneType'"},
+                   {'type': 'TypeError', 'route': 'str', 'msg': 'None is not an allow value',
+                    'validator': 'not_none_validator'}])
+        ])