userstring is a wrapper around the native builtin string type userstring instances should behave similar to builtin string objects overwrite the three testing methods because userstring can t cope with arguments propagated to userstring and we don t test with subclasses we don t fix the arguments because userstring can t cope with it we don t fix the arguments because userstring can t cope with it we don t fix the arguments because userstring can t cope with it check that encoding defaults to utf8 check that errors defaults to strict check that encoding defaults to utf8 check that errors defaults to strict userstring is a wrapper around the native builtin string type userstring instances should behave similar to builtin string objects overwrite the three testing methods because userstring can t cope with arguments propagated to userstring and we don t test with subclasses we don t fix the arguments because userstring can t cope with it we don t fix the arguments because userstring can t cope with it we don t fix the arguments because userstring can t cope with it check that encoding defaults to utf 8 check that errors defaults to strict check that encoding defaults to utf 8 check that errors defaults to strict
import unittest
from test import string_tests

from collections import UserString

class UserStringTest(
    string_tests.StringLikeTest,
    unittest.TestCase
    ):

    type2test = UserString

    
    
    
    def checkequal(self, result, object, methodname, *args, **kwargs):
        result = self.fixtype(result)
        object = self.fixtype(object)
        
        realresult = getattr(object, methodname)(*args, **kwargs)
        self.assertEqual(
            result,
            realresult
        )

    def checkraises(self, exc, obj, methodname, *args, expected_msg=None):
        obj = self.fixtype(obj)
        
        with self.assertRaises(exc) as cm:
            getattr(obj, methodname)(*args)
        self.assertNotEqual(str(cm.exception), '')
        if expected_msg is not None:
            self.assertEqual(str(cm.exception), expected_msg)

    def checkcall(self, object, methodname, *args):
        object = self.fixtype(object)
        
        getattr(object, methodname)(*args)

    def test_rmod(self):
        class ustr2(UserString):
            pass

        class ustr3(ustr2):
            def __rmod__(self, other):
                return super().__rmod__(other)

        fmt2 = ustr2('value is %s')
        str3 = ustr3('TEST')
        self.assertEqual(fmt2 % str3, 'value is TEST')

    def test_encode_default_args(self):
        self.checkequal(b'hello', 'hello', 'encode')
        
        self.checkequal(b'\xf0\xa3\x91\x96', '\U00023456', 'encode')
        
        self.checkraises(UnicodeError, '\ud800', 'encode')

    def test_encode_explicit_none_args(self):
        self.checkequal(b'hello', 'hello', 'encode', None, None)
        
        self.checkequal(b'\xf0\xa3\x91\x96', '\U00023456', 'encode', None, None)
        
        self.checkraises(UnicodeError, '\ud800', 'encode', None, None)


if __name__ == "__main__":
    unittest.main()