<?hh

/**

* Copyright (c) 2015-present, Facebook, Inc.

* All rights reserved.

*

* This source code is licensed under the BSD-style license found in the

* LICENSE file in the root directory of this source tree. An additional grant

* of patent rights can be found in the PATENTS file in the same directory.

*/

final class RefactorCodegenTestCase extends CodegenBaseTest {

/**

* In this example, the contents of OldClass get put into NewClass.

* OldClass had some partially generated code that's going to get

* transported into the like sections of NewClass.

*/

public function testClassRename() {

$old_file_name = Filesystem::createTemporaryFile('codegen', true);

$old_class = codegen_class('OldClass')

->setHasManualDeclarations(true, null, "// Let's see if this shows up")

->setHasManualMethodSection(true, null, "// Will this also show up?");

$codegen_old_file = test_codegen_file($old_file_name)

->addClass($old_class);

$codegen_old_file->save();

$new_file_name = Filesystem::createTemporaryFile('codegen', true);

$new_class = codegen_class('NewClass')

->setHasManualMethodSection()

->setHasManualDeclarations();

$codegen_new_file = test_codegen_file($new_file_name)

->addClass($new_class)

->addOriginalFile($old_file_name)

->rekeyManualSection('OldClass_header', 'NewClass_header')

->rekeyManualSection('OldClass_footer', 'NewClass_footer');

self::assertUnchanged($codegen_new_file->render());

}

/**

* Same as testClassRename, but we're going to combine the contents

* of the manual sections into one section instead of two.

*/

public function testManualSectionMerge() {

$old_file_name = Filesystem::createTemporaryFile('codegen', true);

$old_class = codegen_class('OldClass')

->setHasManualDeclarations(true, null, "// Let's see if this shows up")

->setHasManualMethodSection(true, null, "// Will this also show up?");

$codegen_old_file = test_codegen_file($old_file_name)

->addClass($old_class);

$codegen_old_file->save();

$new_file_name = Filesystem::createTemporaryFile('codegen', true);

$new_class = codegen_class('NewClass')

->setHasManualMethodSection(true, "NewClass_manual");

$codegen_new_file = test_codegen_file($new_file_name)

->addClass($new_class)

->addOriginalFile($old_file_name)

->rekeyManualSection('OldClass_header', 'NewClass_manual')

->rekeyManualSection('OldClass_footer', 'NewClass_manual');

self::assertUnchanged($codegen_new_file->render());

}

}