scottk/vault
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/vault.git synced 2025-11-01 19:17:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add test for custiom mssql revoke statement

Browse Source
This commit is contained in:
Brian Kassouf
2017-05-01 15:43:21 -07:00
parent 6ca436cdf5
commit 66630f642d
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
plugins/database/mssql/mssql_test.go
View File

@@ -122,6 +122,26 @@ func TestMSSQL_RevokeUser(t *testing.T) {
if err := testCredsExist(t, connURL, username, password); err == nil { if err := testCredsExist(t, connURL, username, password); err == nil {
t.Fatal("Credentials were not revoked") t.Fatal("Credentials were not revoked")
} }
username, password, err = db.CreateUser(statements, "test", time.Now().Add(2*time.Second))
if err != nil {
t.Fatalf("err: %s", err)
}
if err = testCredsExist(t, connURL, username, password); err != nil {
t.Fatalf("Could not connect with new credentials: %s", err)
}
// Test custom revoke statememt
statements.RevocationStatements = testMSSQLDrop
err = db.RevokeUser(statements, username)
if err != nil {
t.Fatalf("err: %s", err)
}
if err := testCredsExist(t, connURL, username, password); err == nil {
t.Fatal("Credentials were not revoked")
}
} }
func testCredsExist(t testing.TB, connURL, username, password string) error { func testCredsExist(t testing.TB, connURL, username, password string) error {
@@ -140,3 +160,8 @@ const testMSSQLRole = `
CREATE LOGIN [{{name}}] WITH PASSWORD = '{{password}}'; CREATE LOGIN [{{name}}] WITH PASSWORD = '{{password}}';
CREATE USER [{{name}}] FOR LOGIN [{{name}}]; CREATE USER [{{name}}] FOR LOGIN [{{name}}];
GRANT SELECT, INSERT, UPDATE, DELETE ON SCHEMA::dbo TO [{{name}}];` GRANT SELECT, INSERT, UPDATE, DELETE ON SCHEMA::dbo TO [{{name}}];`
const testMSSQLDrop = `
DROP USER [{{name}}];
DROP LOGIN [{{name}}];
`