class AndroidMocks {

    public static AlertDialog.Builder mockDialogBuilder() {
        AlertDialog.Builder builder = mock(AlertDialog.Builder.class);
        when(builder.setTitle(anyInt())).thenReturn(builder);
        when(builder.setTitle(anyString())).thenReturn(builder);
        when(builder.setMessage(anyInt())).thenReturn(builder);
        when(builder.setMessage(anyString())).thenReturn(builder);
        when(builder.setPositiveButton(anyInt(), any(DialogInterface.OnClickListener.class))).thenReturn(builder);
        when(builder.setPositiveButton(anyString(), any(DialogInterface.OnClickListener.class))).thenReturn(builder);
        when(builder.setNegativeButton(anyInt(), any(DialogInterface.OnClickListener.class))).thenReturn(builder);
        when(builder.setNegativeButton(anyString(), any(DialogInterface.OnClickListener.class))).thenReturn(builder);
        return builder;
    }

}