#!/usr/bin/env python

#

# test_codecencodings_kr.py

#   Codec encoding tests for ROK encodings.

#

from test import test_support

from test import test_multibytecodec_support

import unittest

class Test_CP949(test_multibytecodec_support.TestBase, unittest.TestCase):

    encoding = 'cp949'

    tstring = test_multibytecodec_support.load_teststring('cp949')

    codectests = (

        # invalid bytes

        ("abc\x80\x80\xc1\xc4", "strict",  None),

        ("abc\xc8", "strict",  None),

        ("abc\x80\x80\xc1\xc4", "replace", u"abc\ufffd\uc894"),

        ("abc\x80\x80\xc1\xc4\xc8", "replace", u"abc\ufffd\uc894\ufffd"),

        ("abc\x80\x80\xc1\xc4", "ignore",  u"abc\uc894"),

    )

class Test_EUCKR(test_multibytecodec_support.TestBase, unittest.TestCase):

    encoding = 'euc_kr'

    tstring = test_multibytecodec_support.load_teststring('euc_kr')

    codectests = (

        # invalid bytes

        ("abc\x80\x80\xc1\xc4", "strict",  None),

        ("abc\xc8", "strict",  None),

        ("abc\x80\x80\xc1\xc4", "replace", u"abc\ufffd\uc894"),

        ("abc\x80\x80\xc1\xc4\xc8", "replace", u"abc\ufffd\uc894\ufffd"),

        ("abc\x80\x80\xc1\xc4", "ignore",  u"abc\uc894"),

        # composed make-up sequence errors

        ("\xa4\xd4", "strict", None),

        ("\xa4\xd4\xa4", "strict", None),

        ("\xa4\xd4\xa4\xb6", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa4", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa4\xd0", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa4\xd0\xa4", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa4\xd0\xa4\xd4", "strict", u"\uc4d4"),

        ("\xa4\xd4\xa4\xb6\xa4\xd0\xa4\xd4x", "strict", u"\uc4d4x"),

        ("a\xa4\xd4\xa4\xb6\xa4", "replace", u"a\ufffd"),

        ("\xa4\xd4\xa3\xb6\xa4\xd0\xa4\xd4", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa3\xd0\xa4\xd4", "strict", None),

        ("\xa4\xd4\xa4\xb6\xa4\xd0\xa3\xd4", "strict", None),

        ("\xa4\xd4\xa4\xff\xa4\xd0\xa4\xd4", "replace", u"\ufffd"),

        ("\xa4\xd4\xa4\xb6\xa4\xff\xa4\xd4", "replace", u"\ufffd"),

        ("\xa4\xd4\xa4\xb6\xa4\xd0\xa4\xff", "replace", u"\ufffd"),

        ("\xc1\xc4", "strict", u"\uc894"),

    )

class Test_JOHAB(test_multibytecodec_support.TestBase, unittest.TestCase):

    encoding = 'johab'

    tstring = test_multibytecodec_support.load_teststring('johab')

    codectests = (

        # invalid bytes

        ("abc\x80\x80\xc1\xc4", "strict",  None),

        ("abc\xc8", "strict",  None),

        ("abc\x80\x80\xc1\xc4", "replace", u"abc\ufffd\ucd27"),

        ("abc\x80\x80\xc1\xc4\xc8", "replace", u"abc\ufffd\ucd27\ufffd"),

        ("abc\x80\x80\xc1\xc4", "ignore",  u"abc\ucd27"),

    )

def test_main():

    test_support.run_unittest(__name__)

if __name__ == "__main__":

    test_main()