go-viya/batch_servers_test.go at main · dingdayu/go-viya · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package viya

import (
	"net/http"
	"net/http/httptest"
	"net/url"
	"testing"
)

func TestGetBatchServersListDecodesServers(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if got, want := r.Method, http.MethodGet; got != want {
			t.Fatalf("method = %q, want %q", got, want)
		}
		if got, want := r.URL.Path, "/batch/servers"; got != want {
			t.Fatalf("path = %q, want %q", got, want)
		}

		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{"count":1,"items":[{"id":"server-1","contextId":"context-1","hostName":"host","state":"available","isPermanent":true,"version":1}]}`))
	}))
	defer server.Close()

	u, err := url.Parse(server.URL)
	if err != nil {
		t.Fatalf("url.Parse() error = %v", err)
	}
	client := NewClient(t.Context(), WithBaseURL(u))

	servers, err := client.GetBatchServersList(t.Context())
	if err != nil {
		t.Fatalf("GetBatchServersList() error = %v", err)
	}
	if got, want := servers.Count, 1; got != want {
		t.Fatalf("Count = %d, want %d", got, want)
	}
	if len(servers.Items) != 1 {
		t.Fatalf("len(Items) = %d, want 1", len(servers.Items))
	}
	if got, want := servers.Items[0].State, "available"; got != want {
		t.Fatalf("State = %q, want %q", got, want)
	}
	if got, want := servers.Items[0].HostName, "host"; got != want {
		t.Fatalf("HostName = %q, want %q", got, want)
	}
}

func TestGetBatchServerInfoDecodesServer(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if got, want := r.Method, http.MethodGet; got != want {
			t.Fatalf("method = %q, want %q", got, want)
		}
		if got, want := r.RequestURI, "/batch/servers/server%201"; got != want {
			t.Fatalf("request URI = %q, want %q", got, want)
		}

		w.Header().Set("Content-Type", "application/json")
		_, _ = w.Write([]byte(`{"id":"server 1","contextId":"context-1","hostName":"host","processId":"proc-1","queueName":"default","state":"busy","version":1}`))
	}))
	defer server.Close()

	u, err := url.Parse(server.URL)
	if err != nil {
		t.Fatalf("url.Parse() error = %v", err)
	}
	client := NewClient(t.Context(), WithBaseURL(u))

	serverInfo, err := client.GetBatchServerInfo(t.Context(), "server 1")
	if err != nil {
		t.Fatalf("GetBatchServerInfo() error = %v", err)
	}
	if got, want := serverInfo.ID, "server 1"; got != want {
		t.Fatalf("ID = %q, want %q", got, want)
	}
	if got, want := serverInfo.ProcessID, "proc-1"; got != want {
		t.Fatalf("ProcessID = %q, want %q", got, want)
	}
}

func TestDeleteBatchServerSendsDelete(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if got, want := r.Method, http.MethodDelete; got != want {
			t.Fatalf("method = %q, want %q", got, want)
		}
		if got, want := r.RequestURI, "/batch/servers/server%201"; got != want {
			t.Fatalf("request URI = %q, want %q", got, want)
		}
		if got, want := r.Header.Get("Accept"), "application/vnd.sas.error+json"; got != want {
			t.Fatalf("Accept = %q, want %q", got, want)
		}

		w.WriteHeader(http.StatusNoContent)
	}))
	defer server.Close()

	u, err := url.Parse(server.URL)
	if err != nil {
		t.Fatalf("url.Parse() error = %v", err)
	}
	client := NewClient(t.Context(), WithBaseURL(u))

	err = client.DeleteBatchServer(t.Context(), "server 1")
	if err != nil {
		t.Fatalf("DeleteBatchServer() error = %v", err)
	}
}