# coding=utf-8
# ------------------------------------
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
# ------------------------------------

import pytest
import platform
import functools

from azure.core.exceptions import HttpResponseError, ClientAuthenticationError
from azure.core.credentials import AzureKeyCredential
from testcase import GlobalTextAnalyticsAccountPreparer
from asynctestcase import AsyncTextAnalyticsTest
from testcase import TextAnalyticsClientPreparer as _TextAnalyticsClientPreparer
from azure.ai.textanalytics.aio import TextAnalyticsClient

# pre-apply the client_cls positional argument so it needn't be explicitly passed below
# the first one
TextAnalyticsClientPreparer = functools.partial(_TextAnalyticsClientPreparer, TextAnalyticsClient)

class TestEncoding(AsyncTextAnalyticsTest):
    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_emoji(self, client):
        result = await client.recognize_pii_entities(["👩 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 7)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_emoji_with_skin_tone_modifier(self, client):
        result = await client.recognize_pii_entities(["👩🏻 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 8)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_emoji_family(self, client):
        result = await client.recognize_pii_entities(["👩‍👩‍👧‍👧 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 13)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_emoji_family_with_skin_tone_modifier(self, client):
        result = await client.recognize_pii_entities(["👩🏻‍👩🏽‍👧🏾‍👦🏿 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 17)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_diacritics_nfc(self, client):
        result = await client.recognize_pii_entities(["año SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 9)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_diacritics_nfd(self, client):
        result = await client.recognize_pii_entities(["año SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 10)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_korean_nfc(self, client):
        result = await client.recognize_pii_entities(["아가 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 8)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_korean_nfd(self, client):
        result = await client.recognize_pii_entities(["아가 SSN: 859-98-0987"])
        self.assertEqual(result[0].entities[0].offset, 8)

    @GlobalTextAnalyticsAccountPreparer()
    @TextAnalyticsClientPreparer()
    async def test_zalgo_text(self, client):
        result = await client.recognize_pii_entities(["ơ̵̧̧̢̳̘̘͕͔͕̭̟̙͎͈̞͔̈̇̒̃͋̇̅͛̋͛̎́͑̄̐̂̎͗͝m̵͍͉̗̄̏͌̂̑̽̕͝͠g̵̢̡̢̡̨̡̧̛͉̞̯̠̤̣͕̟̫̫̼̰͓̦͖̣̣͎̋͒̈́̓̒̈̍̌̓̅͑̒̓̅̅͒̿̏́͗̀̇͛̏̀̈́̀̊̾̀̔͜͠͝ͅ SSN: 859-98-0987"])


        self.assertEqual(result[0].entities[0].offset, 121)