44import org .apache .commons .io .IOUtils ;
55import org .junit .Assert ;
66import org .junit .Test ;
7+ import org .kohsuke .github .GHRepositoryTrafficInfo .DayInfo ;
78import org .mockito .Mockito ;
89
910import java .io .IOException ;
@@ -24,27 +25,18 @@ private <T extends GHRepositoryTrafficInfo> void checkResponse(T expected, T act
2425 Assert .assertEquals (expected .getCount (), actual .getCount ());
2526 Assert .assertEquals (expected .getUniques (), actual .getUniques ());
2627
27- List <T .DayInfo > expectedList = null ;
28- List <T .DayInfo > actualList = null ;
29- Iterator <T .DayInfo > expectedIt ;
30- Iterator <T .DayInfo > actualIt ;
31-
32- if (expected instanceof GHRepositoryViews ){
33- expectedList = (List <T .DayInfo >)((List <?>)((GHRepositoryViews ) expected ).getViews ());
34- actualList = (List <T .DayInfo >)((List <?>)((GHRepositoryViews ) actual ).getViews ());
35- }
36- else if (expected instanceof GHRepositoryClones ){
37- expectedList = (List <T .DayInfo >)((List <?>)((GHRepositoryClones ) expected ).getClones ());
38- actualList = (List <T .DayInfo >)((List <?>)((GHRepositoryClones ) actual ).getClones ());
39- }
28+ List <? extends DayInfo > expectedList = expected .getDailyInfo ();
29+ List <? extends DayInfo > actualList = actual .getDailyInfo ();
30+ Iterator <? extends DayInfo > expectedIt ;
31+ Iterator <? extends DayInfo > actualIt ;
4032
4133 Assert .assertEquals (expectedList .size (), actualList .size ());
4234 expectedIt = expectedList .iterator ();
4335 actualIt = actualList .iterator ();
4436
4537 while (expectedIt .hasNext () && actualIt .hasNext ()) {
46- T . DayInfo expectedDayInfo = expectedIt .next ();
47- T . DayInfo actualDayInfo = actualIt .next ();
38+ DayInfo expectedDayInfo = expectedIt .next ();
39+ DayInfo actualDayInfo = actualIt .next ();
4840 Assert .assertEquals (expectedDayInfo .getCount (), actualDayInfo .getCount ());
4941 Assert .assertEquals (expectedDayInfo .getUniques (), actualDayInfo .getUniques ());
5042 Assert .assertEquals (expectedDayInfo .getTimestamp (), actualDayInfo .getTimestamp ());
0 commit comments