mirror of https://github.com/prometheus/prometheus
259 lines
9.6 KiB
Go
259 lines
9.6 KiB
Go
// Copyright 2020 The Prometheus Authors
|
|
// 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 eureka
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/common/model"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/prometheus/prometheus/discovery"
|
|
"github.com/prometheus/prometheus/discovery/targetgroup"
|
|
)
|
|
|
|
func testUpdateServices(respHandler http.HandlerFunc) ([]*targetgroup.Group, error) {
|
|
// Create a test server with mock HTTP handler.
|
|
ts := httptest.NewServer(respHandler)
|
|
defer ts.Close()
|
|
|
|
conf := SDConfig{
|
|
Server: ts.URL,
|
|
}
|
|
|
|
reg := prometheus.NewRegistry()
|
|
refreshMetrics := discovery.NewRefreshMetrics(reg)
|
|
metrics := conf.NewDiscovererMetrics(reg, refreshMetrics)
|
|
err := metrics.Register()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer metrics.Unregister()
|
|
defer refreshMetrics.Unregister()
|
|
|
|
md, err := NewDiscovery(&conf, nil, metrics)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return md.refresh(context.Background())
|
|
}
|
|
|
|
func TestEurekaSDHandleError(t *testing.T) {
|
|
var (
|
|
errTesting = "non 2xx status '500' response during eureka service discovery"
|
|
respHandler = func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Header().Set("Content-Type", "application/xml")
|
|
io.WriteString(w, ``)
|
|
}
|
|
)
|
|
tgs, err := testUpdateServices(respHandler)
|
|
|
|
require.EqualError(t, err, errTesting)
|
|
require.Empty(t, tgs)
|
|
}
|
|
|
|
func TestEurekaSDEmptyList(t *testing.T) {
|
|
var (
|
|
appsXML = `<applications>
|
|
<versions__delta>1</versions__delta>
|
|
<apps__hashcode/>
|
|
</applications>`
|
|
respHandler = func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Header().Set("Content-Type", "application/xml")
|
|
io.WriteString(w, appsXML)
|
|
}
|
|
)
|
|
tgs, err := testUpdateServices(respHandler)
|
|
require.NoError(t, err)
|
|
require.Len(t, tgs, 1)
|
|
}
|
|
|
|
func TestEurekaSDSendGroup(t *testing.T) {
|
|
var (
|
|
appsXML = `<applications>
|
|
<versions__delta>1</versions__delta>
|
|
<apps__hashcode>UP_4_</apps__hashcode>
|
|
<application>
|
|
<name>CONFIG-SERVICE</name>
|
|
<instance>
|
|
<instanceId>config-service001.test.com:config-service:8080</instanceId>
|
|
<hostName>config-service001.test.com</hostName>
|
|
<app>CONFIG-SERVICE</app>
|
|
<ipAddr>192.133.83.31</ipAddr>
|
|
<status>UP</status>
|
|
<overriddenstatus>UNKNOWN</overriddenstatus>
|
|
<port enabled="true">8080</port>
|
|
<securePort enabled="false">8080</securePort>
|
|
<countryId>1</countryId>
|
|
<dataCenterInfo class="com.netflix.appinfo.InstanceInfo$DefaultDataCenterInfo">
|
|
<name>MyOwn</name>
|
|
</dataCenterInfo>
|
|
<leaseInfo>
|
|
<renewalIntervalInSecs>30</renewalIntervalInSecs>
|
|
<durationInSecs>90</durationInSecs>
|
|
<registrationTimestamp>1596003469304</registrationTimestamp>
|
|
<lastRenewalTimestamp>1596110179310</lastRenewalTimestamp>
|
|
<evictionTimestamp>0</evictionTimestamp>
|
|
<serviceUpTimestamp>1547190033103</serviceUpTimestamp>
|
|
</leaseInfo>
|
|
<metadata>
|
|
<instanceId>config-service001.test.com:config-service:8080</instanceId>
|
|
</metadata>
|
|
<homePageUrl>http://config-service001.test.com:8080/</homePageUrl>
|
|
<statusPageUrl>http://config-service001.test.com:8080/info</statusPageUrl>
|
|
<healthCheckUrl>http://config-service001.test.com 8080/health</healthCheckUrl>
|
|
<vipAddress>config-service</vipAddress>
|
|
<isCoordinatingDiscoveryServer>false</isCoordinatingDiscoveryServer>
|
|
<lastUpdatedTimestamp>1596003469304</lastUpdatedTimestamp>
|
|
<lastDirtyTimestamp>1596003469304</lastDirtyTimestamp>
|
|
<actionType>ADDED</actionType>
|
|
</instance>
|
|
<instance>
|
|
<instanceId>config-service002.test.com:config-service:8080</instanceId>
|
|
<hostName>config-service002.test.com</hostName>
|
|
<app>CONFIG-SERVICE</app>
|
|
<ipAddr>192.133.83.31</ipAddr>
|
|
<status>UP</status>
|
|
<overriddenstatus>UNKNOWN</overriddenstatus>
|
|
<port enabled="true">8080</port>
|
|
<securePort enabled="false">8080</securePort>
|
|
<countryId>1</countryId>
|
|
<dataCenterInfo class="com.netflix.appinfo.InstanceInfo$DefaultDataCenterInfo">
|
|
<name>MyOwn</name>
|
|
</dataCenterInfo>
|
|
<leaseInfo>
|
|
<renewalIntervalInSecs>30</renewalIntervalInSecs>
|
|
<durationInSecs>90</durationInSecs>
|
|
<registrationTimestamp>1596003469304</registrationTimestamp>
|
|
<lastRenewalTimestamp>1596110179310</lastRenewalTimestamp>
|
|
<evictionTimestamp>0</evictionTimestamp>
|
|
<serviceUpTimestamp>1547190033103</serviceUpTimestamp>
|
|
</leaseInfo>
|
|
<metadata>
|
|
<instanceId>config-service002.test.com:config-service:8080</instanceId>
|
|
</metadata>
|
|
<homePageUrl>http://config-service002.test.com:8080/</homePageUrl>
|
|
<statusPageUrl>http://config-service002.test.com:8080/info</statusPageUrl>
|
|
<healthCheckUrl>http://config-service002.test.com:8080/health</healthCheckUrl>
|
|
<vipAddress>config-service</vipAddress>
|
|
<isCoordinatingDiscoveryServer>false</isCoordinatingDiscoveryServer>
|
|
<lastUpdatedTimestamp>1596003469304</lastUpdatedTimestamp>
|
|
<lastDirtyTimestamp>1596003469304</lastDirtyTimestamp>
|
|
<actionType>ADDED</actionType>
|
|
</instance>
|
|
</application>
|
|
<application>
|
|
<name>META-SERVICE</name>
|
|
<instance>
|
|
<instanceId>meta-service002.test.com:meta-service:8080</instanceId>
|
|
<hostName>meta-service002.test.com</hostName>
|
|
<app>META-SERVICE</app>
|
|
<ipAddr>192.133.87.237</ipAddr>
|
|
<status>UP</status>
|
|
<overriddenstatus>UNKNOWN</overriddenstatus>
|
|
<port enabled="true">8080</port>
|
|
<securePort enabled="false">443</securePort>
|
|
<countryId>1</countryId>
|
|
<dataCenterInfo class="com.netflix.appinfo.InstanceInfo$DefaultDataCenterInfo">
|
|
<name>MyOwn</name>
|
|
</dataCenterInfo>
|
|
<leaseInfo>
|
|
<renewalIntervalInSecs>30</renewalIntervalInSecs>
|
|
<durationInSecs>90</durationInSecs>
|
|
<registrationTimestamp>1535444352472</registrationTimestamp>
|
|
<lastRenewalTimestamp>1596110168846</lastRenewalTimestamp>
|
|
<evictionTimestamp>0</evictionTimestamp>
|
|
<serviceUpTimestamp>1535444352472</serviceUpTimestamp>
|
|
</leaseInfo>
|
|
<metadata>
|
|
<project>meta-service</project>
|
|
<management.port>8090</management.port>
|
|
</metadata>
|
|
<homePageUrl>http://meta-service002.test.com:8080/</homePageUrl>
|
|
<statusPageUrl>http://meta-service002.test.com:8080/info</statusPageUrl>
|
|
<healthCheckUrl>http://meta-service002.test.com:8080/health</healthCheckUrl>
|
|
<vipAddress>meta-service</vipAddress>
|
|
<secureVipAddress>meta-service</secureVipAddress>
|
|
<isCoordinatingDiscoveryServer>false</isCoordinatingDiscoveryServer>
|
|
<lastUpdatedTimestamp>1535444352472</lastUpdatedTimestamp>
|
|
<lastDirtyTimestamp>1535444352398</lastDirtyTimestamp>
|
|
<actionType>ADDED</actionType>
|
|
</instance>
|
|
<instance>
|
|
<instanceId>meta-service001.test.com:meta-service:8080</instanceId>
|
|
<hostName>meta-service001.test.com</hostName>
|
|
<app>META-SERVICE</app>
|
|
<ipAddr>192.133.87.236</ipAddr>
|
|
<status>UP</status>
|
|
<overriddenstatus>UNKNOWN</overriddenstatus>
|
|
<port enabled="true">8080</port>
|
|
<securePort enabled="false">443</securePort>
|
|
<countryId>1</countryId>
|
|
<dataCenterInfo class="com.netflix.appinfo.InstanceInfo$DefaultDataCenterInfo">
|
|
<name>MyOwn</name>
|
|
</dataCenterInfo>
|
|
<leaseInfo>
|
|
<renewalIntervalInSecs>30</renewalIntervalInSecs>
|
|
<durationInSecs>90</durationInSecs>
|
|
<registrationTimestamp>1535444352472</registrationTimestamp>
|
|
<lastRenewalTimestamp>1596110168846</lastRenewalTimestamp>
|
|
<evictionTimestamp>0</evictionTimestamp>
|
|
<serviceUpTimestamp>1535444352472</serviceUpTimestamp>
|
|
</leaseInfo>
|
|
<metadata>
|
|
<project>meta-service</project>
|
|
<management.port>8090</management.port>
|
|
</metadata>
|
|
<homePageUrl>http://meta-service001.test.com:8080/</homePageUrl>
|
|
<statusPageUrl>http://meta-service001.test.com:8080/info</statusPageUrl>
|
|
<healthCheckUrl>http://meta-service001.test.com:8080/health</healthCheckUrl>
|
|
<vipAddress>meta-service</vipAddress>
|
|
<secureVipAddress>meta-service</secureVipAddress>
|
|
<isCoordinatingDiscoveryServer>false</isCoordinatingDiscoveryServer>
|
|
<lastUpdatedTimestamp>1535444352472</lastUpdatedTimestamp>
|
|
<lastDirtyTimestamp>1535444352398</lastDirtyTimestamp>
|
|
<actionType>ADDED</actionType>
|
|
</instance>
|
|
</application>
|
|
</applications>`
|
|
respHandler = func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
w.Header().Set("Content-Type", "application/xml")
|
|
io.WriteString(w, appsXML)
|
|
}
|
|
)
|
|
|
|
tgs, err := testUpdateServices(respHandler)
|
|
require.NoError(t, err)
|
|
require.Len(t, tgs, 1)
|
|
|
|
tg := tgs[0]
|
|
require.Equal(t, "eureka", tg.Source)
|
|
require.Len(t, tg.Targets, 4)
|
|
|
|
tgt := tg.Targets[0]
|
|
require.Equal(t, tgt[model.AddressLabel], model.LabelValue("config-service001.test.com:8080"))
|
|
|
|
tgt = tg.Targets[2]
|
|
require.Equal(t, tgt[model.AddressLabel], model.LabelValue("meta-service002.test.com:8080"))
|
|
}
|