com.orientechnologies.orient.core.metadata.security.OUser Java Examples

@Test
public void testOQueryProvider()
{
	OQueryDataProvider<OUser> provider = new OQueryDataProvider<OUser>("select from OUser where name <> :other", OUser.class);
	provider.setSort("name", SortOrder.ASCENDING);
	provider.setParameter("other", Model.of("blalba"));
	Iterator<OUser> it = provider.iterator(0, -1);
	List<ODocument> allUsers = wicket.getTester().getMetadata().getSecurity().getAllUsers();
	assertTrue(provider.size()==allUsers.size());
	while(it.hasNext())
	{
		OUser oUser = it.next();
		assertTrue(allUsers.contains(provider.model(oUser).getObject().getDocument()));
	}
	provider.detach();
	assertTrue(provider.size()==allUsers.size());
}
 

@Test
public void testOQueryProviderContextVariables()
{
	OQueryDataProvider<OUser> provider = new OQueryDataProvider<OUser>("select from OUser where name = $name", OUser.class);
	provider.setSort("name", SortOrder.ASCENDING);
	provider.setContextVariable("name", Model.of("admin"));
	Iterator<OUser> it = provider.iterator(0, -1);
	assertEquals(1, provider.size());
	assertEquals("admin", it.next().getName());
}
 

@Test
public void testDocumentWrapper() throws Exception
{
	ORID orid = new ORecordId("#5:0"); //Admin ORID
	ODocument adminDocument = orid.getRecord();
	OUser admin = wicket.getTester().getMetadata().getSecurity().getUser("admin");
	DocumentWrapperTransformer<OUser> transformer = new DocumentWrapperTransformer<OUser>(OUser.class);
	assertEquals(admin, transformer.apply(adminDocument));
}
 

/**
 * Changes current connection user. See {@link #executeWithTxUser(
 *com.orientechnologies.orient.core.metadata.security.OSecurityUser, SpecificUserAction)}.
 * <p>
 * LIMITATION: current user must have read right on users table.
 *
 * @param user       user login
 * @param userAction logic to execute with specific user
 * @param <T>        type of returned result (may be Void)
 * @return action result (may be null)
 */
public <T> T executeWithTxUser(final String user, final SpecificUserAction<T> userAction) {
    final boolean userChanged = checkSpecificUserConditions(user);
    final ODatabaseDocument db = connectionProvider.get();
    final T res;
    if (userChanged) {
        // this may cause security exception if current user has no access rights to users table
        final OUser specificUser = db.getMetadata().getSecurity().getUser(user);
        Preconditions.checkState(specificUser != null, "User '%s' not found", user);
        res = executeWithTxUser(specificUser, userAction);
    } else {
        res = executeWithTxUser(db.getUser(), userAction);
    }
    return res;
}
 

@Override
public boolean checkPassword(String userId, String password) {
	
	OPersistenceSession session  = (OPersistenceSession)getSession(PersistenceSession.class);
	OUser oUser = session.getDatabase().getMetadata().getSecurity().getUser(userId);
	return oUser!=null?oUser.checkPassword(password):false;
}
 

@After
public void destroy() {
    DBClosure.sudoConsumer(db -> {
        String sql = String.format("delete from %s where %s = ?", OUser.CLASS_NAME, OrienteerUser.PROP_EMAIL);
        db.command(new OCommandSQL(sql)).execute(testUser.getEmail());
    });
    tester.signOut();
}
 

@Before
public void init() {
    DBClosure.sudoConsumer(db -> {
        user = new OrienteerUser(OUser.CLASS_NAME);
        user.setName(UUID.randomUUID().toString())
                .setPassword(UUID.randomUUID().toString())
                .setAccountStatus(OSecurityUser.STATUSES.ACTIVE);
        user.setEmail(UUID.randomUUID().toString() + "@gmail.com");
        user.save();

        OProperty property = user.getDocument().getSchemaClass().getProperty(OrienteerUser.PROP_RESTORE_ID);
        OrienteerUsersModule.REMOVE_CRON_RULE.setValue(property, "0/7 0/1 * 1/1 * ? *");
        OrienteerUsersModule.REMOVE_SCHEDULE_START_TIMEOUT.setValue(property, "3000");
    });
}
 

@Override
public ODocument onInstall(OrienteerWebApplication app, ODatabaseDocument db) {
    super.onInstall(app, db);
    OSchemaHelper helper = OSchemaHelper.bind(db);

    helper.oClass(OUser.CLASS_NAME)
            .oProperty(PROP_ONLINE, OType.BOOLEAN)
            .oProperty(PROP_LAST_SESSION_FIELD, OType.STRING)
            .switchDisplayable(true, PROP_ONLINE, PROP_LAST_SESSION_FIELD);

    return null;
}
 

private ISessionListener createUserOnlineListener() {
    return new ISessionListener() {
        @Override
        public void onCreated(Session session) {}

        @Override
        public void onUnbound(final String sessionId) {
            DBClosure.sudoConsumer(db -> {
                String sql = String.format("update %s set %s = ? where %s = ?", OUser.CLASS_NAME,
                        PROP_ONLINE, PROP_LAST_SESSION_FIELD);
                db.command(new OCommandSQL(sql)).execute(false, sessionId);
            });
        }
    };
}
 

/**
 * @return currently signed in {@link OUser}. Returns null in case of no user was signed in.
 */
public OSecurityUser getUser()
{
	ODocument userDoc = getUserAsODocument();
	return userDoc!=null?new OUser(userDoc):null;
}
 

public OrienteerUserHook(ODatabaseDocument database) {
    super(database);
    setIncludeClasses(OUser.CLASS_NAME);
}
 

private void resetUsersOnline(ODatabaseDocument db) {
    String sql = String.format("update %s set %s = ?", OUser.CLASS_NAME, PROP_ONLINE);
    db.command(new OCommandSQL(sql)).execute(false);
}