使用Robolectric& amp;测试CursorLoader的Mockito

时间:2013-07-05 10:28:38

标签: android mockito robolectric

鉴于我正在开发一个简单的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方法在工作线程上运行。)

2 个答案:

答案 0 :(得分:1)

我刚刚在我的代码中使用了Loader测试。在我的情况下,我发现更直接地测试加载器本身而不是尝试通过Fragment代码路由它。

我使用这篇文章稍微修改过的代码版本结束了: 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。

答案 1 :(得分:1)

解决方案是使用:

Robolectric.flushForegroundThreadScheduler();

(Robolectric 3.0)

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