blob: 19e143cd7393dce1e529fafa05333ce839942117 [file] [log] [blame]
/*
* 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 skywalking_test
import (
"context"
"github.com/apache/servicecomb-mesher/proxy/config"
"github.com/apache/servicecomb-mesher/proxy/pkg/skywalking"
gcconfig "github.com/go-chassis/go-chassis/core/config"
"github.com/go-chassis/go-chassis/core/config/model"
"github.com/go-chassis/go-chassis/core/invocation"
"github.com/stretchr/testify/assert"
"testing"
)
const (
Port = ":49800"
ServerUrl = "127.0.0.1:49800"
)
//initConfig
func initConfig() {
var micCfg model.ServiceSpec
micCfg.Name = "TEST"
gcconfig.MicroserviceDefinition = &micCfg
}
//initMesherConfig
func initMesherConfig() {
config.SetConfig(&config.MesherConfig{ServiceComb: &config.ServiceComb{config.APM{config.Tracing{Enable: true, ServerURI: "192.168.0.1:17289"}}}})
}
//initInv
func initInv() *invocation.Invocation {
var i invocation.Invocation
i.MicroServiceName = "test"
i.Ctx = context.Background()
i.Endpoint = "calculator"
i.URLPathFormat = "/bmi"
return &i
}
//TestInit init skywalking manager
func TestInit(t *testing.T) {
initConfig()
initMesherConfig()
skywalking.Init()
assert.NotEqual(t, gcconfig.MicroserviceDefinition, nil)
}
//TestCreateEntrySpan test skywalking manager creating entryspan
func TestCreateEntrySpan(t *testing.T) {
initConfig()
initMesherConfig()
skywalking.Init()
span, _, err := skywalking.CreateEntrySpan(initInv())
assert.Equal(t, err, nil)
assert.NotEqual(t, span, nil)
span.End()
}
//TestCreateExitSpan test skywalking manager creating endspan
func TestCreateExitSpan(t *testing.T) {
initConfig()
initMesherConfig()
skywalking.Init()
inv := initInv()
span, ctx, err := skywalking.CreateEntrySpan(inv)
assert.Equal(t, err, nil)
assert.NotEqual(t, span, nil)
spanExit, err := skywalking.CreateExitSpan(ctx, inv)
assert.Equal(t, err, nil)
assert.NotEqual(t, spanExit, nil)
spanExit.End()
span.End()
}
//TestCreateLocalSpan test skywalking manager creating localspan
func TestCreateLocalSpan(t *testing.T) {
initConfig()
initMesherConfig()
skywalking.Init()
span, _, err := skywalking.CreateLocalSpan(context.Background())
assert.Equal(t, err, nil)
assert.NotEqual(t, span, nil)
}