使用 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+Awaitity来证明碎片在各种条件下（例如空列表或无效数据等）表现良好。我的测试课是这样的：

@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测试。在我的案例中，我发现测试加载程序本身比尝试通过Fragment代码路由它更直接。

我最终使用了这篇文章中代码的一个稍微修改过的版本：https://groups.google.com/forum/#！msg/机器人电动/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();

（Robolectric 3.0）

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