mirror of
https://github.com/prometheus/prometheus.git
synced 2026-03-27 04:44:19 -04:00
f7c60bf97e
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Signed-off-by: Bartlomiej Plotka <bwplotka@gmail.com>
108 lines
3 KiB
Go
108 lines
3 KiB
Go
// Copyright 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 compliance
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"html/template"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/prometheus/compliance/remotewrite/sender"
|
|
)
|
|
|
|
const (
|
|
scrapeConfigTemplate = `
|
|
global:
|
|
scrape_interval: 1s
|
|
|
|
remote_write:
|
|
- url: "{{.RemoteWriteEndpointURL}}"
|
|
protobuf_message: "{{.RemoteWriteMessage}}"
|
|
send_exemplars: true
|
|
queue_config:
|
|
retry_on_http_429: true
|
|
metadata_config:
|
|
send: true
|
|
|
|
scrape_configs:
|
|
- job_name: "{{.ScrapeTargetJobName}}"
|
|
scrape_interval: 1s
|
|
scrape_protocols:
|
|
- PrometheusProto
|
|
- OpenMetricsText1.0.0
|
|
- PrometheusText0.0.4
|
|
static_configs:
|
|
- targets: ["{{.ScrapeTargetHostPort}}"]
|
|
`
|
|
)
|
|
|
|
var scrapeConfigTmpl = template.Must(template.New("config").Parse(scrapeConfigTemplate))
|
|
|
|
type internalPrometheus struct {
|
|
agentMode bool
|
|
}
|
|
|
|
func (p internalPrometheus) Name() string { return "internal-prometheus" }
|
|
|
|
// Run runs a <REPO>cmd/prometheus main package as a test sender target, until ctx is done.
|
|
func (p internalPrometheus) Run(ctx context.Context, opts sender.Options) error {
|
|
var buf bytes.Buffer
|
|
if err := scrapeConfigTmpl.Execute(&buf, opts); err != nil {
|
|
return fmt.Errorf("failed to execute config template: %w", err)
|
|
}
|
|
|
|
dir, err := os.MkdirTemp("", "test-*")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
configFile := filepath.Join(dir, "config.yaml")
|
|
if err := os.WriteFile(configFile, buf.Bytes(), 0o600); err != nil {
|
|
return err
|
|
}
|
|
defer os.RemoveAll(dir)
|
|
|
|
args := []string{
|
|
"run", ".",
|
|
"--web.listen-address=0.0.0.0:0",
|
|
fmt.Sprintf("--config.file=%s", configFile),
|
|
// Set important flags for the full remote write compliance:
|
|
"--enable-feature=st-storage",
|
|
}
|
|
if p.agentMode {
|
|
args = append(args, fmt.Sprintf("--storage.agent.path=%v", dir), "--agent")
|
|
} else {
|
|
args = append(args, fmt.Sprintf("--storage.tsdb.path=%v", dir))
|
|
}
|
|
return sender.RunCommand(ctx, "../cmd/prometheus", nil, "go", args...)
|
|
}
|
|
|
|
var _ sender.Sender = internalPrometheus{}
|
|
|
|
// TestRemoteWriteSender runs remote write sender compliance tests defined in
|
|
// https://github.com/prometheus/compliance/tree/main/remotewrite/sender against
|
|
// both agent and server modes.
|
|
func TestRemoteWriteSender(t *testing.T) {
|
|
t.Run("mode=server", func(t *testing.T) {
|
|
t.Parallel()
|
|
sender.RunTests(t, internalPrometheus{}, sender.ComplianceTests())
|
|
})
|
|
t.Run("mode=agent", func(t *testing.T) {
|
|
t.Parallel()
|
|
sender.RunTests(t, internalPrometheus{agentMode: true}, sender.ComplianceTests())
|
|
})
|
|
}
|