from __future__ import print_function



import unittest2
import os.path
import re
import sys

from lldbgdbserverutils import *


class TestLldbGdbServerUtils(unittest2.TestCase):
    def test_entry_exact_payload_match(self):
        entry = GdbRemoteEntry(is_send_to_remote=False, exact_payload="$OK#9a")
        entry.assert_match(self, "$OK#9a")

    def test_entry_exact_payload_match_ignores_checksum(self):
        entry = GdbRemoteEntry(is_send_to_remote=False, exact_payload="$OK#9a")
        entry.assert_match(self, "$OK#00")

    def test_entry_creates_context(self):
        entry = GdbRemoteEntry(is_send_to_remote=False, exact_payload="$OK#9a")
        context = entry.assert_match(self, "$OK#9a")
        self.assertIsNotNone(context)

    def test_entry_regex_matches(self):
        entry = GdbRemoteEntry(is_send_to_remote=False, regex=re.compile(r"^\$QC([0-9a-fA-F]+)#"), capture={ 1:"thread_id" })
        context = entry.assert_match(self, "$QC980#00")

    def test_entry_regex_saves_match(self):
        entry = GdbRemoteEntry(is_send_to_remote=False, regex=re.compile(r"^\$QC([0-9a-fA-F]+)#"), capture={ 1:"thread_id" })
        context = entry.assert_match(self, "$QC980#00")
        self.assertEqual(context["thread_id"], "980")

    def test_entry_regex_expect_captures_success(self):
        context = { "thread_id":"980" }
        entry = GdbRemoteEntry(is_send_to_remote=False, regex=re.compile(r"^\$T([0-9a-fA-F]{2})thread:([0-9a-fA-F]+)"), expect_captures={ 2:"thread_id" })
        entry.assert_match(self, "$T11thread:980;", context=context)

    def test_entry_regex_expect_captures_raises_on_fail(self):
        context = { "thread_id":"980" }
        entry = GdbRemoteEntry(is_send_to_remote=False, regex=re.compile(r"^\$T([0-9a-fA-F]{2})thread:([0-9a-fA-F]+)"), expect_captures={ 2:"thread_id" })
        try:
            entry.assert_match(self, "$T11thread:970;", context=context)
            self.fail()
        except AssertionError:
            # okay
            return None
