support sender SetSubject() and SetTo() options (#276)

Author: karenychen

v2/sender.go

@@ -266,6 +266,22 @@ func SetMessageTTL(ttl time.Duration) func(msg *azservicebus.Message) error {
 	}
 }
 
+// SetSubject sets the ServiceBus message's Subject property to a user-specified value
+func SetSubject(subject string) func(msg *azservicebus.Message) error {
+	return func(msg *azservicebus.Message) error {
+		msg.Subject = &subject
+		return nil
+	}
+}
+
+// SetTo sets the ServiceBus message's To property to a user-specified value
+func SetTo(to string) func(msg *azservicebus.Message) error {
+	return func(msg *azservicebus.Message) error {
+		msg.To = &to
+		return nil
+	}
+}
+
 func getMessageType(mb MessageBody) string {
 	var msgType string
 	vo := reflect.ValueOf(mb)

v2/sender_test.go

@@ -95,6 +95,30 @@ func TestHandlers_SetMessageTTL(t *testing.T) {
 	}
 }
 
+func TestHandlers_SetSubject(t *testing.T) {
+	subject := "test-subject"
+	blankMsg := &azservicebus.Message{}
+	handler := SetSubject(subject)
+	if err := handler(blankMsg); err != nil {
+		t.Errorf("Unexpected error in set subject test: %s", err)
+	}
+	if *blankMsg.Subject != subject {
+		t.Errorf("for subject expected %s, got %s", subject, *blankMsg.Subject)
+	}
+}
+
+func TestHandlers_SetTo(t *testing.T) {
+	to := "test-to"
+	blankMsg := &azservicebus.Message{}
+	handler := SetTo(to)
+	if err := handler(blankMsg); err != nil {
+		t.Errorf("Unexpected error in set to test: %s", err)
+	}
+	if *blankMsg.To != to {
+		t.Errorf("for to expected %s, got %s", to, *blankMsg.To)
+	}
+}
+
 func TestSender_SenderTracePropagation(t *testing.T) {
 	g := NewWithT(t)
 	azSender := &fakeAzSender{}