mirror of
				https://github.com/optim-enterprises-bv/vault.git
				synced 2025-10-31 18:48:08 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			108 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			108 lines
		
	
	
		
			2.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package http
 | |
| 
 | |
| import (
 | |
| 	"net/http"
 | |
| 	"net/url"
 | |
| 	"reflect"
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/hashicorp/vault/vault"
 | |
| )
 | |
| 
 | |
| func TestSysHealth_get(t *testing.T) {
 | |
| 	core, _, root := vault.TestCoreUnsealed(t)
 | |
| 	ln, addr := TestServer(t, core)
 | |
| 	defer ln.Close()
 | |
| 
 | |
| 	resp, err := http.Get(addr + "/v1/sys/health")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 
 | |
| 	var actual map[string]interface{}
 | |
| 	expected := map[string]interface{}{
 | |
| 		"initialized": true,
 | |
| 		"sealed":      false,
 | |
| 		"standby":     false,
 | |
| 	}
 | |
| 	testResponseStatus(t, resp, 200)
 | |
| 	testResponseBody(t, resp, &actual)
 | |
| 	expected["server_time_utc"] = actual["server_time_utc"]
 | |
| 	if !reflect.DeepEqual(actual, expected) {
 | |
| 		t.Fatalf("bad: %#v", actual)
 | |
| 	}
 | |
| 
 | |
| 	core.Seal(root)
 | |
| 
 | |
| 	resp, err = http.Get(addr + "/v1/sys/health")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 
 | |
| 	actual = map[string]interface{}{}
 | |
| 	expected = map[string]interface{}{
 | |
| 		"initialized": true,
 | |
| 		"sealed":      true,
 | |
| 		"standby":     false,
 | |
| 	}
 | |
| 	testResponseStatus(t, resp, 500)
 | |
| 	testResponseBody(t, resp, &actual)
 | |
| 	expected["server_time_utc"] = actual["server_time_utc"]
 | |
| 	if !reflect.DeepEqual(actual, expected) {
 | |
| 		t.Fatalf("bad: %#v", actual)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func TestSysHealth_customcodes(t *testing.T) {
 | |
| 	core, _, root := vault.TestCoreUnsealed(t)
 | |
| 	ln, addr := TestServer(t, core)
 | |
| 	defer ln.Close()
 | |
| 
 | |
| 	queryurl, err := url.Parse(addr + "/v1/sys/health?sealedcode=503&activecode=202")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 	resp, err := http.Get(queryurl.String())
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 
 | |
| 	var actual map[string]interface{}
 | |
| 	expected := map[string]interface{}{
 | |
| 		"initialized": true,
 | |
| 		"sealed":      false,
 | |
| 		"standby":     false,
 | |
| 	}
 | |
| 	testResponseStatus(t, resp, 202)
 | |
| 	testResponseBody(t, resp, &actual)
 | |
| 
 | |
| 	expected["server_time_utc"] = actual["server_time_utc"]
 | |
| 	if !reflect.DeepEqual(actual, expected) {
 | |
| 		t.Fatalf("bad: %#v", actual)
 | |
| 	}
 | |
| 
 | |
| 	core.Seal(root)
 | |
| 
 | |
| 	queryurl, err = url.Parse(addr + "/v1/sys/health?sealedcode=503&activecode=202")
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 	resp, err = http.Get(queryurl.String())
 | |
| 	if err != nil {
 | |
| 		t.Fatalf("err: %s", err)
 | |
| 	}
 | |
| 
 | |
| 	actual = map[string]interface{}{}
 | |
| 	expected = map[string]interface{}{
 | |
| 		"initialized": true,
 | |
| 		"sealed":      true,
 | |
| 		"standby":     false,
 | |
| 	}
 | |
| 	testResponseStatus(t, resp, 503)
 | |
| 	testResponseBody(t, resp, &actual)
 | |
| 	expected["server_time_utc"] = actual["server_time_utc"]
 | |
| 	if !reflect.DeepEqual(actual, expected) {
 | |
| 		t.Fatalf("bad: %#v", actual)
 | |
| 	}
 | |
| }
 | 
