send_test.go

//
// Copyright © 2011-2019 Guy M. Allard
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

package stompngo

import (
	"testing"
)

/*
	Test Send Basic, one message.
*/
func TestSendBasic(t *testing.T) {
	for _, sp := range Protocols() {
		n, _ = openConn(t)
		ch := login_headers
		ch = headersProtocol(ch, sp)
		conn, e = Connect(n, ch)
		if e != nil {
			t.Fatalf("TestSendBasic CONNECT expected no error, got [%v]\n", e)
		}
		//
		ms := "A message"
		d := tdest("/queue/send.basiconn.01." + sp)
		sh := Headers{HK_DESTINATION, d}
		e = conn.Send(sh, ms)
		if e != nil {
			t.Fatalf("TestSendBasic Expected nil error, got [%v]\n", e)
		}
		//
		e = conn.Send(empty_headers, ms)
		if e == nil {
			t.Fatalf("TestSendBasic Expected error, got [nil]\n")
		}
		if e != EREQDSTSND {
			t.Fatalf("TestSendBasic Expected [%v], got [%v]\n", EREQDSTSND, e)
		}
		checkReceived(t, conn, false)
		e = conn.Disconnect(empty_headers)
		checkDisconnectError(t, e)
		_ = closeConn(t, n)
	}
}

/*
	Test Send Multiple, multiple messages, 5 to be exact.
*/
func TestSendMultiple(t *testing.T) {
	for _, sp := range Protocols() {
		n, _ = openConn(t)
		ch := login_headers
		ch = headersProtocol(ch, sp)
		conn, e = Connect(n, ch)
		if e != nil {
			t.Fatalf("TestSendMultiple CONNECT expected no error, got [%v]\n", e)
		}
		//
		smd := multi_send_data{conn: conn,
			dest:  tdest("/queue/sendmultiple.01." + sp + "."),
			mpref: "sendmultiple.01.message.prefix ",
			count: 5}
		e = sendMultiple(smd)
		if e != nil {
			t.Fatalf("TestSendMultiple Expected nil error, got [%v]\n", e)
		}
		//
		checkReceived(t, conn, false)
		e = conn.Disconnect(empty_headers)
		checkDisconnectError(t, e)
		_ = closeConn(t, n)
	}
}