<?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 CodegenMethodTestCase extends CodegenBaseTest {

  public function testSimpleGetter() {
    $code = codegen_method('getName')
      ->setReturnType('string')
      ->setBody('return $this->name;')
      ->setDocBlock('Return the name of the user.')
      ->render();

    self::assertUnchanged($code);
  }

  public function testAbstractProtectedAndParams() {
    $code = codegen_method('getSchema')
      ->addParameter('string $name')
      ->setIsAbstract()
      ->setProtected()
      ->render();

    self::assertUnchanged($code);
 }

  public function testAsync() {
    $code = codegen_method('genFoo')
      ->setIsAsync()
      ->render();

    self::assertUnchanged($code);
  }

  public function testPrivateAndStaticWithEmptyBody() {
    $code = codegen_method('doNothing')
      ->setIsStatic()
      ->setPrivate()
      ->render();

    self::assertUnchanged($code);
  }

  public function testManualSection() {
    $method = codegen_method('genProprietorName')
      ->setReturnType('string')
      ->setBody('// insert your code here')
      ->setManualBody();

    codegen_class(EntBlindPig::class)->addMethod($method);
    $code = $method->render();

    self::assertUnchanged($code);
  }

  public function testConstructor() {
    $code = codegen_constructor()
      ->addParameter('string $name')
      ->setBody('$this->name = $name;')
      ->render();

    self::assertUnchanged($code);
  }

  public function testDocBlockCommentsWrap() {
    // 1-3 characters in doc block account for ' * ' in this test.
    $code = codegen_method('getName')
      ->setReturnType('string')
      ->setBody('return $this->name;')
      // 81 characters
      ->setDocBlock(str_repeat('x', 78))
      ->setGeneratedFrom(
        codegen_generated_from_method_with_key(
          'EntTestSchema',
          'getFields',
          'name'
        )
      )->render();

    self::assertUnchanged($code);
  }
}