/*
* 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.catalina.realm;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import org.junit.Assert;
import org.junit.Test;
public class TestGenericPrincipal {
private static final String USER = "user";
private static final List<String> ROLES = Collections.unmodifiableList(
Arrays.asList(new String[] { "ROLE1", "ROLE2" }));
private static final TesterPrincipal PRINCIPAL = new TesterPrincipal("Principal");
private static final TesterPrincipalNonSerializable PRINCIPAL_NON_SERIALIZABLE =
new TesterPrincipalNonSerializable("PrincipalNonSerializable");
@Test
public void testSerialize01() throws ClassNotFoundException, IOException {
GenericPrincipal gpIn = new GenericPrincipal(USER, ROLES);
doTest(gpIn);
}
@Test
public void testSerialize02() throws ClassNotFoundException, IOException {
GenericPrincipal gpIn = new GenericPrincipal(USER, ROLES, PRINCIPAL);
doTest(gpIn);
}
@Test
public void testSerialize03() throws ClassNotFoundException, IOException {
GenericPrincipal gpIn = new GenericPrincipal(USER, ROLES, PRINCIPAL_NON_SERIALIZABLE);
doTest(gpIn);
}
private void doTest(GenericPrincipal gpIn)
throws ClassNotFoundException, IOException {
GenericPrincipal gpOut = serializeAndDeserialize(gpIn);
Assert.assertNull(gpOut.getGssCredential());
Assert.assertEquals(gpIn.getName(), gpOut.getName());
Assert.assertArrayEquals(gpIn.getRoles(), gpOut.getRoles());
if (gpIn == gpIn.getUserPrincipal()) {
Assert.assertEquals(gpOut, gpOut.getUserPrincipal());
} else if (gpIn.getUserPrincipal() instanceof Serializable) {
Assert.assertEquals(gpIn.getUserPrincipal(), gpOut.getUserPrincipal());
} else {
Assert.assertEquals(gpOut, gpOut.getUserPrincipal());
}
}
private GenericPrincipal serializeAndDeserialize(GenericPrincipal gpIn)
throws IOException, ClassNotFoundException {
ByteArrayOutputStream bos = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(bos);
oos.writeObject(gpIn);
byte[] data = bos.toByteArray();
ByteArrayInputStream bis = new ByteArrayInputStream(data);
ObjectInputStream ois = new ObjectInputStream(bis);
return (GenericPrincipal) ois.readObject();
}
}