fragment/src/androidTest/java/androidx/fragment/app/test/FragmentTestActivity.java - platform/frameworks/support - Git at Google

 /*
  * Copyright 2018 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
  *
  *      http://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 androidx.fragment.app.test;

 import static org.junit.Assert.assertFalse;

 import android.app.Activity;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Build.VERSION;
 import android.os.Build.VERSION_CODES;
 import android.os.Bundle;
 import android.transition.Transition;
 import android.transition.Transition.TransitionListener;
 import android.transition.TransitionInflater;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;

 import androidx.annotation.Nullable;
 import androidx.fragment.app.Fragment;
 import androidx.fragment.app.FragmentActivity;
 import androidx.fragment.test.R;

 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;

 /**
  * A simple activity used for Fragment Transitions and lifecycle event ordering
  */
 public class FragmentTestActivity extends FragmentActivity {
     public final CountDownLatch onDestroyLatch = new CountDownLatch(1);

     @Override
     public void onCreate(Bundle icicle) {
         super.onCreate(icicle);
         setContentView(R.layout.activity_content);
         Intent intent = getIntent();
         if (intent != null && intent.getBooleanExtra("finishEarly", false)) {
             finish();
             getSupportFragmentManager().beginTransaction()
                     .add(new AssertNotDestroyed(), "not destroyed")
                     .commit();
         }
     }

     @Override
     protected void onDestroy() {
         super.onDestroy();
         onDestroyLatch.countDown();
     }

     public static class TestFragment extends Fragment {
         public static final int ENTER = 0;
         public static final int RETURN = 1;
         public static final int EXIT = 2;
         public static final int REENTER = 3;
         public static final int SHARED_ELEMENT_ENTER = 4;
         public static final int SHARED_ELEMENT_RETURN = 5;
         private static final int TRANSITION_COUNT = 6;

         private static final String LAYOUT_ID = "layoutId";
         private static final String TRANSITION_KEY = "transition_";
         private int mLayoutId = R.layout.fragment_start;
         private final int[] mTransitionIds = new int[] {
                 R.transition.fade,
                 R.transition.fade,
                 R.transition.fade,
                 R.transition.fade,
                 R.transition.change_bounds,
                 R.transition.change_bounds,
         };
         private final Object[] mListeners = new Object[TRANSITION_COUNT];

         public TestFragment() {
             if (VERSION.SDK_INT >= VERSION_CODES.KITKAT) {
                 for (int i = 0; i < TRANSITION_COUNT; i++) {
                     mListeners[i] = new TransitionCalledListener();
                 }
             }
         }

         public static TestFragment create(int layoutId) {
             TestFragment testFragment = new TestFragment();
             testFragment.mLayoutId = layoutId;
             return testFragment;
         }

         public void clearTransitions() {
             for (int i = 0; i < TRANSITION_COUNT; i++) {
                 mTransitionIds[i] = 0;
             }
         }

         public void clearNotifications() {
             for (int i = 0; i < TRANSITION_COUNT; i++) {
                 ((TransitionCalledListener)mListeners[i]).startLatch = new CountDownLatch(1);
                 ((TransitionCalledListener)mListeners[i]).endLatch = new CountDownLatch(1);
             }
         }

         @Override
         public void onCreate(Bundle savedInstanceState) {
             super.onCreate(savedInstanceState);
             if (savedInstanceState != null) {
                 mLayoutId = savedInstanceState.getInt(LAYOUT_ID, mLayoutId);
                 for (int i = 0; i < TRANSITION_COUNT; i++) {
                     String key = TRANSITION_KEY + i;
                     mTransitionIds[i] = savedInstanceState.getInt(key, mTransitionIds[i]);
                 }
             }
         }

         @Override
         public void onSaveInstanceState(Bundle outState) {
             super.onSaveInstanceState(outState);
             outState.putInt(LAYOUT_ID, mLayoutId);
             for (int i = 0; i < TRANSITION_COUNT; i++) {
                 String key = TRANSITION_KEY + i;
                 outState.putInt(key, mTransitionIds[i]);
             }
         }

         @Override
         public View onCreateView(LayoutInflater inflater, ViewGroup container,
                 Bundle savedInstanceState) {
             return inflater.inflate(mLayoutId, container, false);
         }

         @SuppressWarnings("deprecation")
         @Override
         public void onAttach(Activity activity) {
             super.onAttach(activity);
             if (VERSION.SDK_INT > VERSION_CODES.KITKAT) {
                 setEnterTransition(loadTransition(ENTER));
                 setReenterTransition(loadTransition(REENTER));
                 setExitTransition(loadTransition(EXIT));
                 setReturnTransition(loadTransition(RETURN));
                 setSharedElementEnterTransition(loadTransition(SHARED_ELEMENT_ENTER));
                 setSharedElementReturnTransition(loadTransition(SHARED_ELEMENT_RETURN));
             }
         }

         public boolean wasStartCalled(int transitionKey) {
             return ((TransitionCalledListener)mListeners[transitionKey]).startLatch.getCount() == 0;
         }

         public boolean wasEndCalled(int transitionKey) {
             return ((TransitionCalledListener)mListeners[transitionKey]).endLatch.getCount() == 0;
         }

         public boolean waitForStart(int transitionKey)
                 throws InterruptedException {
             TransitionCalledListener l = ((TransitionCalledListener)mListeners[transitionKey]);
             return l.startLatch.await(500,TimeUnit.MILLISECONDS);
         }

         public boolean waitForEnd(int transitionKey)
                 throws InterruptedException {
             TransitionCalledListener l = ((TransitionCalledListener)mListeners[transitionKey]);
             return l.endLatch.await(500,TimeUnit.MILLISECONDS);
         }

         private Transition loadTransition(int key) {
             final int id = mTransitionIds[key];
             if (id == 0) {
                 return null;
             }
             Transition transition = TransitionInflater.from(getActivity()).inflateTransition(id);
             transition.addListener(((TransitionCalledListener)mListeners[key]));
             return transition;
         }

         private class TransitionCalledListener implements TransitionListener {
             public CountDownLatch startLatch = new CountDownLatch(1);
             public CountDownLatch endLatch = new CountDownLatch(1);

             public TransitionCalledListener() {
             }

             @Override
             public void onTransitionStart(Transition transition) {
                 startLatch.countDown();
             }

             @Override
             public void onTransitionEnd(Transition transition) {
                 endLatch.countDown();
             }

             @Override
             public void onTransitionCancel(Transition transition) {
             }

             @Override
             public void onTransitionPause(Transition transition) {
             }

             @Override
             public void onTransitionResume(Transition transition) {
             }
         }
     }

     public static class ParentFragment extends Fragment {
         static final String CHILD_FRAGMENT_TAG = "childFragment";
         public boolean wasAttachedInTime;

         private boolean mRetainChild;

         @Override
         public void onCreate(Bundle savedInstanceState) {
             super.onCreate(savedInstanceState);

             ChildFragment f = getChildFragment();
             if (f == null) {
                 f = new ChildFragment();
                 if (mRetainChild) {
                     f.setRetainInstance(true);
                 }
                 getChildFragmentManager().beginTransaction().add(f, CHILD_FRAGMENT_TAG).commitNow();
             }
             wasAttachedInTime = f.attached;
         }

         public ChildFragment getChildFragment() {
             return (ChildFragment) getChildFragmentManager().findFragmentByTag(CHILD_FRAGMENT_TAG);
         }

         public void setRetainChildInstance(boolean retainChild) {
             mRetainChild = retainChild;
         }
     }

     public static class ChildFragment extends Fragment {
         private OnAttachListener mOnAttachListener;

         public boolean attached;
         public boolean onActivityResultCalled;
         public int onActivityResultRequestCode;
         public int onActivityResultResultCode;

         @Override
         public void onAttach(Context activity) {
             super.onAttach(activity);
             attached = true;
             if (mOnAttachListener != null) {
                 mOnAttachListener.onAttach(activity, this);
             }
         }

         public void setOnAttachListener(OnAttachListener listener) {
             mOnAttachListener = listener;
         }

         public interface OnAttachListener {
             void onAttach(Context activity, ChildFragment fragment);
         }

         @Override
         public void onActivityResult(int requestCode, int resultCode, Intent data) {
             onActivityResultCalled = true;
             onActivityResultRequestCode = requestCode;
             onActivityResultResultCode = resultCode;
         }
     }

     public static class AssertNotDestroyed extends Fragment {
         @Override
         public void onActivityCreated(@Nullable Bundle savedInstanceState) {
             super.onActivityCreated(savedInstanceState);
             if (VERSION.SDK_INT >= VERSION_CODES.JELLY_BEAN_MR1) {
                 assertFalse(getActivity().isDestroyed());
             }
         }
     }
 }
	/*
	* Copyright 2018 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
	*
	* http://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 androidx.fragment.app.test;

	import static org.junit.Assert.assertFalse;

	import android.app.Activity;
	import android.content.Context;
	import android.content.Intent;
	import android.os.Build.VERSION;
	import android.os.Build.VERSION_CODES;
	import android.os.Bundle;
	import android.transition.Transition;
	import android.transition.Transition.TransitionListener;
	import android.transition.TransitionInflater;
	import android.view.LayoutInflater;
	import android.view.View;
	import android.view.ViewGroup;

	import androidx.annotation.Nullable;
	import androidx.fragment.app.Fragment;
	import androidx.fragment.app.FragmentActivity;
	import androidx.fragment.test.R;

	import java.util.concurrent.CountDownLatch;
	import java.util.concurrent.TimeUnit;

	/**
	* A simple activity used for Fragment Transitions and lifecycle event ordering
	*/
	public class FragmentTestActivity extends FragmentActivity {
	public final CountDownLatch onDestroyLatch = new CountDownLatch(1);

	@Override
	public void onCreate(Bundle icicle) {
	super.onCreate(icicle);
	setContentView(R.layout.activity_content);
	Intent intent = getIntent();
	if (intent != null && intent.getBooleanExtra("finishEarly", false)) {
	finish();
	getSupportFragmentManager().beginTransaction()
	.add(new AssertNotDestroyed(), "not destroyed")
	.commit();
	}
	}

	@Override
	protected void onDestroy() {
	super.onDestroy();
	onDestroyLatch.countDown();
	}

	public static class TestFragment extends Fragment {
	public static final int ENTER = 0;
	public static final int RETURN = 1;
	public static final int EXIT = 2;
	public static final int REENTER = 3;
	public static final int SHARED_ELEMENT_ENTER = 4;
	public static final int SHARED_ELEMENT_RETURN = 5;
	private static final int TRANSITION_COUNT = 6;

	private static final String LAYOUT_ID = "layoutId";
	private static final String TRANSITION_KEY = "transition_";
	private int mLayoutId = R.layout.fragment_start;
	private final int[] mTransitionIds = new int[] {
	R.transition.fade,
	R.transition.fade,
	R.transition.fade,
	R.transition.fade,
	R.transition.change_bounds,
	R.transition.change_bounds,
	};
	private final Object[] mListeners = new Object[TRANSITION_COUNT];

	public TestFragment() {
	if (VERSION.SDK_INT >= VERSION_CODES.KITKAT) {
	for (int i = 0; i < TRANSITION_COUNT; i++) {
	mListeners[i] = new TransitionCalledListener();
	}
	}
	}

	public static TestFragment create(int layoutId) {
	TestFragment testFragment = new TestFragment();
	testFragment.mLayoutId = layoutId;
	return testFragment;
	}

	public void clearTransitions() {
	for (int i = 0; i < TRANSITION_COUNT; i++) {
	mTransitionIds[i] = 0;
	}
	}

	public void clearNotifications() {
	for (int i = 0; i < TRANSITION_COUNT; i++) {
	((TransitionCalledListener)mListeners[i]).startLatch = new CountDownLatch(1);
	((TransitionCalledListener)mListeners[i]).endLatch = new CountDownLatch(1);
	}
	}

	@Override
	public void onCreate(Bundle savedInstanceState) {
	super.onCreate(savedInstanceState);
	if (savedInstanceState != null) {
	mLayoutId = savedInstanceState.getInt(LAYOUT_ID, mLayoutId);
	for (int i = 0; i < TRANSITION_COUNT; i++) {
	String key = TRANSITION_KEY + i;
	mTransitionIds[i] = savedInstanceState.getInt(key, mTransitionIds[i]);
	}
	}
	}

	@Override
	public void onSaveInstanceState(Bundle outState) {
	super.onSaveInstanceState(outState);
	outState.putInt(LAYOUT_ID, mLayoutId);
	for (int i = 0; i < TRANSITION_COUNT; i++) {
	String key = TRANSITION_KEY + i;
	outState.putInt(key, mTransitionIds[i]);
	}
	}

	@Override
	public View onCreateView(LayoutInflater inflater, ViewGroup container,
	Bundle savedInstanceState) {
	return inflater.inflate(mLayoutId, container, false);
	}

	@SuppressWarnings("deprecation")
	@Override
	public void onAttach(Activity activity) {
	super.onAttach(activity);
	if (VERSION.SDK_INT > VERSION_CODES.KITKAT) {
	setEnterTransition(loadTransition(ENTER));
	setReenterTransition(loadTransition(REENTER));
	setExitTransition(loadTransition(EXIT));
	setReturnTransition(loadTransition(RETURN));
	setSharedElementEnterTransition(loadTransition(SHARED_ELEMENT_ENTER));
	setSharedElementReturnTransition(loadTransition(SHARED_ELEMENT_RETURN));
	}
	}

	public boolean wasStartCalled(int transitionKey) {
	return ((TransitionCalledListener)mListeners[transitionKey]).startLatch.getCount() == 0;
	}

	public boolean wasEndCalled(int transitionKey) {
	return ((TransitionCalledListener)mListeners[transitionKey]).endLatch.getCount() == 0;
	}

	public boolean waitForStart(int transitionKey)
	throws InterruptedException {
	TransitionCalledListener l = ((TransitionCalledListener)mListeners[transitionKey]);
	return l.startLatch.await(500,TimeUnit.MILLISECONDS);
	}

	public boolean waitForEnd(int transitionKey)
	throws InterruptedException {
	TransitionCalledListener l = ((TransitionCalledListener)mListeners[transitionKey]);
	return l.endLatch.await(500,TimeUnit.MILLISECONDS);
	}

	private Transition loadTransition(int key) {
	final int id = mTransitionIds[key];
	if (id == 0) {
	return null;
	}
	Transition transition = TransitionInflater.from(getActivity()).inflateTransition(id);
	transition.addListener(((TransitionCalledListener)mListeners[key]));
	return transition;
	}

	private class TransitionCalledListener implements TransitionListener {
	public CountDownLatch startLatch = new CountDownLatch(1);
	public CountDownLatch endLatch = new CountDownLatch(1);

	public TransitionCalledListener() {
	}

	@Override
	public void onTransitionStart(Transition transition) {
	startLatch.countDown();
	}

	@Override
	public void onTransitionEnd(Transition transition) {
	endLatch.countDown();
	}

	@Override
	public void onTransitionCancel(Transition transition) {
	}

	@Override
	public void onTransitionPause(Transition transition) {
	}

	@Override
	public void onTransitionResume(Transition transition) {
	}
	}
	}

	public static class ParentFragment extends Fragment {
	static final String CHILD_FRAGMENT_TAG = "childFragment";
	public boolean wasAttachedInTime;

	private boolean mRetainChild;

	@Override
	public void onCreate(Bundle savedInstanceState) {
	super.onCreate(savedInstanceState);

	ChildFragment f = getChildFragment();
	if (f == null) {
	f = new ChildFragment();
	if (mRetainChild) {
	f.setRetainInstance(true);
	}
	getChildFragmentManager().beginTransaction().add(f, CHILD_FRAGMENT_TAG).commitNow();
	}
	wasAttachedInTime = f.attached;
	}

	public ChildFragment getChildFragment() {
	return (ChildFragment) getChildFragmentManager().findFragmentByTag(CHILD_FRAGMENT_TAG);
	}

	public void setRetainChildInstance(boolean retainChild) {
	mRetainChild = retainChild;
	}
	}

	public static class ChildFragment extends Fragment {
	private OnAttachListener mOnAttachListener;

	public boolean attached;
	public boolean onActivityResultCalled;
	public int onActivityResultRequestCode;
	public int onActivityResultResultCode;

	@Override
	public void onAttach(Context activity) {
	super.onAttach(activity);
	attached = true;
	if (mOnAttachListener != null) {
	mOnAttachListener.onAttach(activity, this);
	}
	}

	public void setOnAttachListener(OnAttachListener listener) {
	mOnAttachListener = listener;
	}

	public interface OnAttachListener {
	void onAttach(Context activity, ChildFragment fragment);
	}

	@Override
	public void onActivityResult(int requestCode, int resultCode, Intent data) {
	onActivityResultCalled = true;
	onActivityResultRequestCode = requestCode;
	onActivityResultResultCode = resultCode;
	}
	}

	public static class AssertNotDestroyed extends Fragment {
	@Override
	public void onActivityCreated(@Nullable Bundle savedInstanceState) {
	super.onActivityCreated(savedInstanceState);
	if (VERSION.SDK_INT >= VERSION_CODES.JELLY_BEAN_MR1) {
	assertFalse(getActivity().isDestroyed());
	}
	}
	}
	}