/*
* Copyright (c) 2012, Paul Merlin. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*/
package org.qi4j.library.shiro;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.authc.credential.PasswordService;
import org.apache.shiro.subject.Subject;
import org.junit.Test;
import org.qi4j.api.common.Visibility;
import org.qi4j.api.entity.EntityBuilder;
import org.qi4j.api.injection.scope.Service;
import org.qi4j.api.injection.scope.Structure;
import org.qi4j.api.mixin.Mixins;
import org.qi4j.api.structure.Module;
import org.qi4j.api.unitofwork.UnitOfWork;
import org.qi4j.api.unitofwork.UnitOfWorkCompletionException;
import org.qi4j.bootstrap.AssemblyException;
import org.qi4j.bootstrap.ModuleAssembly;
import org.qi4j.index.rdf.assembly.RdfMemoryStoreAssembler;
import org.qi4j.library.shiro.assembly.PasswordDomainAssembler;
import org.qi4j.library.shiro.assembly.StandaloneShiroAssembler;
import org.qi4j.library.shiro.domain.passwords.PasswordSecurable;
import org.qi4j.library.shiro.ini.ShiroIniConfiguration;
import org.qi4j.test.AbstractQi4jTest;
import org.qi4j.test.EntityTestAssembler;
import static org.junit.Assert.*;
public class PasswordDomainTest
extends AbstractQi4jTest
{
// START SNIPPET: domain
public interface User
extends PasswordSecurable
{
}
// END SNIPPET: domain
// START SNIPPET: domain
@Mixins( UserFactoryMixin.class )
public interface UserFactory
{
User createNewUser( String username, String password );
}
// END SNIPPET: domain
// START SNIPPET: domain
public static class UserFactoryMixin
implements UserFactory
{
@Structure
private Module module;
@Service
private PasswordService passwordService;
@Override
public User createNewUser( String username, String password )
{
EntityBuilder<User> userBuilder = module.currentUnitOfWork().newEntityBuilder( User.class );
User user = userBuilder.instance();
user.subjectIdentifier().set( username );
user.password().set( passwordService.encryptPassword( password ) );
return userBuilder.newInstance();
}
}
// END SNIPPET: domain
// START SNIPPET: assembly
@Override
public void assemble( ModuleAssembly module )
throws AssemblyException
{
// END SNIPPET: assembly
new EntityTestAssembler().assemble( module );
new RdfMemoryStoreAssembler().assemble( module );
ModuleAssembly configModule = module;
// START SNIPPET: assembly
new StandaloneShiroAssembler().
withConfig( configModule, Visibility.layer ).
assemble( module );
new PasswordDomainAssembler().
withConfig( configModule, Visibility.layer ).
assemble( module );
module.entities( User.class );
module.services( UserFactory.class );
// END SNIPPET: assembly
configModule.forMixin( ShiroIniConfiguration.class ).
declareDefaults().
iniResourcePath().set( "classpath:standalone-shiro.ini" );
// START SNIPPET: assembly
}
// END SNIPPET: assembly
@Test
public void test()
throws UnitOfWorkCompletionException
{
UnitOfWork uow = module.newUnitOfWork();
UserFactory userFactory = module.findService( UserFactory.class ).get();
// START SNIPPET: usage
User user = userFactory.createNewUser( "foo", "bar" );
// END SNIPPET: usage
uow.complete();
uow = module.newUnitOfWork();
// START SNIPPET: usage
Subject currentUser = SecurityUtils.getSubject();
currentUser.login( new UsernamePasswordToken( "foo", "bar" ) );
// END SNIPPET: usage
assertNotNull( "Unable to authenticate against PasswordRealmService", currentUser.getPrincipal() );
assertFalse( currentUser.hasRole( "role-one" ) );
uow.discard();
}
}