package gqueue

import (
	"context"
	"fmt"
	"github.com/Shopify/sarama"
	"testing"
	"time"
)

func TestNewQueue(t *testing.T) {
	NewProducer()
	go func() {
		for e := range Errors() {
			println(e.Error())
		}
	}()
	c1, _ := NewConsumer("test", "id1")
	c2, _ := NewConsumer("test", "id2")
	i := 0
	go func() {
		for {
			m := NewMessage()
			m.Topic = "test"
			i++
			m.Value = sarama.StringEncoder(fmt.Sprintf("%d", i))
			Input() <- m
			time.Sleep(time.Second)
		}
	}()
	go func() {
		for {
			ctx, _ := context.WithTimeout(context.Background(), time.Second*3)
			list := c1.GetBatch(ctx, 5)
			for _, one := range list {
				fmt.Printf("[id1]%s", string(one.Value))
			}

		}
	}()

	for {
		ctx, _ := context.WithTimeout(context.Background(), time.Second*3)
		list := c2.GetBatch(ctx, 5)
		for _, one := range list {
			fmt.Printf("[id2]%s", string(one.Value))
		}

	}
}