fix(stats): fix stats with updated api

https://zanata.atlassian.net/browse/ZNTA-874
zanata · Jan 15, 2016 · 5384f2c · 5384f2c
1 parent 9841bf8
commit 5384f2c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 8 deletions.
diff --git a/stub-server/src/main/java/org/zanata/rest/service/MockStatisticsResource.java b/stub-server/src/main/java/org/zanata/rest/service/MockStatisticsResource.java
@@ -32,6 +32,9 @@
 import org.zanata.rest.dto.stats.contribution.ContributionStatistics;
 import org.zanata.rest.dto.stats.contribution.LocaleStatistics;
 
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * @author Patrick Huang <a
  *         href="mailto:pahuang@redhat.com">pahuang@redhat.com</a>
@@ -54,8 +57,7 @@ private ContainerTranslationStatistics generateStatistics(
         stats.setId(id);
         for (String locale : locales) {
             stats.addStats(new TranslationStatistics(new TransUnitCount(
-                    100, 0,
-                    0, 100, 0), locale.trim()));
+                    100, 0, 0, 100, 0), locale.trim()));
         }
         return stats;
     }
@@ -70,12 +72,19 @@ public ContainerTranslationStatistics getStatistics(String projectSlug,
     @Override
     public ContributionStatistics getContributionStatistics(String projectSlug,
             String versionSlug, String username, String dateRange, boolean includeAutomatedEntry) {
+
+        BaseContributionStatistic transStats = new BaseContributionStatistic(100, 90, 100, 0);
+        BaseContributionStatistic reviewStats = new BaseContributionStatistic(100, 0, 0, 10);
+
+        LocaleStatistics localeStatistics =
+            new LocaleStatistics(new LocaleId("zh"), transStats, reviewStats);
+
+        List<LocaleStatistics> localeStatisticsList = new ArrayList<>();
+        localeStatisticsList.add(localeStatistics);
+
         ContributionStatistics contributionStatistics =
-                new ContributionStatistics();
-        LocaleStatistics localeStatistics = new LocaleStatistics();
-        localeStatistics.put(new LocaleId("zh"), new BaseContributionStatistic(
-                100, 90, 100, 0));
-        contributionStatistics.put(username, localeStatistics);
+            new ContributionStatistics(username, localeStatisticsList);
+
         return contributionStatistics;
     }
 }

diff --git a/zanata-rest-client/src/test/java/org/zanata/rest/client/StatisticsResourceClientTest.java b/zanata-rest-client/src/test/java/org/zanata/rest/client/StatisticsResourceClientTest.java
@@ -74,7 +74,7 @@ public void testGetContributorStatistics() {
         ContributionStatistics statistics =
                 client.getContributionStatistics("about-fedora", "master",
                         "pahuang", "2014-10-01..2014-11-10", false);
-        assertThat(statistics, Matchers.hasKey("pahuang"));
+        assertThat(statistics.getUsername(), Matchers.equalTo("pahuang"));
     }
 }