feat: add AdvisoryUpdateEvent with tests

Author: katarinazaprazna

base/mqueue/advisory_update_event.go

@@ -0,0 +1,40 @@
+package mqueue
+
+import (
+	"app/base/types"
+	"context"
+
+	"github.com/bytedance/sonic"
+	"github.com/google/uuid"
+	"github.com/pkg/errors"
+)
+
+type AdvisoryUpdateEvent struct {
+	RhAccountID int                    `json:"rh_account_id"`
+	WorkspaceID uuid.UUID              `json:"workspace_id"`
+	InventoryID uuid.UUID              `json:"inventory_id"`
+	AdvisoryIDs []int64                `json:"advisory_ids"`
+	ProducedAt  types.Rfc3339Timestamp `json:"produced_at"`
+}
+
+type AdvisoryUpdateEvents []AdvisoryUpdateEvent
+
+func (event *AdvisoryUpdateEvent) createKafkaMessage() (KafkaMessage, error) {
+	data, err := sonic.Marshal(event)
+	if err != nil {
+		return KafkaMessage{}, errors.Wrap(err, "Serializing advisory update event")
+	}
+	return KafkaMessage{Value: data}, nil
+}
+
+func (events AdvisoryUpdateEvents) WriteEvents(ctx context.Context, w Writer) error {
+	msgs := make([]KafkaMessage, 0, len(events))
+	for i := range events {
+		msg, err := events[i].createKafkaMessage()
+		if err != nil {
+			return err
+		}
+		msgs = append(msgs, msg)
+	}
+	return w.WriteMessages(ctx, msgs...)
+}

base/mqueue/advisory_update_event_test.go

@@ -0,0 +1,84 @@
+package mqueue
+
+import (
+	"app/base/types"
+	"context"
+	"testing"
+	"time"
+
+	"github.com/bytedance/sonic"
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/assert"
+)
+
+var (
+	testWorkspaceID = uuid.New()
+	testNow         = types.Rfc3339Timestamp(time.Now())
+)
+
+func TestAdvisoryUpdateEventMarshal(t *testing.T) {
+	event := AdvisoryUpdateEvent{
+		RhAccountID: 1,
+		WorkspaceID: testWorkspaceID,
+		InventoryID: uuid.New(),
+		AdvisoryIDs: []int64{101, 202, 303},
+		ProducedAt:  testNow,
+	}
+
+	data, err := sonic.Marshal(&event)
+	assert.NoError(t, err)
+
+	var parsed AdvisoryUpdateEvent
+	err = sonic.Unmarshal(data, &parsed)
+	assert.NoError(t, err)
+	assert.Equal(t, event.RhAccountID, parsed.RhAccountID)
+	assert.Equal(t, event.WorkspaceID, parsed.WorkspaceID)
+	assert.Equal(t, event.InventoryID, parsed.InventoryID)
+	assert.Equal(t, event.AdvisoryIDs, parsed.AdvisoryIDs)
+	assert.NotNil(t, parsed.ProducedAt)
+}
+
+func TestAdvisoryUpdateEventsWriteEvents(t *testing.T) {
+	var writer Writer = &MockKafkaWriter{}
+
+	events := AdvisoryUpdateEvents{
+		{
+			RhAccountID: 1,
+			WorkspaceID: testWorkspaceID,
+			InventoryID: uuid.New(),
+			AdvisoryIDs: []int64{100, 200},
+			ProducedAt:  testNow,
+		},
+		{
+			RhAccountID: 2,
+			WorkspaceID: testWorkspaceID,
+			InventoryID: uuid.New(),
+			AdvisoryIDs: []int64{300},
+			ProducedAt:  testNow,
+		},
+	}
+
+	err := SendMessages(context.Background(), writer, &events)
+	assert.NoError(t, err)
+
+	mockWriter := writer.(*MockKafkaWriter)
+	assert.Equal(t, 2, len(mockWriter.Messages))
+
+	var firstEvent AdvisoryUpdateEvent
+	err = sonic.Unmarshal(mockWriter.Messages[0].Value, &firstEvent)
+	assert.NoError(t, err)
+	assert.Equal(t, events[0].RhAccountID, firstEvent.RhAccountID)
+	assert.Equal(t, events[0].WorkspaceID, firstEvent.WorkspaceID)
+	assert.Equal(t, events[0].InventoryID, firstEvent.InventoryID)
+	assert.Equal(t, events[0].AdvisoryIDs, firstEvent.AdvisoryIDs)
+	assert.NotNil(t, firstEvent.ProducedAt)
+
+	var secondEvent AdvisoryUpdateEvent
+	err = sonic.Unmarshal(mockWriter.Messages[1].Value, &secondEvent)
+	assert.NoError(t, err)
+	assert.Equal(t, events[1].RhAccountID, secondEvent.RhAccountID)
+	assert.Equal(t, events[1].WorkspaceID, secondEvent.WorkspaceID)
+	assert.Equal(t, events[1].InventoryID, secondEvent.InventoryID)
+	assert.Equal(t, events[1].AdvisoryIDs, secondEvent.AdvisoryIDs)
+	assert.NotNil(t, secondEvent.ProducedAt)
+}