io.servicecomb.core.unittest.UnitTestMeta Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of java-chassis-core Show documentation
There is a newer version: 0.5.0
/*
 * Copyright 2017 Huawei Technologies Co., Ltd
 *
 * 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 io.servicecomb.core.unittest;

import java.util.Collections;
import java.util.List;

import io.servicecomb.core.handler.ConsumerHandlerManager;
import org.mockito.Mockito;
import org.springframework.context.ApplicationContext;

import io.servicecomb.core.Handler;
import io.servicecomb.core.definition.MicroserviceMeta;
import io.servicecomb.core.definition.MicroserviceMetaManager;
import io.servicecomb.core.definition.SchemaMeta;
import io.servicecomb.core.definition.loader.SchemaLoader;
import io.servicecomb.core.handler.ProducerHandlerManager;
import io.servicecomb.core.handler.config.Config;
import io.servicecomb.core.handler.impl.SimpleLoadBalanceHandler;
import io.servicecomb.serviceregistry.RegistryUtils;
import io.servicecomb.serviceregistry.api.registry.Microservice;
import io.servicecomb.swagger.generator.core.unittest.UnitTestSwaggerUtils;
import io.servicecomb.foundation.common.utils.BeanUtils;

import io.swagger.models.Swagger;
import mockit.Mock;
import mockit.MockUp;

public class UnitTestMeta {
    private static boolean inited = false;

    @SuppressWarnings("unchecked")
    public static synchronized void init() {
        if (inited) {
            return;
        }

        Config config = new Config();
        Class cls = SimpleLoadBalanceHandler.class;
        config.getHandlerClassMap().put("simpleLB", (Class) cls);
        ProducerHandlerManager.INSTANCE.init(config);
        ConsumerHandlerManager.INSTANCE.init(config);

        ApplicationContext applicationContext = Mockito.mock(ApplicationContext.class);
        BeanUtils.setContext(applicationContext);
        inited = true;
    }

    static {
        init();
    }

    private MicroserviceMetaManager microserviceMetaManager = new MicroserviceMetaManager();

    private SchemaLoader schemaLoader = new SchemaLoader() {
        public void putSelfBasePathIfAbsent(String microserviceName, String basePath) {
        };
    };

    private Microservice microservice = new Microservice();

    public UnitTestMeta() {
        microservice.setAppId("app");
        microservice.setServiceName("testname");
        new MockUp() {
            @Mock
            private Microservice createMicroserviceFromDefinition() {
                return microservice;
            }
        };

        new MockUp() {
            @Mock
            public List getOrCreate(String name) {
                return Collections.emptyList();
            }
        };
        new MockUp() {
            @Mock
            public List getOrCreate(String name) {
                return Collections.emptyList();
            }
        };

        schemaLoader.setMicroserviceMetaManager(microserviceMetaManager);
    }

    public void setMicroservice(Microservice microservice) {
        this.microservice = microservice;
    }

    public MicroserviceMetaManager getMicroserviceMetaManager() {
        return microserviceMetaManager;
    }

    public SchemaMeta getOrCreateSchemaMeta(Class impl) {
        return getOrCreateSchemaMeta("app", "test", impl.getName(), impl);
    }

    public SchemaMeta getOrCreateSchemaMeta(String appId, String microserviceName, String schemaId, Class impl) {
        String longName = appId + ":" + microserviceName;
        MicroserviceMeta microserviceMeta = microserviceMetaManager.getOrCreateMicroserviceMeta(longName);
        SchemaMeta schemaMeta = microserviceMeta.findSchemaMeta(schemaId);
        if (schemaMeta != null) {
            return schemaMeta;
        }

        Swagger swagger = UnitTestSwaggerUtils.generateSwagger(impl).getSwagger();
        return schemaLoader.registerSchema(microserviceMeta, schemaId, swagger);
    }
}