File: client_test.go

package info (click to toggle)
golang-github-go-kit-kit 0.13.0-8
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,784 kB
  • sloc: sh: 22; makefile: 11
file content (80 lines) | stat: -rw-r--r-- 1,615 bytes parent folder | download | duplicates (2)
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
package etcdv3

import (
	"context"
	"testing"
	"time"

	"google.golang.org/grpc"
)

const (
	// irrelevantEndpoint is an address which does not exists.
	irrelevantEndpoint = "http://irrelevant:12345"
)

func TestNewClient(t *testing.T) {
	client, err := NewClient(
		context.Background(),
		[]string{irrelevantEndpoint},
		ClientOptions{
			DialTimeout:   3 * time.Second,
			DialKeepAlive: 3 * time.Second,
		},
	)
	if err != nil {
		t.Fatalf("unexpected error creating client: %v", err)
	}
	if client == nil {
		t.Fatal("expected new Client, got nil")
	}
}

func TestClientOptions(t *testing.T) {
	client, err := NewClient(
		context.Background(),
		[]string{},
		ClientOptions{
			Cert:          "",
			Key:           "",
			CACert:        "",
			DialTimeout:   3 * time.Second,
			DialKeepAlive: 3 * time.Second,
		},
	)
	if err == nil {
		t.Errorf("expected error: %v", err)
	}
	if client != nil {
		t.Fatalf("expected client to be nil on failure")
	}

	_, err = NewClient(
		context.Background(),
		[]string{irrelevantEndpoint},
		ClientOptions{
			Cert:          "does-not-exist.crt",
			Key:           "does-not-exist.key",
			CACert:        "does-not-exist.CACert",
			DialTimeout:   3 * time.Second,
			DialKeepAlive: 3 * time.Second,
		},
	)
	if err == nil {
		t.Errorf("expected error: %v", err)
	}

	client, err = NewClient(
		context.Background(),
		[]string{irrelevantEndpoint},
		ClientOptions{
			DialOptions: []grpc.DialOption{grpc.WithBlock()},
		},
	)
	if err == nil {
		t.Errorf("expected connection should fail")
	}
	if client != nil {
		t.Errorf("expected client to be nil on failure")
	}
}