TestDefaultStatsService done.

openid-connect-server/src/test/java/org/mitre/openid/connect/service/impl/TestDefaultStatsService.java

@@ -0,0 +1,165 @@
+/*******************************************************************************
+ * Copyright 2013 The MITRE Corporation 
+ *   and the MIT Kerberos and Internet Trust Consortium
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ ******************************************************************************/
+package org.mitre.openid.connect.service.impl;
+
+import static org.junit.Assert.*;
+import static org.hamcrest.CoreMatchers.*;
+
+import java.util.HashSet;
+import java.util.Map;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mitre.oauth2.model.ClientDetailsEntity;
+import org.mitre.oauth2.service.ClientDetailsEntityService;
+import org.mitre.openid.connect.model.ApprovedSite;
+import org.mitre.openid.connect.service.ApprovedSiteService;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import com.google.common.collect.Sets;
+
+/**
+ * @author wkim
+ *
+ */
+@RunWith(MockitoJUnitRunner.class)
+public class TestDefaultStatsService {
+
+	// Test fixtures:
+	// Currently tests 4 approved sites with a total of 2 users and 3 clients for those sites.
+	// There is an extra client in the system to make sure the stats only count for approved sites.
+
+	private String userId1 = "batman";
+	private String userId2 = "alfred";
+
+	private String clientId1 = "bar";
+	private String clientId2 = "pawnshop";
+	private String clientId3 = "pizzastore";
+	private String clientId4 = "gasstation";
+
+	private ApprovedSite ap1 = Mockito.mock(ApprovedSite.class);
+	private ApprovedSite ap2 = Mockito.mock(ApprovedSite.class);
+	private ApprovedSite ap3 = Mockito.mock(ApprovedSite.class);
+	private ApprovedSite ap4 = Mockito.mock(ApprovedSite.class);
+
+	private ClientDetailsEntity client1 = Mockito.mock(ClientDetailsEntity.class);
+	private ClientDetailsEntity client2 = Mockito.mock(ClientDetailsEntity.class);
+	private ClientDetailsEntity client3 = Mockito.mock(ClientDetailsEntity.class);
+	private ClientDetailsEntity client4 = Mockito.mock(ClientDetailsEntity.class);
+
+	@Mock
+	private ApprovedSiteService approvedSiteService;
+
+	@Mock
+	private ClientDetailsEntityService clientService;
+
+	@InjectMocks
+	private DefaultStatsService service = new DefaultStatsService();
+
+	/**
+	 * Prepares a collection of ApprovedSite mocks to be returned from the approvedSiteService
+	 * and a collection of ClientDetailEntity mocks to be returned from the clientService.
+	 */
+	@Before
+	public void prepare() {
+
+		Mockito.reset(approvedSiteService, clientService);
+
+		Mockito.when(ap1.getUserId()).thenReturn(userId1);
+		Mockito.when(ap1.getClientId()).thenReturn(clientId1);
+
+		Mockito.when(ap2.getUserId()).thenReturn(userId1);
+		Mockito.when(ap2.getClientId()).thenReturn(clientId1);
+
+		Mockito.when(ap3.getUserId()).thenReturn(userId2);
+		Mockito.when(ap3.getClientId()).thenReturn(clientId2);
+
+		Mockito.when(ap4.getUserId()).thenReturn(userId2);
+		Mockito.when(ap4.getClientId()).thenReturn(clientId3);
+
+		Mockito.when(approvedSiteService.getAll()).thenReturn(Sets.newHashSet(ap1, ap2, ap3, ap4));
+
+		Mockito.when(client1.getId()).thenReturn(1L);
+		Mockito.when(client2.getId()).thenReturn(2L);
+		Mockito.when(client3.getId()).thenReturn(3L);
+		Mockito.when(client4.getId()).thenReturn(4L);
+
+		Mockito.when(clientService.getAllClients()).thenReturn(Sets.newHashSet(client1, client2, client3, client4));
+		Mockito.when(clientService.loadClientByClientId(clientId1)).thenReturn(client1);
+		Mockito.when(clientService.loadClientByClientId(clientId2)).thenReturn(client2);
+		Mockito.when(clientService.loadClientByClientId(clientId3)).thenReturn(client3);
+		Mockito.when(clientService.loadClientByClientId(clientId4)).thenReturn(client4);
+	}
+
+	@Test
+	public void calculateSummaryStats_empty() {
+
+		Mockito.when(approvedSiteService.getAll()).thenReturn(new HashSet<ApprovedSite>());
+
+		Map<String, Integer> stats = service.calculateSummaryStats();
+
+		assertThat(stats.get("approvalCount"), is(0));
+		assertThat(stats.get("userCount"), is(0));
+		assertThat(stats.get("clientCount"), is(0));
+	}
+
+	@Test
+	public void calculateSummaryStats() {
+		Map<String, Integer> stats = service.calculateSummaryStats();
+
+		assertThat(stats.get("approvalCount"), is(4));
+		assertThat(stats.get("userCount"), is(2));
+		assertThat(stats.get("clientCount"), is(3));
+	}
+	
+	@Test
+	public void calculateByClientId_empty() {
+		
+		Mockito.when(approvedSiteService.getAll()).thenReturn(new HashSet<ApprovedSite>());
+		
+		Map<Long, Integer> stats = service.calculateByClientId();
+		
+		assertThat(stats.get(1L), is(0));
+		assertThat(stats.get(2L), is(0));
+		assertThat(stats.get(3L), is(0));
+		assertThat(stats.get(4L), is(0));
+	}
+
+	@Test
+	public void calculateByClientId() {
+		
+		Map<Long, Integer> stats = service.calculateByClientId();
+		
+		assertThat(stats.get(1L), is(2));
+		assertThat(stats.get(2L), is(1));
+		assertThat(stats.get(3L), is(1));
+		assertThat(stats.get(4L), is(0));
+	}
+	
+	@Test
+	public void countForClientId() {
+
+		assertThat(service.countForClientId(1L), is(2));
+		assertThat(service.countForClientId(2L), is(1));
+		assertThat(service.countForClientId(3L), is(1));
+		assertThat(service.countForClientId(4L), is(0));
+	}
+}