How to unit test custom view with attribute

Question

I'm having trouble to create a unit test for my custom view. I try to add an attribute and test it if my custom view class get it right.

Here's what my test looks like:

@RunWith(AndroidJUnit4.class)
@SmallTest
public class BaseRatingBarMinRatingTest {

    private Context mContext;

    @Before
    public void setUp(){
        mContext = InstrumentationRegistry.getTargetContext();
    }

    @Test
    public void constructor_should_setMinRating_when_attriSetHasOne() throws Exception{
        // 1. ARRANGE DATA
        float minRating = 2.5f;
        AttributeSet as = mock(AttributeSet.class);
        when(as.getAttributeFloatValue(eq(R.styleable.BaseRatingBar_srb_minRating), anyFloat())).thenReturn(minRating);


        // 2. ACT
        BaseRatingBar brb = new BaseRatingBar(mContext, as);


        // 3. ASSERT
        assertThat(brb.getMinRating(), is(minRating));
    }

  // ...

}

Which gets this Exception:

java.lang.ClassCastException: android.util.AttributeSet$MockitoMock$1142631110 cannot be cast to android.content.res.XmlBlock$Parser

I tried mocking TypeArray like this article did, but my view treats the mocked context as null.

Is there any good way to create a test case for custom view?

You can implement a presenter for your view and unit test it that way. You might also be able to use RoboElectric — dazza5000, Feb 24 '18 at 16:54
Unfortunately, the custom view I'm implementing is from an open source library, which does not have presenter layer in the original project structure. RoboElectric seems like a good way. Thank you for this suggestion. =) — Tomaz Wang, Feb 25 '18 at 09:06
Refer to [this related post](https://stackoverflow.com/q/4913875/8583692). — Mahozad, Sep 20 '21 at 18:09

score 5 · Answer 1 · edited Nov 21 '19 at 15:05

5

Consider using Robolectric, as it eliminates the need for mocks.

AttributeSet as =  Robolectric.buildAttributeSet().addAttribute(R.style.BaseRatingBar_srb_minRating, "2.5f").build()
BaseRatingBar brb = new BaseRatingBar(mContext, as);

edited Nov 21 '19 at 15:05

noloman

11,411
20
82
129

answered Jun 07 '19 at 19:35

amram99

699
8
19

score 3 · Answer 2 · answered Mar 06 '18 at 11:01

I had a similar problem as you did. Like you, I wanted to test a custom View. This is how I solved it:

public class CustomViewTest {
    @Mock
    private Context context;
    @Mock
    private AttributeSet attributes;
    @Mock
    private TypedArray typedArray;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        when(context.obtainStyledAttributes(attributes, R.styleable.CustomView)).thenReturn(typedArray);
        when(typedArray.getInteger(eq(R.styleable.CustomView_customAttribute), anyInt())).thenReturn(23);
    }

    @Test
    public void constructor() {
        CustomView customView = new CustomView(context, attributes);
        assertEquals(23, customView.getCustomAttribute());
    }
}

And in my CustomView class:

public CustomView(Context context, AttributeSet attrs) {
    super(context, attrs);
    if (attrs != null) {
        TypedArray a = context.obtainStyledAttributes(attrs, R.styleable.CustomView);
        customAttribute = a.getInteger(R.styleable.CustomView_customAttribute, 19);
        a.recycle();
    } else {
        customAttribute = 19;
    }
}

Try this approach, and post the exact error message if it doesn't work. Hope it helps.

Problem is this is using a standard unit test (i.e. not instrumented test annotated with `@RunWith(AndroidJUnit4.class)`). So if you want to test some behaviour which requires the use of real UI components it won't work - you need to use a real context not mocked context — Adam Burley, Aug 18 '22 at 13:28

How to unit test custom view with attribute

2 Answers2