private void setupStandardMocks() throws Exception {
Map<String, List<String>> externalGroups = new HashMap<>();
externalGroups.put(
EzSecurityConstant.EZ_INTERNAL_PROJECT,
Lists.newArrayList(EzSecurityConstant.EZ_INTERNAL_ADMIN_GROUP));
EzSecurityPrincipal principal = new EzSecurityPrincipal();
principal.setPrincipal(POC).setName(POC);
expect(
mockPool.getClient(
InternalNameServiceConstants.SERVICE_NAME, InternalNameService.Client.class))
.andReturn(insClientMock)
.anyTimes();
expect(
mockPool.getClient(
EzSecurityRegistrationConstants.SERVICE_NAME, EzSecurityRegistration.Client.class))
.andReturn(regClientMock)
.anyTimes();
expect(mockPool.getSecurityId(EzSecurityRegistrationConstants.SERVICE_NAME))
.andReturn("")
.anyTimes();
expect(mockPool.getSecurityId(EzDeployServiceConstants.SERVICE_NAME)).andReturn("").anyTimes();
expect(securityClientMock.fetchTokenForProxiedUser()).andReturn(ezTokenMock).anyTimes();
expect(securityClientMock.fetchTokenForProxiedUser("")).andReturn(ezTokenMock).anyTimes();
expect(insClientMock.getAppById("48454c4c4f-20-574f524c44", ezTokenMock))
.andReturn(getApplication());
expect(ezTokenMock.getExternalProjectGroups()).andReturn(externalGroups).anyTimes();
expect(ezTokenMock.getTokenPrincipal()).andReturn(principal).anyTimes();
mockPool.returnToPool(insClientMock);
expectLastCall().anyTimes();
mockPool.returnToPool(regClientMock);
expectLastCall().anyTimes();
}
public void run()
throws TException, EzSecurityTokenException, UserNotFoundException, IOException,
AppNotRegisteredException {
Properties config;
try {
config =
new EzConfiguration(new DirectoryConfigurationLoader(new File(this.config).toPath()))
.getProperties();
} catch (EzConfigurationLoaderException e) {
try {
config = new EzConfiguration(new ClasspathConfigurationLoader()).getProperties();
} catch (EzConfigurationLoaderException e1) {
throw new RuntimeException("Unable to load EzConfiguration");
}
}
if (config.get(EzBakePropertyConstants.ZOOKEEPER_CONNECTION_STRING) == null) {
config.setProperty(EzBakePropertyConstants.ZOOKEEPER_CONNECTION_STRING, this.zoo);
}
if (config.get(EzBakePropertyConstants.EZBAKE_CERTIFICATES_DIRECTORY) == null) {
config.setProperty(EzBakePropertyConstants.EZBAKE_CERTIFICATES_DIRECTORY, this.sslDir);
}
if (config.get(EzBakePropertyConstants.EZBAKE_SECURITY_ID) == null) {
config.setProperty(EzBakePropertyConstants.EZBAKE_SECURITY_ID, this.appId);
}
EzbakeSecurityClient client = new EzbakeSecurityClient(config);
switch (this.request) {
case User:
EzSecurityToken usertoken =
client.fetchTokenForProxiedUser(issuedTokenPrincipal(dn), this.target);
if (this.outputFile != null) {
writeTokenToFile(this.outputFile, usertoken);
}
break;
case App:
EzSecurityToken appToken = client.fetchAppToken(dn);
if (this.outputFile != null) {
writeTokenToFile(this.outputFile, appToken);
}
break;
case DN:
ProxyTokenRequest req = new ProxyTokenRequest();
req.setX509(new X509Info(dn));
req.setValidity(
new ValidityCaveats("EFE", "EzSecurity", System.currentTimeMillis() + 1000, ""));
EzSecurity.Client c = client.getClient();
ProxyTokenResponse principal = c.requestProxyToken(req);
client.returnClient(c);
System.out.println(principal.getToken());
System.out.println(principal.getSignature());
break;
case PROXY_DN:
ProxyTokenRequest proxyReq = new ProxyTokenRequest();
proxyReq.setX509(new X509Info(dn));
proxyReq.setValidity(
new ValidityCaveats("EFE", "EzSecurity", System.currentTimeMillis() + 1000, ""));
proxyReq.getValidity().setIssuedTime(System.currentTimeMillis());
EzSecurity.Client pc = client.getClient();
ProxyTokenResponse presp = pc.requestProxyToken(proxyReq);
client.returnClient(pc);
System.out.println(presp.getToken());
System.out.println(presp.getSignature());
break;
}
Closeables.close(client, true);
}
