/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project 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 io.netty.testsuite.transport.socket;

import io.netty.bootstrap.Bootstrap;
import io.netty.buffer.ByteBufAllocator;
import io.netty.channel.ChannelOption;
import io.netty.testsuite.transport.socket.SocketTestPermutation.Factory;
import io.netty.testsuite.util.TestUtils;
import io.netty.util.NetUtil;
import io.netty.util.internal.StringUtil;
import io.netty.util.internal.logging.InternalLogger;
import io.netty.util.internal.logging.InternalLoggerFactory;
import org.junit.Rule;
import org.junit.rules.TestName;

import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.InetSocketAddress;
import java.util.List;

public abstract class AbstractClientSocketTest {

    private static final List<Factory<Bootstrap>> COMBO = SocketTestPermutation.clientSocket();
    private static final List<ByteBufAllocator> ALLOCATORS = SocketTestPermutation.allocator();

    @Rule
    public final TestName testName = new TestName();

    protected final InternalLogger logger = InternalLoggerFactory.getInstance(getClass());

    protected volatile Bootstrap cb;
    protected volatile InetSocketAddress addr;

    protected void run() throws Throwable {
        for (ByteBufAllocator allocator: ALLOCATORS) {
            int i = 0;
            for (Factory<Bootstrap> e: COMBO) {
                cb = e.newInstance();
                addr = new InetSocketAddress(NetUtil.LOCALHOST, TestUtils.getFreePort());
                cb.remoteAddress(addr);
                cb.option(ChannelOption.ALLOCATOR, allocator);
                logger.info(String.format(
                        "Running: %s %d of %d with %s",
                        testName.getMethodName(), ++ i, COMBO.size(), StringUtil.simpleClassName(allocator)));
                try {
                    Method m = getClass().getDeclaredMethod(
                            testName.getMethodName(), Bootstrap.class);
                    m.invoke(this, cb);
                } catch (InvocationTargetException ex) {
                    throw ex.getCause();
                }
            }
        }
    }
}