diff options
author | 2019-03-14 08:12:28 +0100 | |
---|---|---|
committer | 2019-03-14 07:12:28 +0000 | |
commit | 7b6cb76237d151ffa056c742ec281a17548fa089 (patch) | |
tree | 2100dd26d9e41f0be6fb365de9aeb334d7a94075 /plugin/grpc/setup_policy_test.go | |
parent | 0d8e1cf8b4ac157ef08837d0ff1e83597b8d1211 (diff) | |
download | coredns-7b6cb76237d151ffa056c742ec281a17548fa089.tar.gz coredns-7b6cb76237d151ffa056c742ec281a17548fa089.tar.zst coredns-7b6cb76237d151ffa056c742ec281a17548fa089.zip |
plugin/grpc: New gRPC plugin (#2667)
* plugin/grpc: New gRPC plugin
* some changes after the first review:
- remove healthcheck. gRPC already has this implicitly implemented
- some naming and stetic changes
- fix some comments
- other minor fixes
* plugin/grpc: New gRPC plugin
* some changes after the first review:
- remove healthcheck. gRPC already has this implicitly implemented
- some naming and stetic changes
- fix some comments
- other minor fixes
* add OWNERS file and change plugin order
* remove Rcode checker
Diffstat (limited to 'plugin/grpc/setup_policy_test.go')
-rw-r--r-- | plugin/grpc/setup_policy_test.go | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/plugin/grpc/setup_policy_test.go b/plugin/grpc/setup_policy_test.go new file mode 100644 index 000000000..db7da6262 --- /dev/null +++ b/plugin/grpc/setup_policy_test.go @@ -0,0 +1,47 @@ +package grpc + +import ( + "strings" + "testing" + + "github.com/mholt/caddy" +) + +func TestSetupPolicy(t *testing.T) { + tests := []struct { + input string + shouldErr bool + expectedPolicy string + expectedErr string + }{ + // positive + {"grpc . 127.0.0.1 {\npolicy random\n}\n", false, "random", ""}, + {"grpc . 127.0.0.1 {\npolicy round_robin\n}\n", false, "round_robin", ""}, + {"grpc . 127.0.0.1 {\npolicy sequential\n}\n", false, "sequential", ""}, + // negative + {"grpc . 127.0.0.1 {\npolicy random2\n}\n", true, "random", "unknown policy"}, + } + + for i, test := range tests { + c := caddy.NewTestController("dns", test.input) + g, err := parseGRPC(c) + + if test.shouldErr && err == nil { + t.Errorf("Test %d: expected error but found %s for input %s", i, err, test.input) + } + + if err != nil { + if !test.shouldErr { + t.Errorf("Test %d: expected no error but found one for input %s, got: %v", i, test.input, err) + } + + if !strings.Contains(err.Error(), test.expectedErr) { + t.Errorf("Test %d: expected error to contain: %v, found error: %v, input: %s", i, test.expectedErr, err, test.input) + } + } + + if !test.shouldErr && g.p.String() != test.expectedPolicy { + t.Errorf("Test %d: expected: %s, got: %s", i, test.expectedPolicy, g.p.String()) + } + } +} |