1 package org.robolectric.shadows; 2 3 import android.view.View; 4 import android.widget.HeaderViewListAdapter; 5 import android.widget.ListView; 6 import java.util.ArrayList; 7 import java.util.List; 8 import org.robolectric.annotation.Implements; 9 import org.robolectric.annotation.RealObject; 10 11 @SuppressWarnings({"UnusedDeclaration"}) 12 @Implements(ListView.class) 13 public class ShadowListView extends ShadowAbsListView { 14 @RealObject private ListView realListView; 15 16 public List<View> getHeaderViews() { 17 HeaderViewListAdapter adapter = (HeaderViewListAdapter) realListView.getAdapter(); 18 ArrayList<View> headerViews = new ArrayList<>(); 19 int headersCount = adapter.getHeadersCount(); 20 for (int i = 0; i < headersCount; i++) { 21 headerViews.add(adapter.getView(i, null, realListView)); 22 } 23 return headerViews; 24 } 25 26 public List<View> getFooterViews() { 27 HeaderViewListAdapter adapter = (HeaderViewListAdapter) realListView.getAdapter(); 28 ArrayList<View> footerViews = new ArrayList<>(); 29 int offset = adapter.getHeadersCount() + adapter.getCount() - adapter.getFootersCount(); 30 int itemCount = adapter.getCount(); 31 for (int i = offset; i < itemCount; i++) { 32 footerViews.add(adapter.getView(i, null, realListView)); 33 } 34 return footerViews; 35 } 36 } 37
