使用 Robolectric 和 Mockito 测试 CursorLoader

Question

假设我正在开发一个简单的 ListFragment（在这种情况下，它会从 MediaStore 读取艺术家列表，但稍后还会从其他来源读取数据），如下所示：

@EFragment
public class ArtistsFragment extends ListFragment implements LoaderManager.LoaderCallbacks<Cursor> {
    private static final String TAG = ArtistsFragment.class.getName();
    private SimpleCursorAdapter mAdapter;

    Uri uri = MediaStore.Audio.Artists.EXTERNAL_CONTENT_URI;

    CursorLoader mCursorLoader;

    @AfterViews
    void setupView() {
        mAdapter = new SimpleCursorAdapter(getActivity(),
                android.R.layout.simple_list_item_1, null,
                new String[]{MediaStore.Audio.Artists.ARTIST}, // lists path of files
                new int[]{android.R.id.text1}, 0);

        setListAdapter(mAdapter);

        getLoaderManager().initLoader(0, null, this);
    }

    @Override
    public Loader<Cursor> onCreateLoader(int id, Bundle args) {
        if (mCursorLoader == null) {
            mCursorLoader = new CursorLoader(getActivity(), uri, new String[]{MediaStore.Audio.Artists._ID, MediaStore.Audio.Artists.ARTIST},
                    null, null, MediaStore.Audio.Artists.ARTIST + " ASC");
        } else {
            System.out.println("mCursorLoader.count: " + mCursorLoader.loadInBackground().getCount());            
        }
        return mCursorLoader;
    }

    @Override
    public void onLoadFinished(Loader<Cursor> loader, Cursor data) {
        setListShown(true);
        mAdapter.swapCursor(data);
    }

    @Override
    public void onLoaderReset(Loader<Cursor> loader) {
        mAdapter.swapCursor(null);
    }
}

我想使用 Robolectric + Mockito + awaitility 来证明 Fragment 在各种条件下（例如空列表或无效数据等）。我的测试类如下所示：

@RunWith(RobolectricTestRunner.class)
public class ArtistsFragmentTest {
    @Test
    public void shouldNotBeNull() {
        final ArtistsFragment myFragment = ArtistsFragment_.builder().build();
        assertNotNull(myFragment);

        // Create a mock cursor.
        final Cursor mc = getSampleArtistCursor();
        when(mc.getCount()).thenReturn(1);
        when(mc.getInt(0)).thenReturn(1);
        when(mc.getString(1)).thenReturn("Sample Title");

        myFragment.mCursorLoader = mock(CursorLoader.class);
        when(myFragment.mCursorLoader.loadInBackground()).thenReturn(mc);

        startFragment(myFragment);

        assertNotNull(myFragment.getListView());
        await().atMost(5, TimeUnit.SECONDS).until(new Callable<Integer>() {
            @Override
            public Integer call() throws Exception {
                return myFragment.getListAdapter().getCount();
            }
        }, equalTo(1));

        System.out.println(myFragment.getListAdapter().getCount());
    }

    private Cursor getSampleArtistCursor() {
        return new CursorWrapper(mock(MockCursor.class));
    }
}

然后在 IntelliJ 或 maven 中运行此测试时，测试将失败，适配器将始终返回零计数。

onCreateLoader 中的 System.out.println 语句返回 1。我需要特别注意后台线程中的 Mockito 吗？ （loadInBackground 方法在工作线程上运行）。

Answer 1

我刚刚在我的代码中进行了 Loader 测试。就我而言，我发现测试加载器本身比尝试通过片段代码路由它更直接。

我最终使用了这篇文章中代码的略微修改版本： https://groups.google.com/forum/#!msg/robolectric/xY5MF399jA8/V5PnUfh1D-wJ

首先，我必须实现一些影子类，因为 Robolectric2 不包含 AsyncTaskLoader 或 Loader 类的影子代码。如果您从未添加过影子类，请知道将它们放在正确的包中很重要。这两个阴影都应该存在于 android.support.v4.content 中。

ShadowLoader

@Implements(Loader.class)
public class ShadowLoader<D> {

// //////////////////////
// Constants

// //////////////////////
// Fields
protected boolean reset;

// //////////////////////
// Constructors

// //////////////////////
// Getter & Setter

// //////////////////////
// Methods from SuperClass/Interfaces

@Implementation
public void reset() {
    reset = true;
}

@Implementation
public boolean isReset() {
    return reset;
}
// //////////////////////
// Methods

// //////////////////////
// Inner and Anonymous Classes
}

ShadowAsyncTaskLoader

@Implements(AsyncTaskLoader.class)
public class ShadowAsyncTaskLoader<D> extends ShadowLoader {

@RealObject
private AsyncTaskLoader<D> asyncTaskLoader;

@Implementation
void executePendingTask() {
    new AsyncTask<Void, Void, D>() {
        @Override
        protected D doInBackground(Void... params) {
            return (D) asyncTaskLoader.loadInBackground();
        }

        @Override
        protected void onPostExecute(D data) {
            updateLastLoadCompleteTimeField();
            asyncTaskLoader.deliverResult(data);
        }

        @Override
        protected void onCancelled(D data) {
            updateLastLoadCompleteTimeField();
            asyncTaskLoader.onCanceled(data);

            executePendingTask();
        }
    }.execute((Void)null);
}


public void setReset(boolean reset) {
    this.reset = reset;
}

private void updateLastLoadCompleteTimeField() {
    try {
        Field mLastLoadCompleteTimeField = AsyncTaskLoader.class.getDeclaredField("mLastLoadCompleteTime");
        if(!mLastLoadCompleteTimeField.isAccessible()) {
            mLastLoadCompleteTimeField.setAccessible(true);
        }
        mLastLoadCompleteTimeField.set(asyncTaskLoader, SystemClock.uptimeMillis());

    } catch(NoSuchFieldException e) {
        e.printStackTrace();
    } catch(IllegalAccessException e) {
        e.printStackTrace();
    }
}
}

然后，根据您的配置，您可以添加注释以使用自定义类

@Config( shadows = { ShadowAsyncTaskLoader.class, ShadowLoader.class})

此时调用 loader.onStartLoading() 会导致加载程序按预期运行，而无需将任何等待命令破解到我的测试用例中。

希望这会有所帮助。我还没有尝试将 LoaderManager 与这种测试方法一起使用，所以我无法验证它是否通过该调用工作。

注意：我添加 ShadowLoader 的原因是因为我发现 isReset() 在我没想到时返回 true。

Answer 2

解决方法是使用：

Robolectric.flushForegroundThreadScheduler();

（机器人 3.0）

这将立即运行所有任务，包括 CursorLoader。