/*
 * Copyright (C) 2009 The Android Open Source Project
 *
 * 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.conscrypt;

import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;
import javax.net.ssl.SSLSession;
import junit.framework.TestCase;
import libcore.javax.net.ssl.FakeSSLSession;

public final class ClientSessionContextTest extends TestCase {

    public void testSimpleAddition() {
        ClientSessionContext context = new ClientSessionContext();
        SSLSession a = new ValidSSLSession("a");
        SSLSession b = new ValidSSLSession("b");

        context.putSession(a);
        assertSessionContextContents(context, new SSLSession[] { a }, new SSLSession[] { b });

        context.putSession(b);
        assertSessionContextContents(context, new SSLSession[] { a, b }, new SSLSession[0]);
    }

    public void testTrimToSize() {
        ClientSessionContext context = new ClientSessionContext();
        ValidSSLSession a = new ValidSSLSession("a");
        ValidSSLSession b = new ValidSSLSession("b");
        ValidSSLSession c = new ValidSSLSession("c");
        ValidSSLSession d = new ValidSSLSession("d");

        context.putSession(a);
        assertSessionContextContents(context, new SSLSession[] { a }, new SSLSession[] { b, c, d });

        context.putSession(b);
        assertSessionContextContents(context, new SSLSession[] { a, b }, new SSLSession[] { c, d });

        context.putSession(c);
        assertSessionContextContents(context, new SSLSession[] { a, b, c }, new SSLSession[] { d });

        context.putSession(d);
        assertSessionContextContents(context, new SSLSession[] { a, b, c, d }, new SSLSession[0]);

        context.setSessionCacheSize(2);
        assertSessionContextContents(context, new SSLSession[] { c, d }, new SSLSession[] { a, b });
    }

    public void testImplicitRemovalOfOldest() {
        ClientSessionContext context = new ClientSessionContext();
        context.setSessionCacheSize(2);
        ValidSSLSession a = new ValidSSLSession("a");
        ValidSSLSession b = new ValidSSLSession("b");
        ValidSSLSession c = new ValidSSLSession("c");
        ValidSSLSession d = new ValidSSLSession("d");

        context.putSession(a);
        assertSessionContextContents(context, new SSLSession[] { a }, new SSLSession[] { b, c, d });

        context.putSession(b);
        assertSessionContextContents(context, new SSLSession[] { a, b }, new SSLSession[] { c, d });

        context.putSession(c);
        assertSessionContextContents(context, new SSLSession[] { b, c }, new SSLSession[] { a, d });

        context.putSession(d);
        assertSessionContextContents(context, new SSLSession[] { c, d }, new SSLSession[] { a, b });
    }

    private static void assertSessionContextContents(ClientSessionContext context,
                                                     SSLSession[] contains,
                                                     SSLSession[] exludes) {
        assertEquals(contains.length, context.size());

        for (SSLSession s : contains) {
            assertSame(s.getPeerHost(), s, context.getSession(s.getId()));
            assertSame(s.getPeerHost(), s, context.getSession(s.getPeerHost(), 443));
        }
        for (SSLSession s : exludes) {
            assertNull(s.getPeerHost(), context.getSession(s.getId()));
            assertNull(s.getPeerHost(), context.getSession(s.getPeerHost(), 443));
        }

        Set<SSLSession> sessions = new HashSet<SSLSession>();
        Enumeration<byte[]> ids = context.getIds();
        while (ids.hasMoreElements()) {
            byte[] id = ids.nextElement();
            sessions.add(context.getSession(id));
        }

        Set<SSLSession> expected = new HashSet<SSLSession>();
        for (SSLSession s : sessions) {
            expected.add(s);
        }
        assertEquals(expected, sessions);
    }

    static class ValidSSLSession extends FakeSSLSession {
        ValidSSLSession(String host) {
            super(host);
        }
        @Override public boolean isValid() {
            return true;
        }
    }
}