/*
 * Copyright (c) 2017-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.
 */

package com.facebook.litho.testing.assertj;

import com.facebook.litho.Component;
import com.facebook.litho.ComponentContext;
import com.facebook.litho.LithoView;
import org.assertj.core.api.Java6Assertions;

/**
 * Common entry point for Litho assertions. Can be statically imported instead of {@link
 * Java6Assertions#assertThat}.
 */
public class LithoAssertions extends Java6Assertions {
  public static ComponentAssert assertThat(ComponentContext c, Component component) {
    return ComponentAssert.assertThat(c, component);
  }

  public static LithoViewAssert assertThat(LithoView lithoView) {
    return LithoViewAssert.assertThat(lithoView);
  }
}

@@ -428,16 +428,62 @@
 ontext;%0A
+import com.facebook.litho.ComponentLifecycle;%0A
 import c
@@ -885,16 +885,185 @@
 );%0A  %7D%0A%0A
+  public static %3CL extends ComponentLifecycle%3E ComponentAssert assertThat(%0A      Component.Builder%3CL, ?%3E builder) %7B%0A    return ComponentAssert.assertThat(builder);%0A  %7D%0A%0A
   public

Add missing entry point to LithoAssertions