From 339d868e029e7d18c8ff09674307acfc2f670146 Mon Sep 17 00:00:00 2001
From: Justin Richer <jricher@mit.edu>
Date: Tue, 13 May 2014 14:17:40 -0400
Subject: [PATCH] added unit test for data service
---
.../impl/TestMITREidDataService_1_0.java | 98 +++++++++++++++++++
1 file changed, 98 insertions(+)
create mode 100644 openid-connect-server/src/test/java/org/mitre/openid/connect/service/impl/TestMITREidDataService_1_0.java
diff --git a/openid-connect-server/src/test/java/org/mitre/openid/connect/service/impl/TestMITREidDataService_1_0.java b/openid-connect-server/src/test/java/org/mitre/openid/connect/service/impl/TestMITREidDataService_1_0.java
new file mode 100644
index 000000000..3a4c0d885
--- /dev/null
+++ b/openid-connect-server/src/test/java/org/mitre/openid/connect/service/impl/TestMITREidDataService_1_0.java
@@ -0,0 +1,98 @@
+package org.mitre.openid.connect.service.impl;
+
+import java.io.IOException;
+import java.io.StringWriter;
+import java.util.HashSet;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mitre.oauth2.model.AuthenticationHolderEntity;
+import org.mitre.oauth2.model.ClientDetailsEntity;
+import org.mitre.oauth2.model.OAuth2AccessTokenEntity;
+import org.mitre.oauth2.model.OAuth2RefreshTokenEntity;
+import org.mitre.oauth2.model.SystemScope;
+import org.mitre.oauth2.repository.AuthenticationHolderRepository;
+import org.mitre.oauth2.repository.OAuth2ClientRepository;
+import org.mitre.oauth2.repository.OAuth2TokenRepository;
+import org.mitre.oauth2.repository.SystemScopeRepository;
+import org.mitre.openid.connect.model.ApprovedSite;
+import org.mitre.openid.connect.repository.ApprovedSiteRepository;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.runners.MockitoJUnitRunner;
+
+import com.google.common.collect.ImmutableSet;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonParser;
+import com.google.gson.stream.JsonWriter;
+
+@RunWith(MockitoJUnitRunner.class)
+public class TestMITREidDataService_1_0 {
+
+ @Mock
+ private OAuth2ClientRepository clientRepository;
+ @Mock
+ private ApprovedSiteRepository approvedSiteRepository;
+ @Mock
+ private AuthenticationHolderRepository authHolderRepository;
+ @Mock
+ private OAuth2TokenRepository tokenRepository;
+ @Mock
+ private SystemScopeRepository sysScopeRepository;
+
+ @InjectMocks
+ private MITREidDataService_1_0 dataService;
+
+ @Before
+ public void prepare() {
+ Mockito.reset(clientRepository, approvedSiteRepository, authHolderRepository, tokenRepository, sysScopeRepository);
+ }
+
+ @Test
+ public void testExportClients() throws IOException {
+
+ ClientDetailsEntity client1 = new ClientDetailsEntity();
+ client1.setId(1L);
+ client1.setAccessTokenValiditySeconds(3600);
+ client1.setClientId("client1");
+ client1.setClientSecret("clientsecret1");
+ client1.setRedirectUris(ImmutableSet.of("http://foo.com/"));
+ client1.setScope(ImmutableSet.of("foo", "bar", "baz", "dolphin"));
+ client1.setGrantTypes(ImmutableSet.of("implicit", "authorization_code", "urn:ietf:params:oauth:grant_type:redelegate", "refresh_token"));
+ client1.setAllowIntrospection(true);
+
+ ClientDetailsEntity client2 = new ClientDetailsEntity();
+ client2.setId(2L);
+ client2.setAccessTokenValiditySeconds(3600);
+ client2.setClientId("client2");
+ client2.setClientSecret("clientsecret2");
+ client2.setScope(ImmutableSet.of("foo", "dolphin", "electric-wombat"));
+ client2.setGrantTypes(ImmutableSet.of("client_credentials", "urn:ietf:params:oauth:grant_type:redelegate"));
+ client2.setAllowIntrospection(false);
+
+ Mockito.when(clientRepository.getAllClients()).thenReturn(ImmutableSet.of(client1, client2));
+ Mockito.when(approvedSiteRepository.getAll()).thenReturn(new HashSet<ApprovedSite>());
+ Mockito.when(authHolderRepository.getAll()).thenReturn(new HashSet<AuthenticationHolderEntity>());
+ Mockito.when(tokenRepository.getAllAccessTokens()).thenReturn(new HashSet<OAuth2AccessTokenEntity>());
+ Mockito.when(tokenRepository.getAllRefreshTokens()).thenReturn(new HashSet<OAuth2RefreshTokenEntity>());
+ Mockito.when(sysScopeRepository.getAll()).thenReturn(new HashSet<SystemScope>());
+
+ // do the data export
+ StringWriter stringWriter = new StringWriter();
+ JsonWriter writer = new JsonWriter(stringWriter);
+ writer.beginObject();
+ dataService.exportData(writer);
+ writer.endObject();
+ writer.close();
+
+ // parse the output as a JSON object for testing
+ JsonElement elem = new JsonParser().parse(stringWriter.toString());
+
+
+
+
+ }
+
+}