def test_method_one_line_empty_body(): cpp_code_in = """\ void A::B() {}""" cpp_code_out = """\ #include "MyInclude.h" void A::B() {} """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_class_definition(): cpp_code_in = """\ class TeamCityPrinter : public ::testing::EmptyTestEventListener { };""" cpp_code_out = """\ #include "MyInclude.h" class TeamCityPrinter : public ::testing::EmptyTestEventListener { }; """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_one_line_open_bracket_same_line(): cpp_code_in = """\ void A::B() { }""" cpp_code_out = """\ #include "MyInclude.h" void A::B() { MACRO_INITIAL(); } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_destructor(): cpp_code_in = """\ A::~A(){ }""" cpp_code_out = """\ #include "MyInclude.h" A::~A(){ } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_with_comments(): cpp_code_in = """\ A::B(int foo) //{ //}""" cpp_code_out = """\ #include "MyInclude.h" A::B(int foo) //{ //} """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_one_line_blank_lines_before(): cpp_code_in = """\ void A::B() { }""" cpp_code_out = """\ #include "MyInclude.h" void A::B() { MACRO_INITIAL(); } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_with_initialization_and_trailing_spaces(): cpp_code_in = """\ A::B(int foo): _member(0) { }""" cpp_code_out = """\ #include "MyInclude.h" A::B(int foo): _member(0) { MACRO_INITIAL(); } """ assert process_code(cpp_code_in) == cpp_code_out
def test_namespace(): cpp_code_in = """\ static const int = J::K(); namespace { A::B(int foo) { }""" cpp_code_out = """\ #include "MyInclude.h" static const int = J::K(); namespace { A::B(int foo) { MACRO_INITIAL(); } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_extra_bracket(): cpp_code_in = """\ A::B() { for ( , , ){ } }""" cpp_code_out = """\ #include "MyInclude.h" A::B() { MACRO_INITIAL(); for ( , , ){ } } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_several_lines(): cpp_code_in = """\ void A::B(int foo, int bar) { }""" cpp_code_out = """\ #include "MyInclude.h" void A::B(int foo, int bar) { MACRO_INITIAL(); } """ assert process_code(cpp_code_in) == cpp_code_out
def test_method_constructor_and_destructor(): cpp_code_in = """\ A::A() { } A::~A(){ }""" cpp_code_out = """\ #include "MyInclude.h" A::A() { MACRO_INITIAL(); } A::~A(){ } """ assert process_code(cpp_code_in) == cpp_code_out