message-id tests

.travis.yml

@@ -16,9 +16,6 @@ before_install:
 install:
   - travis_retry pip install -r requirements/tests-$TRAVIS_PYTHON_VERSION.txt
 
-env:
-  - PIP_DOWNLOAD_CACHE=$HOME/.pip-cache
-
 cache:
   directories:
     - $HOME/.pip-cache/

emails/testsuite/message/test_message.py

@@ -5,7 +5,7 @@ from dateutil.parser import parse as dateutil_parse
 import pytest
 import emails
 import emails.exc
-from emails.compat import to_unicode, NativeStringIO
+from emails.compat import to_unicode, StringIO
 from .helpers import common_email_data
 
 
@@ -21,7 +21,7 @@ def test_message_build():
         emails.Message().as_string()
 
     # Test file-like html and text
-    m = emails.Message(html=NativeStringIO('X'), text=NativeStringIO('Y'))
+    m = emails.Message(html=StringIO('X'), text=StringIO('Y'))
     assert m.html == 'X'
     assert m.text == 'Y'
 

emails/testsuite/test_utils.py

@@ -2,7 +2,7 @@
 from __future__ import unicode_literals
 import pytest
 from emails.utils import (parse_name_and_email,
-    encode_header, decode_header, sanitize_address, fetch_url)
+    encode_header, decode_header, sanitize_address, fetch_url, MessageID)
 from emails.exc import HTTPLoaderError
 
 def test_parse_name_and_email():
@@ -34,3 +34,21 @@ def test_fetch_url():
     with pytest.raises(HTTPLoaderError):
         fetch_url('http://google.com/nonexistent-no-page')
 
+
+def test_message_id():
+    # Test message-id generate
+    assert MessageID()()
+    assert '___xxx___' in MessageID(idstring='___xxx___')()
+    assert '___yyy___' in MessageID(domain='___yyy___')()
+
+    # Test message-id generate
+    _ids = set()
+    gen = MessageID()
+    for _ in range(100):
+        _id = gen()
+        if len(_ids) == 1:
+            _ids.add(_id)
+            continue
+        else:
+            assert _id not in _ids
+            _ids.add(_id)