Java Code Examples for org.wso2.carbon.identity.core.util.IdentityTenantUtil#setRealmService()

@Before
public void setUp() throws Exception {
    String dbConfigPath = System.getProperty("APIManagerDBConfigurationPath");
    APIManagerConfiguration config = new APIManagerConfiguration();
    initializeDatabase(dbConfigPath);
    config.load(dbConfigPath);
    ServiceReferenceHolder.getInstance().setAPIManagerConfigurationService(new APIManagerConfigurationServiceImpl
            (config));
    List<Notifier> notifierList = new ArrayList<>();
    SubscriptionsNotifier subscriptionsNotifier = new SubscriptionsNotifier();
    notifierList.add(subscriptionsNotifier);
    ServiceReferenceHolder.getInstance().getNotifiersMap().put(subscriptionsNotifier.getType(), notifierList);
    PowerMockito.mockStatic(KeyManagerHolder.class);
    keyManager = Mockito.mock(KeyManager.class);
    APIMgtDBUtil.initialize();
    apiMgtDAO = ApiMgtDAO.getInstance();
    IdentityTenantUtil.setRealmService(new TestRealmService());
    String identityConfigPath = System.getProperty("IdentityConfigurationPath");
    IdentityConfigParser.getInstance(identityConfigPath);
    OAuthServerConfiguration oAuthServerConfiguration = OAuthServerConfiguration.getInstance();
    ServiceReferenceHolder.getInstance().setOauthServerConfiguration(oAuthServerConfiguration);

}
 

private RealmService createRealmService(WithRealmService withRealmService, boolean reuseIfAvailable) throws
        UserStoreException {

    if (testSessionRealmService != null && reuseIfAvailable) {
        return testSessionRealmService;
    }
    try {
        int tenantId = withRealmService.tenantId();
        String tenantDomain = withRealmService.tenantDomain();

        PrivilegedCarbonContext.startTenantFlow();
        PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantId(tenantId);
        PrivilegedCarbonContext.getThreadLocalCarbonContext().setTenantDomain(tenantDomain);
        testSessionRealmService = new InMemoryRealmService(tenantId);
        UserStoreManager userStoreManager = testSessionRealmService.getTenantUserRealm(tenantId)
                .getUserStoreManager();
        ((MockUserStoreManager) userStoreManager)
                .addSecondaryUserStoreManager("PRIMARY", (MockUserStoreManager) userStoreManager);
        IdentityTenantUtil.setRealmService(testSessionRealmService);
        RegistryDataHolder.getInstance().setRealmService(testSessionRealmService);
        OSGiDataHolder.getInstance().setUserRealmService(testSessionRealmService);

        Class[] singletonClasses = withRealmService.injectToSingletons();
        for (Class singletonClass : singletonClasses) {
            Object instance = getSingletonInstance(singletonClass);
            if (instance != null) {
                setInstanceValue(testSessionRealmService, RealmService.class, singletonClass, instance);
            } else {
                setInstanceValue(testSessionRealmService, RealmService.class, singletonClass, null);
            }

        }
    } finally {
        PrivilegedCarbonContext.endTenantFlow();
    }
    return testSessionRealmService;
}
 

@BeforeMethod
protected void setUp() throws UserStoreException, NoSuchFieldException, IllegalAccessException,
        NoSuchMethodException, URISyntaxException {

    URL root = this.getClass().getClassLoader().getResource(".");
    File file = new File(root.getPath());
    System.setProperty("carbon.home", file.toString());
    resetAuthenticators();

    FrameworkServiceDataHolder.getInstance().setRealmService(mock(RealmService.class));

    CacheBackedIdPMgtDAO cacheBackedIdPMgtDAO = mock(CacheBackedIdPMgtDAO.class);
    Field daoField = IdentityProviderManager.class.getDeclaredField("dao");
    daoField.setAccessible(true);
    daoField.set(IdentityProviderManager.getInstance(), cacheBackedIdPMgtDAO);

    RealmService mockRealmService = mock(RealmService.class);
    TenantManager tenantManager = mock(TenantManager.class);
    when(tenantManager.getTenantId(anyString())).thenReturn(1);
    when(mockRealmService.getTenantManager()).thenReturn(tenantManager);
    IdentityTenantUtil.setRealmService(mockRealmService);

    Field configFilePathField = FileBasedConfigurationBuilder.class.getDeclaredField("configFilePath");
    configFilePathField.setAccessible(true);
    URL url = this.getClass().getResource(APPLICATION_AUTHENTICATION_FILE_NAME);
    configFilePathField.set(null, Paths.get(url.toURI()).toString());
}
 

/**
 * @param realmService
 */
@Reference(
        name = "user.realmservice.default",
        service = RealmService.class,
        cardinality = ReferenceCardinality.MANDATORY,
        policy = ReferencePolicy.DYNAMIC,
        unbind = "unsetRealmService"
)
protected void setRealmService(RealmService realmService) {
    IdentityTenantUtil.setRealmService(realmService);
    defaultKeystoreManagerExtension.setRealmService(realmService);
}
 

/**
 * @param realmService
 */
protected void unsetRealmService(RealmService realmService) {
    defaultKeystoreManagerExtension.setRealmService(null);
    IdentityTenantUtil.setRealmService(null);
}
 

/**
 * @param realmService
 */
protected void setRealmService(RealmService realmService) {
    IdentityTenantUtil.setRealmService(realmService);
}
 

/**
 * @param realmService
 */
protected void unsetRealmService(RealmService realmService) {
    IdentityTenantUtil.setRealmService(null);
}