fix: added the ability to setup the consumer and producer names

hidromatologia-v2 · Aug 26, 2023 · 060b12d · 060b12d
1 parent 875a991
commit 060b12d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/common/connection/connection.go b/common/connection/connection.go
@@ -39,37 +39,37 @@ const (
 )
 
 func DefaultConsumer(t *testing.T) *memphis.Consumer {
- return NewConsumer(t, testingStation)
+ return NewConsumer(t, testingStation, random.String())
 }
 
 func DefaultProducer(t *testing.T) *memphis.Producer {
- return NewProducer(t, testingStation)
+ return NewProducer(t, testingStation, random.String())
 }
 
-func NewConsumer(t *testing.T, station string) *memphis.Consumer {
+func NewConsumer(t *testing.T, station, name string) *memphis.Consumer {
  conn, cErr := memphis.Connect(
  "127.0.0.1",
  "root",
  memphis.Password("memphis"),
  // memphis.ConnectionToken("memphis"),
  )
  assert.Nil(t, cErr)
- consumerName := fmt.Sprintf("%s-%s", station, uuid.NewV4().String())
- c, err := conn.CreateConsumer(consumerName, random.String())
+ name = fmt.Sprintf("%s-%s", name, uuid.NewV4().String())
+ c, err := conn.CreateConsumer(station, name)
  assert.Nil(t, err)
  return c
 }
 
-func NewProducer(t *testing.T, station string) *memphis.Producer {
+func NewProducer(t *testing.T, station, name string) *memphis.Producer {
  conn, cErr := memphis.Connect(
  "127.0.0.1",
  "root",
  memphis.Password("memphis"),
  // memphis.ConnectionToken("memphis"),
  )
  assert.Nil(t, cErr)
- producerName := fmt.Sprintf("%s-%s", station, uuid.NewV4().String())
- prod, err := conn.CreateProducer(producerName, random.String())
+ name = fmt.Sprintf("%s-%s", name, uuid.NewV4().String())
+ prod, err := conn.CreateProducer(station, name)
  assert.Nil(t, err)
  return prod
 }
diff --git a/common/connection/connection_test.go b/common/connection/connection_test.go
@@ -26,13 +26,13 @@ func TestDefaultProducer(t *testing.T) {
 }
 
 func TestNewConsumer(t *testing.T) {
- c := NewConsumer(t, random.String())
+ c := NewConsumer(t, random.String(), random.String())
  assert.NotNil(t, c)
  defer c.Destroy()
 }
 
 func TestNewProducer(t *testing.T) {
- c := NewProducer(t, random.String())
+ c := NewProducer(t, random.String(), random.String())
  assert.NotNil(t, c)
  defer c.Destroy()
 }