github.com/glide-im/glide@v1.6.0/pkg/store/kafka_store_consumer_test.go (about)

     1  package store
     2  
     3  import (
     4  	"encoding/json"
     5  	"github.com/glide-im/glide/pkg/logger"
     6  	"github.com/glide-im/glide/pkg/messages"
     7  	"testing"
     8  	"time"
     9  )
    10  
    11  func TestNewConsumer(t *testing.T) {
    12  
    13  	consumer, err := NewKafkaConsumer([]string{"localhost:9092"})
    14  	if err != nil {
    15  		t.Error(consumer)
    16  	}
    17  	defer consumer.Close()
    18  
    19  	consumer.ConsumeChatMessage(func(m *messages.ChatMessage) {
    20  		b, _ := json.Marshal(m)
    21  		logger.D("message: %s", string(b))
    22  	})
    23  	consumer.ConsumeChannelMessage(func(m *messages.ChatMessage) {
    24  		b, _ := json.Marshal(m)
    25  		logger.D("message: %s", string(b))
    26  	})
    27  	consumer.ConsumeOfflineMessage(func(m *messages.ChatMessage) {
    28  		b, _ := json.Marshal(m)
    29  		logger.D("message: %s", string(b))
    30  	})
    31  
    32  	time.Sleep(time.Second * 5)
    33  }