RepoMirrors/postgres_exporter
1
0
Fork 0
mirror of https://github.com/prometheus-community/postgres_exporter synced 2025-02-02 11:31:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

use check for tests

Browse Source
This commit is contained in:
Matthias Berkenkamp 2017-08-01 08:23:52 +02:00 committed by Will Rouesnel
parent fce6b005a6
commit 142716dcb0
1 changed files with 28 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
postgres_exporter_test.go
View File

@ -86,66 +86,66 @@ func (s *FunctionalSuite) TestSemanticVersionColumnDiscard(c *C) {
} }
} }
func TestEnvironmentSettingWithSecretsFiles(t *testing.T) { // test read username and password from file
func (s *FunctionalSuite) TestEnvironmentSettingWithSecretsFiles(c *C) {
err := os.Setenv("DATA_SOURCE_USER_FILE", "./tests/username_file") err := os.Setenv("DATA_SOURCE_USER_FILE", "./tests/username_file")
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_USER_FILE could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_USER_FILE")
}
err = os.Setenv("DATA_SOURCE_PASS_FILE", "./tests/userpass_file") err = os.Setenv("DATA_SOURCE_PASS_FILE", "./tests/userpass_file")
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_PASS_FILE could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_PASS_FILE")
}
err = os.Setenv("DATA_SOURCE_URI", "localhost:5432/?sslmode=disable") err = os.Setenv("DATA_SOURCE_URI", "localhost:5432/?sslmode=disable")
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_URI could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_URI")
}
var expected = "postgresql://custom_username:custom_password@localhost:5432/?sslmode=disable" var expected = "postgresql://custom_username:custom_password@localhost:5432/?sslmode=disable"
dsn := getDataSource() dsn := getDataSource()
if dsn != expected { if dsn != expected {
t.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, expected) c.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, expected)
} }
} }
func TestEnvironmentSettingWithDns(t *testing.T) { // test read DATA_SOURCE_NAME from environment
func (s *FunctionalSuite) TestEnvironmentSettingWithDns(c *C) {
envDsn := "postgresql://user:password@localhost:5432/?sslmode=enabled" envDsn := "postgresql://user:password@localhost:5432/?sslmode=enabled"
err := os.Setenv("DATA_SOURCE_NAME", envDsn) err := os.Setenv("DATA_SOURCE_NAME", envDsn)
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_NAME could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_NAME")
}
dsn := getDataSource() dsn := getDataSource()
if dsn != envDsn { if dsn != envDsn {
t.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, envDsn) c.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, envDsn)
} }
} }
// test DATA_SOURCE_NAME is used even if username and password environment wariables are set // test DATA_SOURCE_NAME is used even if username and password environment variables are set
func TestEnvironmentSettingWithDnsAndSecrets(t *testing.T) { func (s *FunctionalSuite) TestEnvironmentSettingWithDnsAndSecrets(c *C) {
envDsn := "postgresql://userDsn:passwordDsn@localhost:55432/?sslmode=disabled" envDsn := "postgresql://userDsn:passwordDsn@localhost:55432/?sslmode=disabled"
err := os.Setenv("DATA_SOURCE_NAME", envDsn) err := os.Setenv("DATA_SOURCE_NAME", envDsn)
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_NAME could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_NAME")
}
err = os.Setenv("DATA_SOURCE_USER_FILE", "./tests/username_file") err = os.Setenv("DATA_SOURCE_USER_FILE", "./tests/username_file")
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_USER_FILE could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_USER_FILE")
}
err = os.Setenv("DATA_SOURCE_PASS", "envUserPass") err = os.Setenv("DATA_SOURCE_PASS", "envUserPass")
if err != nil { c.Assert(err, IsNil)
t.Errorf("DATA_SOURCE_PASS could not be set") defer UnsetEnvironment(c, "DATA_SOURCE_PASS")
}
dsn := getDataSource() dsn := getDataSource()
if dsn != envDsn { if dsn != envDsn {
t.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, envDsn) c.Errorf("Expected Username to be read from file. Found=%v, expected=%v", dsn, envDsn)
} }
} }
func UnsetEnvironment(c *C, d string) {
err := os.Unsetenv(d)
c.Assert(err, IsNil)
}