2016-12-28 02:16:47 +00:00
|
|
|
// Copyright 2016 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 triton
|
|
|
|
|
|
|
|
import (
|
2017-10-25 04:21:42 +00:00
|
|
|
"context"
|
2016-12-28 02:16:47 +00:00
|
|
|
"fmt"
|
|
|
|
"net"
|
|
|
|
"net/http"
|
|
|
|
"net/http/httptest"
|
|
|
|
"net/url"
|
|
|
|
"strconv"
|
|
|
|
"testing"
|
|
|
|
"time"
|
|
|
|
|
2017-10-25 04:21:42 +00:00
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
|
2018-01-11 15:10:25 +00:00
|
|
|
"github.com/prometheus/common/config"
|
2016-12-28 02:16:47 +00:00
|
|
|
"github.com/prometheus/common/model"
|
Refactor SD configuration to remove `config` dependency (#3629)
* refactor: move targetGroup struct and CheckOverflow() to their own package
* refactor: move auth and security related structs to a utility package, fix import error in utility package
* refactor: Azure SD, remove SD struct from config
* refactor: DNS SD, remove SD struct from config into dns package
* refactor: ec2 SD, move SD struct from config into the ec2 package
* refactor: file SD, move SD struct from config to file discovery package
* refactor: gce, move SD struct from config to gce discovery package
* refactor: move HTTPClientConfig and URL into util/config, fix import error in httputil
* refactor: consul, move SD struct from config into consul discovery package
* refactor: marathon, move SD struct from config into marathon discovery package
* refactor: triton, move SD struct from config to triton discovery package, fix test
* refactor: zookeeper, move SD structs from config to zookeeper discovery package
* refactor: openstack, remove SD struct from config, move into openstack discovery package
* refactor: kubernetes, move SD struct from config into kubernetes discovery package
* refactor: notifier, use targetgroup package instead of config
* refactor: tests for file, marathon, triton SD - use targetgroup package instead of config.TargetGroup
* refactor: retrieval, use targetgroup package instead of config.TargetGroup
* refactor: storage, use config util package
* refactor: discovery manager, use targetgroup package instead of config.TargetGroup
* refactor: use HTTPClient and TLS config from configUtil instead of config
* refactor: tests, use targetgroup package instead of config.TargetGroup
* refactor: fix tagetgroup.Group pointers that were removed by mistake
* refactor: openstack, kubernetes: drop prefixes
* refactor: remove import aliases forced due to vscode bug
* refactor: move main SD struct out of config into discovery/config
* refactor: rename configUtil to config_util
* refactor: rename yamlUtil to yaml_config
* refactor: kubernetes, remove prefixes
* refactor: move the TargetGroup package to discovery/
* refactor: fix order of imports
2017-12-29 20:01:34 +00:00
|
|
|
"github.com/prometheus/prometheus/discovery/targetgroup"
|
2016-12-28 02:16:47 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
Refactor SD configuration to remove `config` dependency (#3629)
* refactor: move targetGroup struct and CheckOverflow() to their own package
* refactor: move auth and security related structs to a utility package, fix import error in utility package
* refactor: Azure SD, remove SD struct from config
* refactor: DNS SD, remove SD struct from config into dns package
* refactor: ec2 SD, move SD struct from config into the ec2 package
* refactor: file SD, move SD struct from config to file discovery package
* refactor: gce, move SD struct from config to gce discovery package
* refactor: move HTTPClientConfig and URL into util/config, fix import error in httputil
* refactor: consul, move SD struct from config into consul discovery package
* refactor: marathon, move SD struct from config into marathon discovery package
* refactor: triton, move SD struct from config to triton discovery package, fix test
* refactor: zookeeper, move SD structs from config to zookeeper discovery package
* refactor: openstack, remove SD struct from config, move into openstack discovery package
* refactor: kubernetes, move SD struct from config into kubernetes discovery package
* refactor: notifier, use targetgroup package instead of config
* refactor: tests for file, marathon, triton SD - use targetgroup package instead of config.TargetGroup
* refactor: retrieval, use targetgroup package instead of config.TargetGroup
* refactor: storage, use config util package
* refactor: discovery manager, use targetgroup package instead of config.TargetGroup
* refactor: use HTTPClient and TLS config from configUtil instead of config
* refactor: tests, use targetgroup package instead of config.TargetGroup
* refactor: fix tagetgroup.Group pointers that were removed by mistake
* refactor: openstack, kubernetes: drop prefixes
* refactor: remove import aliases forced due to vscode bug
* refactor: move main SD struct out of config into discovery/config
* refactor: rename configUtil to config_util
* refactor: rename yamlUtil to yaml_config
* refactor: kubernetes, remove prefixes
* refactor: move the TargetGroup package to discovery/
* refactor: fix order of imports
2017-12-29 20:01:34 +00:00
|
|
|
conf = SDConfig{
|
2016-12-28 02:16:47 +00:00
|
|
|
Account: "testAccount",
|
|
|
|
DNSSuffix: "triton.example.com",
|
|
|
|
Endpoint: "127.0.0.1",
|
|
|
|
Port: 443,
|
|
|
|
Version: 1,
|
|
|
|
RefreshInterval: 1,
|
|
|
|
TLSConfig: config.TLSConfig{InsecureSkipVerify: true},
|
|
|
|
}
|
Refactor SD configuration to remove `config` dependency (#3629)
* refactor: move targetGroup struct and CheckOverflow() to their own package
* refactor: move auth and security related structs to a utility package, fix import error in utility package
* refactor: Azure SD, remove SD struct from config
* refactor: DNS SD, remove SD struct from config into dns package
* refactor: ec2 SD, move SD struct from config into the ec2 package
* refactor: file SD, move SD struct from config to file discovery package
* refactor: gce, move SD struct from config to gce discovery package
* refactor: move HTTPClientConfig and URL into util/config, fix import error in httputil
* refactor: consul, move SD struct from config into consul discovery package
* refactor: marathon, move SD struct from config into marathon discovery package
* refactor: triton, move SD struct from config to triton discovery package, fix test
* refactor: zookeeper, move SD structs from config to zookeeper discovery package
* refactor: openstack, remove SD struct from config, move into openstack discovery package
* refactor: kubernetes, move SD struct from config into kubernetes discovery package
* refactor: notifier, use targetgroup package instead of config
* refactor: tests for file, marathon, triton SD - use targetgroup package instead of config.TargetGroup
* refactor: retrieval, use targetgroup package instead of config.TargetGroup
* refactor: storage, use config util package
* refactor: discovery manager, use targetgroup package instead of config.TargetGroup
* refactor: use HTTPClient and TLS config from configUtil instead of config
* refactor: tests, use targetgroup package instead of config.TargetGroup
* refactor: fix tagetgroup.Group pointers that were removed by mistake
* refactor: openstack, kubernetes: drop prefixes
* refactor: remove import aliases forced due to vscode bug
* refactor: move main SD struct out of config into discovery/config
* refactor: rename configUtil to config_util
* refactor: rename yamlUtil to yaml_config
* refactor: kubernetes, remove prefixes
* refactor: move the TargetGroup package to discovery/
* refactor: fix order of imports
2017-12-29 20:01:34 +00:00
|
|
|
badconf = SDConfig{
|
2016-12-28 02:16:47 +00:00
|
|
|
Account: "badTestAccount",
|
|
|
|
DNSSuffix: "bad.triton.example.com",
|
|
|
|
Endpoint: "127.0.0.1",
|
|
|
|
Port: 443,
|
|
|
|
Version: 1,
|
|
|
|
RefreshInterval: 1,
|
|
|
|
TLSConfig: config.TLSConfig{
|
|
|
|
InsecureSkipVerify: false,
|
|
|
|
KeyFile: "shouldnotexist.key",
|
|
|
|
CAFile: "shouldnotexist.ca",
|
|
|
|
CertFile: "shouldnotexist.cert",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestTritonSDNew(t *testing.T) {
|
2017-08-11 18:45:52 +00:00
|
|
|
td, err := New(nil, &conf)
|
2016-12-28 02:16:47 +00:00
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.NotNil(t, td)
|
|
|
|
assert.NotNil(t, td.client)
|
|
|
|
assert.NotNil(t, td.interval)
|
|
|
|
assert.NotNil(t, td.sdConfig)
|
|
|
|
assert.Equal(t, conf.Account, td.sdConfig.Account)
|
|
|
|
assert.Equal(t, conf.DNSSuffix, td.sdConfig.DNSSuffix)
|
|
|
|
assert.Equal(t, conf.Endpoint, td.sdConfig.Endpoint)
|
|
|
|
assert.Equal(t, conf.Port, td.sdConfig.Port)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTritonSDNewBadConfig(t *testing.T) {
|
2017-08-11 18:45:52 +00:00
|
|
|
td, err := New(nil, &badconf)
|
2016-12-28 02:16:47 +00:00
|
|
|
assert.NotNil(t, err)
|
|
|
|
assert.Nil(t, td)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTritonSDRun(t *testing.T) {
|
|
|
|
var (
|
2017-08-11 18:45:52 +00:00
|
|
|
td, err = New(nil, &conf)
|
Refactor SD configuration to remove `config` dependency (#3629)
* refactor: move targetGroup struct and CheckOverflow() to their own package
* refactor: move auth and security related structs to a utility package, fix import error in utility package
* refactor: Azure SD, remove SD struct from config
* refactor: DNS SD, remove SD struct from config into dns package
* refactor: ec2 SD, move SD struct from config into the ec2 package
* refactor: file SD, move SD struct from config to file discovery package
* refactor: gce, move SD struct from config to gce discovery package
* refactor: move HTTPClientConfig and URL into util/config, fix import error in httputil
* refactor: consul, move SD struct from config into consul discovery package
* refactor: marathon, move SD struct from config into marathon discovery package
* refactor: triton, move SD struct from config to triton discovery package, fix test
* refactor: zookeeper, move SD structs from config to zookeeper discovery package
* refactor: openstack, remove SD struct from config, move into openstack discovery package
* refactor: kubernetes, move SD struct from config into kubernetes discovery package
* refactor: notifier, use targetgroup package instead of config
* refactor: tests for file, marathon, triton SD - use targetgroup package instead of config.TargetGroup
* refactor: retrieval, use targetgroup package instead of config.TargetGroup
* refactor: storage, use config util package
* refactor: discovery manager, use targetgroup package instead of config.TargetGroup
* refactor: use HTTPClient and TLS config from configUtil instead of config
* refactor: tests, use targetgroup package instead of config.TargetGroup
* refactor: fix tagetgroup.Group pointers that were removed by mistake
* refactor: openstack, kubernetes: drop prefixes
* refactor: remove import aliases forced due to vscode bug
* refactor: move main SD struct out of config into discovery/config
* refactor: rename configUtil to config_util
* refactor: rename yamlUtil to yaml_config
* refactor: kubernetes, remove prefixes
* refactor: move the TargetGroup package to discovery/
* refactor: fix order of imports
2017-12-29 20:01:34 +00:00
|
|
|
ch = make(chan []*targetgroup.Group)
|
2016-12-28 02:16:47 +00:00
|
|
|
ctx, cancel = context.WithCancel(context.Background())
|
|
|
|
)
|
|
|
|
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.NotNil(t, td)
|
|
|
|
|
|
|
|
go td.Run(ctx, ch)
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-time.After(60 * time.Millisecond):
|
|
|
|
// Expected.
|
|
|
|
case tgs := <-ch:
|
|
|
|
t.Fatalf("Unexpected target groups in triton discovery: %s", tgs)
|
|
|
|
}
|
|
|
|
|
|
|
|
cancel()
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTritonSDRefreshNoTargets(t *testing.T) {
|
|
|
|
tgts := testTritonSDRefresh(t, "{\"containers\":[]}")
|
|
|
|
assert.Nil(t, tgts)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTritonSDRefreshMultipleTargets(t *testing.T) {
|
|
|
|
var (
|
|
|
|
dstr = `{"containers":[
|
|
|
|
{
|
|
|
|
"server_uuid":"44454c4c-5000-104d-8037-b7c04f5a5131",
|
|
|
|
"vm_alias":"server01",
|
2017-04-06 21:09:23 +00:00
|
|
|
"vm_brand":"lx",
|
2016-12-28 02:16:47 +00:00
|
|
|
"vm_image_uuid":"7b27a514-89d7-11e6-bee6-3f96f367bee7",
|
|
|
|
"vm_uuid":"ad466fbf-46a2-4027-9b64-8d3cdb7e9072"
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"server_uuid":"a5894692-bd32-4ca1-908a-e2dda3c3a5e6",
|
|
|
|
"vm_alias":"server02",
|
2017-04-06 21:09:23 +00:00
|
|
|
"vm_brand":"kvm",
|
2016-12-28 02:16:47 +00:00
|
|
|
"vm_image_uuid":"a5894692-bd32-4ca1-908a-e2dda3c3a5e6",
|
|
|
|
"vm_uuid":"7b27a514-89d7-11e6-bee6-3f96f367bee7"
|
|
|
|
}]
|
|
|
|
}`
|
|
|
|
)
|
|
|
|
|
|
|
|
tgts := testTritonSDRefresh(t, dstr)
|
|
|
|
assert.NotNil(t, tgts)
|
|
|
|
assert.Equal(t, 2, len(tgts))
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestTritonSDRefreshNoServer(t *testing.T) {
|
|
|
|
var (
|
2017-08-11 18:45:52 +00:00
|
|
|
td, err = New(nil, &conf)
|
2016-12-28 02:16:47 +00:00
|
|
|
)
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.NotNil(t, td)
|
|
|
|
|
|
|
|
tg, rerr := td.refresh()
|
|
|
|
assert.NotNil(t, rerr)
|
|
|
|
assert.Contains(t, rerr.Error(), "an error occurred when requesting targets from the discovery endpoint.")
|
|
|
|
assert.NotNil(t, tg)
|
|
|
|
assert.Nil(t, tg.Targets)
|
|
|
|
}
|
|
|
|
|
|
|
|
func testTritonSDRefresh(t *testing.T, dstr string) []model.LabelSet {
|
|
|
|
var (
|
2017-08-11 18:45:52 +00:00
|
|
|
td, err = New(nil, &conf)
|
2016-12-28 02:16:47 +00:00
|
|
|
s = httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
fmt.Fprintln(w, dstr)
|
|
|
|
}))
|
|
|
|
)
|
|
|
|
|
|
|
|
defer s.Close()
|
|
|
|
|
|
|
|
u, uperr := url.Parse(s.URL)
|
|
|
|
assert.Nil(t, uperr)
|
|
|
|
assert.NotNil(t, u)
|
|
|
|
|
|
|
|
host, strport, sherr := net.SplitHostPort(u.Host)
|
|
|
|
assert.Nil(t, sherr)
|
|
|
|
assert.NotNil(t, host)
|
|
|
|
assert.NotNil(t, strport)
|
|
|
|
|
|
|
|
port, atoierr := strconv.Atoi(strport)
|
|
|
|
assert.Nil(t, atoierr)
|
|
|
|
assert.NotNil(t, port)
|
|
|
|
|
|
|
|
td.sdConfig.Port = port
|
|
|
|
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.NotNil(t, td)
|
|
|
|
|
|
|
|
tg, err := td.refresh()
|
|
|
|
assert.Nil(t, err)
|
|
|
|
assert.NotNil(t, tg)
|
|
|
|
|
|
|
|
return tg.Targets
|
|
|
|
}
|