diff --git a/zk_test.go b/zk_test.go new file mode 100644 index 0000000..4f05b46 --- /dev/null +++ b/zk_test.go @@ -0,0 +1,69 @@ +/** + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package realis_test + +import ( + "log" + "os" + "testing" + "time" + + "github.com/paypal/gorealis" + "github.com/stretchr/testify/assert" +) + +var backoff realis.Backoff = realis.Backoff{ // Reduce penalties for this test to make it quick + Steps: 5, + Duration: 1 * time.Second, + Factor: 1.0, + Jitter: 0.1} + +// Test for behavior when no endpoints are given to the ZK leader finding function. +func TestZKNoEndpoints(t *testing.T) { + _, err := realis.LeaderFromZKOpts() + + assert.Error(t, err) +} + +// Test for behavior when no path is given to the ZK leader finding function. +func TestZKNoPath(t *testing.T) { + _, err := realis.LeaderFromZKOpts(realis.ZKEndpoints("127.0.0.1:2181")) + + assert.Error(t, err) +} + +// Test for behavior when a valid but non-existent path is given to the ZK leader finding function. +func TestZKPathDoesntExist(t *testing.T) { + _, err := realis.LeaderFromZKOpts(realis.ZKEndpoints("127.0.0.1:2181"), + realis.ZKPath("/somepath"), + realis.ZKBackoff(backoff), + realis.ZKLogger(log.New(os.Stdout, "realis-debug: ", log.Ldate))) + + assert.True(t, realis.IsTimeout(err), "a non-existent path should result in a timeout behaving error") + + retryErr := realis.ToRetryCount(err) + assert.NotNil(t, retryErr, "conversion to retry error failed") + assert.Equal(t, backoff.Steps, retryErr.RetryCount(), "retry count is off") +} + +// Test for behavior when an invalid Zookeeper path is passed. Should fail right away. +func TestZKBadPath(t *testing.T) { + _, err := realis.LeaderFromZKOpts(realis.ZKEndpoints("127.0.0.1:2181"), + realis.ZKPath("invalidpath"), + realis.ZKBackoff(backoff), + realis.ZKLogger(log.New(os.Stdout, "realis-debug: ", log.Ldate))) + + assert.False(t, realis.IsTimeout(err), "a bad path should result in a NON-timeout behaving error") +}