FragmentReceiveResultTest.java revision 4285a0db794444695fd0752e7bbc2ad95082043c
17e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri/* 27e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * Copyright (C) 2016 The Android Open Source Project 37e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * 47e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * Licensed under the Apache License, Version 2.0 (the "License"); 57e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * you may not use this file except in compliance with the License. 67e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * You may obtain a copy of the License at 77e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * 87e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * http://www.apache.org/licenses/LICENSE-2.0 97e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * 107e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * Unless required by applicable law or agreed to in writing, software 117e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * distributed under the License is distributed on an "AS IS" BASIS, 127e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 137e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * See the License for the specific language governing permissions and 147e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * limitations under the License. 157e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri */ 167e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarripackage android.support.v4.app; 177e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 18754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikasimport static junit.framework.TestCase.assertEquals; 19754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikasimport static junit.framework.TestCase.assertTrue; 20754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikasimport static junit.framework.TestCase.fail; 21754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikas 227e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.app.Activity; 237e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.app.PendingIntent; 247e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.content.Intent; 257e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.content.IntentSender; 26754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikasimport android.support.fragment.test.R; 277e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.support.test.InstrumentationRegistry; 28754cb29c50f09a83251dd4bb633ba445b2411adbAurimas Liutikasimport android.support.test.filters.SmallTest; 29c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikasimport android.support.test.rule.ActivityTestRule; 307e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.support.test.runner.AndroidJUnit4; 317e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.support.v4.app.test.FragmentResultActivity; 327e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport android.support.v4.app.test.FragmentTestActivity; 337e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 347e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport org.junit.Before; 35c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikasimport org.junit.Rule; 367e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport org.junit.Test; 377e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarriimport org.junit.runner.RunWith; 387e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 394285a0db794444695fd0752e7bbc2ad95082043cGeorge Mountimport java.util.concurrent.CountDownLatch; 404285a0db794444695fd0752e7bbc2ad95082043cGeorge Mountimport java.util.concurrent.TimeUnit; 414285a0db794444695fd0752e7bbc2ad95082043cGeorge Mount 427e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri/** 437e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri * Tests for Fragment startActivityForResult and startIntentSenderForResult. 447e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri */ 457e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri@RunWith(AndroidJUnit4.class) 46c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikaspublic class FragmentReceiveResultTest { 47c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas @Rule 48c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public ActivityTestRule<FragmentTestActivity> mActivityRule = 49c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas new ActivityTestRule<FragmentTestActivity>(FragmentTestActivity.class); 50c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas 517e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri private FragmentTestActivity mActivity; 527e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri private TestFragment mFragment; 537e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 547e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 557e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Before 56c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public void setup() throws Throwable { 57c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas mActivity = mActivityRule.getActivity(); 587e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mFragment = attachTestFragment(); 597e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 607e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 617e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Test 627e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @SmallTest 63c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public void testStartActivityForResultOk() throws Throwable { 647e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri startActivityForResult(10, Activity.RESULT_OK, "content 10"); 657e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 667e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertTrue("Fragment should receive result", mFragment.mHasResult); 677e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(10, mFragment.mRequestCode); 687e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(Activity.RESULT_OK, mFragment.mResultCode); 697e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals("content 10", mFragment.mResultContent); 707e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 717e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 727e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Test 737e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @SmallTest 74c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public void testStartActivityForResultCanceled() throws Throwable { 757e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri startActivityForResult(20, Activity.RESULT_CANCELED, "content 20"); 767e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 777e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertTrue("Fragment should receive result", mFragment.mHasResult); 787e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(20, mFragment.mRequestCode); 797e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(Activity.RESULT_CANCELED, mFragment.mResultCode); 807e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals("content 20", mFragment.mResultContent); 817e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 827e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 837e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Test 847e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @SmallTest 85c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public void testStartIntentSenderForResultOk() throws Throwable { 867e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri startIntentSenderForResult(30, Activity.RESULT_OK, "content 30"); 877e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 887e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertTrue("Fragment should receive result", mFragment.mHasResult); 897e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(30, mFragment.mRequestCode); 907e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(Activity.RESULT_OK, mFragment.mResultCode); 917e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals("content 30", mFragment.mResultContent); 927e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 937e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 947e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Test 957e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @SmallTest 96c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas public void testStartIntentSenderForResultCanceled() throws Throwable { 977e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri startIntentSenderForResult(40, Activity.RESULT_CANCELED, "content 40"); 987e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 997e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertTrue("Fragment should receive result", mFragment.mHasResult); 1007e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(40, mFragment.mRequestCode); 1017e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals(Activity.RESULT_CANCELED, mFragment.mResultCode); 1027e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri assertEquals("content 40", mFragment.mResultContent); 1037e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1047e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 105c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas private TestFragment attachTestFragment() throws Throwable { 1067e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri final TestFragment fragment = new TestFragment(); 107c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas mActivityRule.runOnUiThread(new Runnable() { 1087e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Override 1097e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri public void run() { 1107e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mActivity.getSupportFragmentManager().beginTransaction() 1117e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri .add(R.id.content, fragment) 1127e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri .addToBackStack(null) 1137e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri .commitAllowingStateLoss(); 1147e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mActivity.getFragmentManager().executePendingTransactions(); 1157e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1167e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri }); 1177e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri InstrumentationRegistry.getInstrumentation().waitForIdleSync(); 1187e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri return fragment; 1197e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1207e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1217e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri private void startActivityForResult(final int requestCode, final int resultCode, 122c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas final String content) throws Throwable { 123c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas mActivityRule.runOnUiThread(new Runnable() { 1247e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Override 1257e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri public void run() { 1267e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri Intent intent = new Intent(mActivity, FragmentResultActivity.class); 1277e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri intent.putExtra(FragmentResultActivity.EXTRA_RESULT_CODE, resultCode); 1287e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri intent.putExtra(FragmentResultActivity.EXTRA_RESULT_CONTENT, content); 1297e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1307e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mFragment.startActivityForResult(intent, requestCode); 1317e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1327e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri }); 1334285a0db794444695fd0752e7bbc2ad95082043cGeorge Mount assertTrue(mFragment.mResultReceiveLatch.await(1, TimeUnit.SECONDS)); 1347e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri InstrumentationRegistry.getInstrumentation().waitForIdleSync(); 1357e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1367e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1377e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri private void startIntentSenderForResult(final int requestCode, final int resultCode, 138c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas final String content) throws Throwable { 139c90b24ff959edb955ff48e345c074a98368a90adAurimas Liutikas mActivityRule.runOnUiThread(new Runnable() { 1407e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Override 1417e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri public void run() { 1427e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri Intent intent = new Intent(mActivity, FragmentResultActivity.class); 1437e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri intent.putExtra(FragmentResultActivity.EXTRA_RESULT_CODE, resultCode); 1447e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri intent.putExtra(FragmentResultActivity.EXTRA_RESULT_CONTENT, content); 1457e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1467e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri PendingIntent pendingIntent = PendingIntent.getActivity(mActivity, 1477e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri requestCode, intent, 0); 1487e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1497e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri try { 1507e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mFragment.startIntentSenderForResult(pendingIntent.getIntentSender(), 1517e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri requestCode, null, 0, 0, 0, null); 1527e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } catch (IntentSender.SendIntentException e) { 1537e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri fail("IntentSender failed"); 1547e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1557e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1567e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri }); 1574285a0db794444695fd0752e7bbc2ad95082043cGeorge Mount assertTrue(mFragment.mResultReceiveLatch.await(1, TimeUnit.SECONDS)); 1587e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri InstrumentationRegistry.getInstrumentation().waitForIdleSync(); 1597e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1607e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 161af556dcfc63652cafa822a3d742ad96c78bff037Clara Bayarri public static class TestFragment extends Fragment { 1627e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri boolean mHasResult = false; 1637e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri int mRequestCode = -1; 1647e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri int mResultCode = 100; 1657e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri String mResultContent; 1664285a0db794444695fd0752e7bbc2ad95082043cGeorge Mount final CountDownLatch mResultReceiveLatch = new CountDownLatch(1); 1677e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri 1687e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri @Override 1697e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri public void onActivityResult(int requestCode, int resultCode, Intent data) { 1707e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mHasResult = true; 1717e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mRequestCode = requestCode; 1727e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mResultCode = resultCode; 1737e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri mResultContent = data.getStringExtra(FragmentResultActivity.EXTRA_RESULT_CONTENT); 1744285a0db794444695fd0752e7bbc2ad95082043cGeorge Mount mResultReceiveLatch.countDown(); 1757e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1767e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri } 1777e866aba6258c1c7c2e0e9b303d065f3ed88e841Clara Bayarri} 178