/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.felix.useradmin.filestore;
import java.io.File;
import java.util.HashMap;
import java.util.Map;
import junit.framework.TestCase;
import org.apache.felix.useradmin.RoleFactory;
import org.osgi.service.useradmin.Group;
import org.osgi.service.useradmin.Role;
import org.osgi.service.useradmin.User;
/**
* Test case for {@link RoleRepositorySerializer}.
*/
public class RoleRepositoryFileStorePerformanceTest extends TestCase {
private static final int USER_COUNT = 25000;
private static final int GROUP_COUNT = 500;
private Role m_anyone;
private Group[] m_groups;
private User[] m_users;
private Map m_repository;
private RoleRepositoryFileStore m_store;
/**
* Executes the performance test.
*/
public void testPerformanceOk() throws Exception {
allocateMemory();
writeRepositoryPerformanceTest();
releaseMemory();
readRepositoryPerformanceTest();
}
/**
* Does a very simple performance test for a large number of users spread over several groups.
*/
protected void readRepositoryPerformanceTest() throws Exception {
long r_st = System.nanoTime();
Map result = m_store.retrieve();
long r_time = System.nanoTime() - r_st;
assertNotNull(result);
assertEquals(GROUP_COUNT + USER_COUNT + 1, result.size());
System.out.println("Read time : " + (r_time / 1.0e9) + "s.");
}
/**
* {@inheritDoc}
*/
protected void setUp() throws Exception {
super.setUp();
m_store = new RoleRepositoryFileStore(new File(System.getProperty("java.io.tmpdir")), false /* disable background writes */);
m_repository = new HashMap(USER_COUNT + GROUP_COUNT + 1);
m_anyone = RoleFactory.createRole(Role.USER_ANYONE);
addToRepository(m_anyone);
}
/**
* Does a very simple performance test for writing a large number of users spread over several groups.
*/
protected void writeRepositoryPerformanceTest() throws Exception {
long w_st = System.nanoTime();
m_store.store(m_repository);
long w_time = System.nanoTime() - w_st;
System.out.println("Write time: " + (w_time / 1.0e9) + "s.");
}
private void addToRepository(Role role) {
m_repository.put(role.getName(), role);
}
/**
*
*/
private void allocateMemory() {
m_groups = new Group[GROUP_COUNT];
for (int i = 0; i < m_groups.length; i++) {
m_groups[i] = createGroup(i+1);
m_groups[i].addRequiredMember(m_anyone);
addToRepository(m_groups[i]);
}
m_users = new User[USER_COUNT];
for (int i = 0; i < m_users.length; i++) {
m_users[i] = createUser(i+1);
int groupIdx = (i % m_groups.length);
m_groups[groupIdx].addMember(m_users[i]);
addToRepository(m_users[i]);
}
}
private Group createGroup(int idx) {
String name = "Group" + idx;
Group result = RoleFactory.createGroup(name);
setCredentials(result);
setProperties(result);
return result;
}
private User createUser(int idx) {
String name = "User" + idx;
User result = RoleFactory.createUser(name);
setCredentials(result);
setProperties(result);
return result;
}
/**
*
*/
private void releaseMemory() {
m_groups = null;
m_users = null;
System.gc();
System.gc();
System.gc();
}
private void setCredentials(User user) {
user.getCredentials().put("name", user.getName());
user.getCredentials().put("password", "secret");
user.getCredentials().put("certificate", new byte[] { (byte) 0x55, (byte) 0xAA } );
}
private void setProperties(Role role) {
role.getProperties().put("key1", "value1");
role.getProperties().put("key2", "hello world".getBytes());
}
}