import difflib

import inspect

import re

import unittest

from code import compile_command as compiler

from codeop import CommandCompiler

from functools import partial

from bpython.curtsiesfrontend.interpreter import code_finished_will_parse

from bpython.curtsiesfrontend.preprocess import preprocess

from bpython.test.fodder import original, processed

preproc = partial(preprocess, compiler=CommandCompiler())

def get_fodder_source(test_name):

pattern = rf"#StartTest-{test_name}\n(.*?)#EndTest"

orig, xformed = (

re.search(pattern, inspect.getsource(module), re.DOTALL)

for module in [original, processed]

)

if not orig:

raise ValueError(

f"Can't locate test {test_name} in original fodder file"

)

if not xformed:

raise ValueError(

f"Can't locate test {test_name} in processed fodder file"

)

return orig.group(1), xformed.group(1)

class TestPreprocessing(unittest.TestCase):

def assertCompiles(self, source):

finished, parsable = code_finished_will_parse(source, compiler)

return finished and parsable

def test_indent_empty_lines_nops(self):

self.assertEqual(preproc("hello"), "hello")

self.assertEqual(preproc("hello\ngoodbye"), "hello\ngoodbye")

self.assertEqual(preproc("a\n b\nc\n"), "a\n b\nc\n")

def assertShowWhitespaceEqual(self, a, b):

self.assertEqual(

a,

b,

"".join(

difflib.context_diff(

a.replace(" ", "~").splitlines(True),

b.replace(" ", "~").splitlines(True),

fromfile="actual",

tofile="expected",

n=5,

)

),

)

def assertDefinitionIndented(self, obj):

name = obj.__name__

obj2 = getattr(processed, name)

orig = inspect.getsource(obj)

xformed = inspect.getsource(obj2)

self.assertShowWhitespaceEqual(preproc(orig), xformed)

self.assertCompiles(xformed)

def assertLinesIndented(self, test_name):

orig, xformed = get_fodder_source(test_name)

self.assertShowWhitespaceEqual(preproc(orig), xformed)

self.assertCompiles(xformed)

def assertIndented(self, obj_or_name):

if isinstance(obj_or_name, str):

self.assertLinesIndented(obj_or_name)

else:

self.assertDefinitionIndented(obj_or_name)

def test_empty_line_between_methods(self):

self.assertIndented(original.BlankLineBetweenMethods)

def test_empty_line_within_class(self):

self.assertIndented(original.BlankLineInFunction)

def test_blank_lines_in_for_loop(self):

self.assertIndented("blank_lines_in_for_loop")

@unittest.skip(

"More advanced technique required: need to try compiling and "

"backtracking"

)

def test_blank_line_in_try_catch(self):

self.assertIndented("blank_line_in_try_catch")

@unittest.skip(

"More advanced technique required: need to try compiling and "

"backtracking"

)

def test_blank_line_in_try_catch_else(self):

self.assertIndented("blank_line_in_try_catch_else")

def test_blank_trailing_line(self):

self.assertIndented("blank_trailing_line")

def test_tabs(self):

self.assertIndented(original.tabs)