mirror of
https://github.com/prometheus-community/postgres_exporter
synced 2025-04-22 15:05:27 +00:00
refactor pgSetting.normaliseUnit tests
This uses the native go testing library to test the outputs in a more standardized fashion. Signed-off-by: Joe Adams <github@joeadams.io>
This commit is contained in:
Joe Adams
parent
134e908885
commit
78abb57f31
@ -16,6 +16,8 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
"github.com/prometheus/client_golang/prometheus"
|
"github.com/prometheus/client_golang/prometheus"
|
||||||
dto "github.com/prometheus/client_model/go"
|
dto "github.com/prometheus/client_model/go"
|
||||||
. "gopkg.in/check.v1"
|
. "gopkg.in/check.v1"
|
||||||
@ -267,3 +269,226 @@ type fixture struct {
|
|||||||
d string
|
d string
|
||||||
v float64
|
v float64
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Test_pgSetting_normaliseUnit(t *testing.T) {
|
||||||
|
type fields struct {
|
||||||
|
name string
|
||||||
|
setting string
|
||||||
|
unit string
|
||||||
|
shortDesc string
|
||||||
|
vartype string
|
||||||
|
}
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
fields fields
|
||||||
|
wantVal float64
|
||||||
|
wantUnit string
|
||||||
|
wantErr bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Seconds",
|
||||||
|
fields: fields{
|
||||||
|
name: "seconds_fixture_metric",
|
||||||
|
setting: "5",
|
||||||
|
unit: "s",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: 5,
|
||||||
|
wantUnit: "seconds",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_seconds_fixture_metric_seconds", help: "Foo foo foo [Units converted to seconds.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 5,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Milliseconds",
|
||||||
|
fields: fields{
|
||||||
|
name: "milliseconds_fixture_metric",
|
||||||
|
setting: "5000",
|
||||||
|
unit: "ms",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: 5,
|
||||||
|
wantUnit: "seconds",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_milliseconds_fixture_metric_seconds", help: "Foo foo foo [Units converted to seconds.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 5,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "8KB",
|
||||||
|
fields: fields{
|
||||||
|
name: "eight_kb_fixture_metric",
|
||||||
|
setting: "17",
|
||||||
|
unit: "8kB",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: 139264,
|
||||||
|
wantUnit: "bytes",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_eight_kb_fixture_metric_bytes", help: "Foo foo foo [Units converted to bytes.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 139264,
|
||||||
|
},
|
||||||
|
|
||||||
|
{
|
||||||
|
name: "16KB real",
|
||||||
|
fields: fields{
|
||||||
|
name: "16_kb_real_fixture_metric",
|
||||||
|
setting: "3.0",
|
||||||
|
unit: "16kB",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "real",
|
||||||
|
},
|
||||||
|
wantVal: 49152,
|
||||||
|
wantUnit: "bytes",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_16_kb_real_fixture_metric_bytes", help: "Foo foo foo [Units converted to bytes.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 49152,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "16MB real",
|
||||||
|
fields: fields{
|
||||||
|
name: "16_mb_real_fixture_metric",
|
||||||
|
setting: "3.0",
|
||||||
|
unit: "16MB",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "real",
|
||||||
|
},
|
||||||
|
wantVal: 5.0331648e+07,
|
||||||
|
wantUnit: "bytes",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_16_mb_real_fixture_metric_bytes", help: "Foo foo foo [Units converted to bytes.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 5.0331648e+07,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "32MB real",
|
||||||
|
fields: fields{
|
||||||
|
name: "32_mb_real_fixture_metric",
|
||||||
|
setting: "3.0",
|
||||||
|
unit: "32MB",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "real",
|
||||||
|
},
|
||||||
|
wantVal: 1.00663296e+08,
|
||||||
|
wantUnit: "bytes",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_32_mb_real_fixture_metric_bytes", help: "Foo foo foo [Units converted to bytes.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 1.00663296e+08,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "64MB real",
|
||||||
|
fields: fields{
|
||||||
|
name: "64_mb_real_fixture_metric",
|
||||||
|
setting: "3.0",
|
||||||
|
unit: "64MB",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "real",
|
||||||
|
},
|
||||||
|
wantVal: 2.01326592e+08,
|
||||||
|
wantUnit: "bytes",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_64_mb_real_fixture_metric_bytes", help: "Foo foo foo [Units converted to bytes.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 2.01326592e+08,
|
||||||
|
},
|
||||||
|
|
||||||
|
// TODO: normalize doesn't handle boolean, that is currently handled in pgSetting.metric
|
||||||
|
// {
|
||||||
|
// name: "Boolean on",
|
||||||
|
// fields: fields{
|
||||||
|
// name: "bool_on_fixture_metric",
|
||||||
|
// setting: "on",
|
||||||
|
// unit: "",
|
||||||
|
// shortDesc: "Foo foo foo",
|
||||||
|
// vartype: "bool",
|
||||||
|
// },
|
||||||
|
// wantVal: 1,
|
||||||
|
// wantUnit: "",
|
||||||
|
// wantErr: false,
|
||||||
|
// // d: `Desc{fqName: "pg_settings_bool_on_fixture_metric", help: "Foo foo foo", constLabels: {}, variableLabels: []}`,
|
||||||
|
// // v: 1,
|
||||||
|
// },
|
||||||
|
|
||||||
|
// {
|
||||||
|
// name: "Boolean off",
|
||||||
|
// fields: fields{
|
||||||
|
// name: "bool_off_fixture_metric",
|
||||||
|
// setting: "off",
|
||||||
|
// unit: "",
|
||||||
|
// shortDesc: "Foo foo foo",
|
||||||
|
// vartype: "bool",
|
||||||
|
// },
|
||||||
|
// wantVal: 0,
|
||||||
|
// wantUnit: "",
|
||||||
|
// wantErr: false,
|
||||||
|
// // d: `Desc{fqName: "pg_settings_bool_off_fixture_metric", help: "Foo foo foo", constLabels: {}, variableLabels: []}`,
|
||||||
|
// // v: 0,
|
||||||
|
// },
|
||||||
|
{
|
||||||
|
name: "Special -1",
|
||||||
|
fields: fields{
|
||||||
|
name: "special_minus_one_value",
|
||||||
|
setting: "-1",
|
||||||
|
unit: "d",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: -1,
|
||||||
|
wantUnit: "seconds",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_special_minus_one_value_seconds", help: "foo foo foo [Units converted to seconds.]", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: -1,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "RDS superuser",
|
||||||
|
fields: fields{
|
||||||
|
name: "rds.rds_superuser_reserved_connections",
|
||||||
|
setting: "2",
|
||||||
|
unit: "",
|
||||||
|
shortDesc: "Sets the number of connection slots reserved for rds_superusers.",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: 2,
|
||||||
|
wantUnit: "",
|
||||||
|
wantErr: false,
|
||||||
|
// d: `Desc{fqName: "pg_settings_rds_rds_superuser_reserved_connections", help: "Sets the number of connection slots reserved for rds_superusers.", constLabels: {}, variableLabels: []}`,
|
||||||
|
// v: 2,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Unknown unit",
|
||||||
|
fields: fields{
|
||||||
|
name: "10",
|
||||||
|
setting: "nonexistent",
|
||||||
|
unit: "",
|
||||||
|
shortDesc: "Foo foo foo",
|
||||||
|
vartype: "integer",
|
||||||
|
},
|
||||||
|
wantVal: 0,
|
||||||
|
wantUnit: "",
|
||||||
|
wantErr: true,
|
||||||
|
// err: `Unknown unit for runtime variable: "nonexistent"`,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s := &pgSetting{
|
||||||
|
name: tt.fields.name,
|
||||||
|
setting: tt.fields.setting,
|
||||||
|
unit: tt.fields.unit,
|
||||||
|
shortDesc: tt.fields.shortDesc,
|
||||||
|
vartype: tt.fields.vartype,
|
||||||
|
}
|
||||||
|
gotVal, gotUnit, err := s.normaliseUnit()
|
||||||
|
if (err != nil) != tt.wantErr {
|
||||||
|
t.Errorf("pgSetting.normaliseUnit() error = %v, wantErr %v", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if gotVal != tt.wantVal {
|
||||||
|
t.Errorf("pgSetting.normaliseUnit() gotVal = %v, want %v", gotVal, tt.wantVal)
|
||||||
|
}
|
||||||
|
if gotUnit != tt.wantUnit {
|
||||||
|
t.Errorf("pgSetting.normaliseUnit() gotUnit = %v, want %v", gotUnit, tt.wantUnit)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user