/*
 * Copyright 2021 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.example.windowmanagersample;

import static androidx.test.espresso.Espresso.onView;
import static androidx.test.espresso.assertion.ViewAssertions.matches;
import static androidx.test.espresso.matcher.ViewMatchers.isDisplayed;
import static androidx.test.espresso.matcher.ViewMatchers.withSubstring;
import static androidx.window.layout.FoldingFeature.Orientation.HORIZONTAL;
import static androidx.window.layout.FoldingFeature.Orientation.VERTICAL;
import static androidx.window.layout.FoldingFeature.State.FLAT;
import static androidx.window.layout.FoldingFeature.State.HALF_OPENED;
import static androidx.window.testing.layout.DisplayFeatureTesting.createFoldingFeature;

import androidx.test.ext.junit.rules.ActivityScenarioRule;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.window.layout.FoldingFeature;
import androidx.window.layout.WindowLayoutInfo;
import androidx.window.testing.layout.WindowLayoutInfoPublisherRule;
import androidx.window.testing.layout.WindowLayoutInfoTesting;
import java.util.Collections;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.RuleChain;
import org.junit.rules.TestRule;
import org.junit.runner.RunWith;

@RunWith(AndroidJUnit4.class)
public class DisplayFeaturesActivityJavaTest {

    private ActivityScenarioRule<DisplayFeaturesActivity> activityRule =
            new ActivityScenarioRule<>(DisplayFeaturesActivity.class);
    private WindowLayoutInfoPublisherRule publisherRule = new WindowLayoutInfoPublisherRule();

    @Rule public TestRule testRule;

    public DisplayFeaturesActivityJavaTest() {
        testRule = RuleChain.outerRule(publisherRule).around(activityRule);
    };

    @Test
    public void testDeviceOpen_Flat() {
        activityRule
                .getScenario()
                .onActivity(
                        activity -> {
                            FoldingFeature feature =
                                    createFoldingFeature(activity, -1, 0, FLAT, HORIZONTAL);

                            WindowLayoutInfo expected =
                                    WindowLayoutInfoTesting.createWindowLayoutInfo(
                                            Collections.singletonList(feature));

                            publisherRule.overrideWindowLayoutInfo(expected);
                        });

        onView(withSubstring("state = FLAT")).check(matches(isDisplayed()));
        onView(withSubstring("is not separated")).check(matches(isDisplayed()));
        onView(withSubstring("Hinge is horizontal")).check(matches(isDisplayed()));
    }

    @Test
    public void testDeviceOpen_TableTop() {
        activityRule
                .getScenario()
                .onActivity(
                        activity -> {
                            FoldingFeature feature =
                                    createFoldingFeature(activity, -1, 0, HALF_OPENED, HORIZONTAL);

                            WindowLayoutInfo expected =
                                    WindowLayoutInfoTesting.createWindowLayoutInfo(
                                            Collections.singletonList(feature));

                            publisherRule.overrideWindowLayoutInfo(expected);
                        });

        onView(withSubstring("state = HALF_OPENED")).check(matches(isDisplayed()));
        onView(withSubstring("are separated")).check(matches(isDisplayed()));
        onView(withSubstring("Hinge is horizontal")).check(matches(isDisplayed()));
    }

    @Test
    public void testDeviceOpen_Book() { // : Unit = testScope.runBlockingTest {
        activityRule
                .getScenario()
                .onActivity(
                        activity -> {
                            FoldingFeature feature =
                                    createFoldingFeature(activity, -1, 0, HALF_OPENED, VERTICAL);

                            WindowLayoutInfo expected =
                                    WindowLayoutInfoTesting.createWindowLayoutInfo(
                                            Collections.singletonList(feature));

                            publisherRule.overrideWindowLayoutInfo(expected);
                        });

        onView(withSubstring("state = HALF_OPENED")).check(matches(isDisplayed()));
        onView(withSubstring("are separated")).check(matches(isDisplayed()));
        onView(withSubstring("Hinge is vertical")).check(matches(isDisplayed()));
    }
}

@@ -1295,16 +1295,109 @@
 Feature;
+%0Aimport static androidx.window.testing.layout.WindowLayoutInfoTesting.createWindowLayoutInfo;
 %0A%0Aimport
@@ -1670,71 +1670,8 @@
 le;%0A
-import androidx.window.testing.layout.WindowLayoutInfoTesting;%0A
 impo
@@ -2741,100 +2741,31 @@
     
-WindowLayoutInfoTesting.createWindowLayoutInfo(%0A                                            
+createWindowLayoutInfo(
 Coll
@@ -3577,100 +3577,31 @@
     
-WindowLayoutInfoTesting.createWindowLayoutInfo(%0A                                            
+createWindowLayoutInfo(
 Coll
@@ -4451,100 +4451,31 @@
     
-WindowLayoutInfoTesting.createWindowLayoutInfo(%0A                                            
+createWindowLayoutInfo(
 Coll

Move an import in Java test to be static.