func TestTCPConsumerStartBadCases(t *testing.T) {
consumer := NewMockConsumer()
tcpConsumer := NewConsumer("qiuweyqjw", consumer)
err := tcpConsumer.Start()
assert.Regexp(t, regexp.MustCompile("missing port.*"), err.Error())
tcpConsumer = NewConsumer("0.0.0.0:qwe", consumer)
err = tcpConsumer.Start()
assert.Regexp(t, regexp.MustCompile("unknown port.*"), err.Error())
tcpConsumer = NewConsumer("0.0.0.0:22", consumer)
err = tcpConsumer.Start()
assert.NotEqual(t, nil, err.Error())
_, err = NewClient(addr)
assert.Regexp(t, regexp.MustCompile(".*connection refused.*"), err.Error())
}
func TestTCPConsumer(t *testing.T) {
consumer := NewMockConsumer()
tcpConsumer := NewConsumer(addr, consumer)
err := tcpConsumer.Start()
assert.Equal(t, nil, err)
client, err := NewClient(addr)
assert.Equal(t, nil, err)
// add
// good case
err = client.Add("asd", 1)
assert.Equal(t, nil, err)
// add existing
err = client.Add("asd", 1)
assert.Regexp(t, regexp.MustCompile(".*already exists.*"), err.Error())
// remove
// good case
err = client.Remove("asd", 1)
assert.Equal(t, nil, err)
// remove non-existing
err = client.Remove("asd", 1)
assert.Regexp(t, regexp.MustCompile(".*does not exist.*"), err.Error())
// assignments
// empty assignments
assignments, err := client.Assignment()
assert.Equal(t, nil, err)
assert.Empty(t, assignments)
// add one
err = client.Add("asd", 1)
assert.Equal(t, nil, err)
// check now
assignments, err = client.Assignment()
assert.Equal(t, nil, err)
assert.Len(t, assignments, 1)
// remove that one and check again
err = client.Remove("asd", 1)
assert.Equal(t, nil, err)
assignments, err = client.Assignment()
assert.Equal(t, nil, err)
assert.Empty(t, assignments)
// commit
// commit something
err = client.Commit("asd", 1, 123)
assert.Equal(t, nil, err)
// commit error
consumer.commitOffsetError = errors.New("boom")
err = client.Commit("asd", 1, 123)
assert.Regexp(t, regexp.MustCompile(".*boom.*"), err)
consumer.commitOffsetError = nil
// get offset
// non-existing
_, err = client.Offset("asd", 1)
assert.Regexp(t, regexp.MustCompile(".*does not exist.*"), err.Error())
// add back
err = client.Add("asd", 1)
assert.Equal(t, nil, err)
// check again
offset, err := client.Offset("asd", 1)
assert.Equal(t, nil, err)
assert.Equal(t, int64(123), offset)
// commit offset
err = client.Commit("asd", 1, 234)
assert.Equal(t, nil, err)
// check once more
offset, err = client.Offset("asd", 1)
assert.Equal(t, nil, err)
assert.Equal(t, int64(234), offset)
// set offset
// remove previous
err = client.Remove("asd", 1)
assert.Equal(t, nil, err)
// non-existing
err = client.SetOffset("asd", 1, 123)
assert.Regexp(t, regexp.MustCompile(".*does not exist.*"), err.Error())
// add first
err = client.Add("asd", 1)
assert.Equal(t, nil, err)
// check again
err = client.SetOffset("asd", 1, 123)
assert.Equal(t, nil, err)
// get offset
offset, err = client.Offset("asd", 1)
assert.Equal(t, nil, err)
assert.Equal(t, int64(123), offset)
// lag
// remove previous
err = client.Remove("asd", 1)
assert.Equal(t, nil, err)
// non-existing
_, err = client.Lag("asd", 1)
assert.Regexp(t, regexp.MustCompile(".*does not exist.*"), err.Error())
// add first
err = client.Add("asd", 1)
assert.Equal(t, nil, err)
// check again
lag, err := client.Lag("asd", 1)
assert.Equal(t, nil, err)
assert.Equal(t, int64(100), lag)
// custom handler
tcpConsumer.RegisterCustomHandler("custom", func(body []byte) (*Response, error) {
return NewResponse(true, "", "hello!"), nil
})
response, err := client.CustomRequest("custom", nil)
assert.Equal(t, nil, err)
var responseValue string
err = json.Unmarshal(response, &responseValue)
assert.Equal(t, nil, err)
assert.Equal(t, "hello!", responseValue)
// unknown key
_, err = client.CustomRequest("foobar", nil)
assert.Regexp(t, regexp.MustCompile("Unknown request key.*"), err.Error())
// await termination
success := make(chan struct{})
go func() {
tcpConsumer.AwaitTermination()
success <- struct{}{}
}()
select {
case <-success:
t.Fatal("Await termination shouldn't unblock until consumer is not stopped")
case <-time.After(timeout):
}
tcpConsumer.Stop()
select {
case <-success:
case <-time.After(timeout):
t.Fatalf("Await termination failed to unblock within %s", timeout)
}
}