119 lines
3.4 KiB
Java
119 lines
3.4 KiB
Java
/**
|
|
*
|
|
*/
|
|
package org.gcube.storagehub;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.util.Properties;
|
|
|
|
import org.gcube.common.authorization.utils.manager.SecretManager;
|
|
import org.gcube.common.authorization.utils.manager.SecretManagerProvider;
|
|
import org.gcube.common.authorization.utils.secret.JWTSecret;
|
|
import org.gcube.common.authorization.utils.secret.Secret;
|
|
import org.gcube.common.authorization.utils.secret.SecretUtility;
|
|
import org.gcube.common.keycloak.KeycloakClientFactory;
|
|
import org.gcube.common.keycloak.model.TokenResponse;
|
|
import org.gcube.common.scope.api.ScopeProvider;
|
|
import org.junit.AfterClass;
|
|
import org.junit.BeforeClass;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
/**
|
|
* @author Luca Frosini (ISTI - CNR)
|
|
*/
|
|
public class ContextTest {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(ContextTest.class);
|
|
|
|
protected static final String CONFIG_INI_FILENAME = "config.ini";
|
|
|
|
public static final String GCUBE;
|
|
public static final String DEVNEXT;
|
|
public static final String NEXTNEXT;
|
|
public static final String DEVSEC;
|
|
public static final String DEVVRE;
|
|
|
|
public static final String ROOT;
|
|
|
|
protected static final Properties properties;
|
|
|
|
protected static final String CLIENT_ID_PROPERTY_KEY = "client_id";
|
|
protected static final String clientID;
|
|
|
|
static {
|
|
GCUBE = "/gcube";
|
|
DEVNEXT = GCUBE + "/devNext";
|
|
NEXTNEXT = DEVNEXT + "/NextNext";
|
|
DEVSEC = GCUBE + "/devsec";
|
|
DEVVRE = DEVSEC + "/devVRE";
|
|
|
|
ROOT = GCUBE;
|
|
|
|
properties = new Properties();
|
|
InputStream input = ContextTest.class.getClassLoader().getResourceAsStream(CONFIG_INI_FILENAME);
|
|
try {
|
|
// load the properties file
|
|
properties.load(input);
|
|
|
|
clientID = properties.getProperty(CLIENT_ID_PROPERTY_KEY);
|
|
|
|
} catch (IOException e) {
|
|
throw new RuntimeException(e);
|
|
}
|
|
|
|
}
|
|
|
|
public static void set(Secret secret) throws Exception {
|
|
SecretManagerProvider.instance.reset();
|
|
SecretManager secretManager = new SecretManager();
|
|
secretManager.addSecret(secret);
|
|
SecretManagerProvider.instance.set(secretManager);
|
|
SecretManagerProvider.instance.get().set();
|
|
}
|
|
|
|
public static void setContextByName(String fullContextName) throws Exception {
|
|
Secret secret = getSecretByContextName(fullContextName);
|
|
set(secret);
|
|
}
|
|
|
|
|
|
private static TokenResponse getJWTAccessToken(String context) throws Exception {
|
|
ScopeProvider.instance.set(context);
|
|
int index = context.indexOf('/', 1);
|
|
String root = context.substring(0, index == -1 ? context.length() : index);
|
|
String clientSecret = properties.getProperty(root);
|
|
TokenResponse tr = KeycloakClientFactory.newInstance().queryUMAToken(clientID, clientSecret, context, null);
|
|
return tr;
|
|
}
|
|
|
|
public static Secret getSecretByContextName(String context) throws Exception {
|
|
TokenResponse tr = getJWTAccessToken(context);
|
|
Secret secret = new JWTSecret(tr.getAccessToken());
|
|
return secret;
|
|
}
|
|
|
|
public static void setContext(String token) throws Exception {
|
|
Secret secret = getSecret(token);
|
|
set(secret);
|
|
}
|
|
|
|
private static Secret getSecret(String token) throws Exception {
|
|
Secret secret = SecretUtility.getSecretByTokenString(token);
|
|
return secret;
|
|
}
|
|
|
|
|
|
@BeforeClass
|
|
public static void beforeClass() throws Exception {
|
|
setContextByName(GCUBE);
|
|
}
|
|
|
|
@AfterClass
|
|
public static void afterClass() throws Exception {
|
|
SecretManagerProvider.instance.reset();
|
|
}
|
|
|
|
}
|