init

test/org/apache/catalina/realm/TestGenericPrincipal.java

@@ -0,0 +1,89 @@
+/*
+* 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 String PASSWORD = "pwd";
+    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, PASSWORD, ROLES);
+        doTest(gpIn);
+    }
+
+    @Test
+    public void testSerialize02() throws ClassNotFoundException, IOException {
+        GenericPrincipal gpIn = new GenericPrincipal(USER, PASSWORD, ROLES, PRINCIPAL);
+        doTest(gpIn);
+    }
+
+    @Test
+    public void testSerialize03() throws ClassNotFoundException, IOException {
+        GenericPrincipal gpIn = new GenericPrincipal(USER, PASSWORD, 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.assertEquals(gpIn.getPassword(), gpOut.getPassword());
+        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();
+    }
+}