func TestShortMessageWindowSize(t *testing.T) { s := smpptest.NewUnstartedServer() s.Handler = func(c smpptest.Conn, p pdu.Body) { time.Sleep(200 * time.Millisecond) r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq r.Fields().Set(pdufield.MessageID, "foobar") c.Write(r) } s.Start() defer s.Close() tx := &Transmitter{ Addr: s.Addr(), User: smpptest.DefaultUser, Passwd: smpptest.DefaultPasswd, WindowSize: 2, RespTimeout: time.Second, } defer tx.Close() conn := <-tx.Bind() switch conn.Status() { case Connected: default: t.Fatal(conn.Error()) } msgc := make(chan *ShortMessage, 3) defer close(msgc) errc := make(chan error, 3) for i := 0; i < 3; i++ { go func(msgc chan *ShortMessage, errc chan error) { m := <-msgc if m == nil { return } _, err := tx.Submit(m) errc <- err }(msgc, errc) msgc <- &ShortMessage{ Src: "root", Dst: "foobar", Text: pdutext.Raw("Lorem ipsum"), Validity: 10 * time.Minute, Register: NoDeliveryReceipt, } } nerr := 0 for i := 0; i < 3; i++ { if <-errc == ErrMaxWindowSize { nerr++ } } if nerr != 1 { t.Fatalf("unexpected # of errors. want 1, have %d", nerr) } }
func TestLongMessage(t *testing.T) { s := smpptest.NewUnstartedServer() s.Handler = func(c smpptest.Conn, p pdu.Body) { switch p.Header().ID { case pdu.SubmitSMID: r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq r.Fields().Set(pdufield.MessageID, "foobar") c.Write(r) default: smpptest.EchoHandler(c, p) } } s.Start() defer s.Close() tx := &Transmitter{ Addr: s.Addr(), User: smpptest.DefaultUser, Passwd: smpptest.DefaultPasswd, } defer tx.Close() conn := <-tx.Bind() switch conn.Status() { case Connected: default: t.Fatal(conn.Error()) } sm, err := tx.SubmitLongMsg(&ShortMessage{ Src: "root", Dst: "foobar", Text: pdutext.Raw("Lorem ipsum dolor sit amet, consectetur adipiscing elit. Nam consequat nisl enim, vel finibus neque aliquet sit amet. Interdum et malesuada fames ac ante ipsum primis in faucibus."), Validity: 10 * time.Minute, Register: NoDeliveryReceipt, }) if err != nil { t.Fatal(err) } msgid := sm.RespID() if msgid == "" { t.Fatalf("pdu does not contain msgid: %#v", sm.Resp()) } if msgid != "foobar" { t.Fatalf("unexpected msgid: want foobar, have %q", msgid) } }
func TestShortMessage(t *testing.T) { s := smpptest.NewUnstartedServer() s.Handler = func(c smpptest.Conn, p pdu.Body) { switch p.Header().ID { case pdu.SubmitSMID: r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq r.Fields().Set(pdufield.MessageID, "foobar") c.Write(r) default: smpptest.EchoHandler(c, p) } } s.Start() defer s.Close() tx := &Transmitter{ Addr: s.Addr(), User: smpptest.DefaultUser, Passwd: smpptest.DefaultPasswd, } defer tx.Close() conn := <-tx.Bind() switch conn.Status() { case Connected: default: t.Fatal(conn.Error()) } sm, err := tx.Submit(&ShortMessage{ Src: "root", Dst: "foobar", Text: pdutext.Raw("Lorem ipsum"), Register: NoDeliveryReceipt, }) if err != nil { t.Fatal(err) } msgid := sm.RespID() if msgid == "" { t.Fatalf("pdu does not contain msgid: %#v", sm.Resp()) } if msgid != "foobar" { t.Fatalf("unexpected msgid: want foobar, have %q", msgid) } }
func pduHandler(c smpptest.Conn, p pdu.Body) { src := p.Fields()[pdufield.SourceAddr] fail := src != nil && src.String() == "root" switch p.Header().ID { case pdu.SubmitSMID: r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq if fail { r.Header().Status = 0x00000045 // submitsm failed } else { r.Fields().Set(pdufield.MessageID, "foobar") } c.Write(r) rd := p.Fields()[pdufield.RegisteredDelivery] if rd == nil || rd.Bytes()[0] == 0 { return } r = pdu.NewDeliverSM() rf := r.Fields() pf := p.Fields() rf.Set(pdufield.SourceAddr, pf[pdufield.SourceAddr]) rf.Set(pdufield.DestinationAddr, pf[pdufield.DestinationAddr]) rf.Set(pdufield.ShortMessage, "delivery receipt here") c.Write(r) case pdu.QuerySMID: r := pdu.NewQuerySMResp() r.Header().Seq = p.Header().Seq if fail { r.Header().Status = 0x00000067 // querysm failed } else { pf := p.Fields() rf := r.Fields() rf.Set(pdufield.MessageID, pf[pdufield.MessageID]) rf.Set(pdufield.MessageState, 2) // DELIVERED } c.Write(r) default: smpptest.EchoHandler(c, p) } }
func TestNotConnected(t *testing.T) { s := smpptest.NewUnstartedServer() s.Handler = func(c smpptest.Conn, p pdu.Body) { switch p.Header().ID { case pdu.SubmitSMID: r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq r.Fields().Set(pdufield.MessageID, "foobar") c.Write(r) default: smpptest.EchoHandler(c, p) } } s.Start() defer s.Close() tx := &Transmitter{ Addr: s.Addr(), User: smpptest.DefaultUser, Passwd: smpptest.DefaultPasswd, } // Open connection and then close it conn := <-tx.Bind() switch conn.Status() { case Connected: default: t.Fatal(conn.Error()) } tx.Close() _, err := tx.Submit(&ShortMessage{ Src: "root", Dst: "foobar", Text: pdutext.Raw("Lorem ipsum"), Validity: 10 * time.Minute, Register: NoDeliveryReceipt, }) if err != ErrNotConnected { t.Fatalf("Error should be not connect, got %s", err.Error()) } }
func TestTransceiver(t *testing.T) { s := smpptest.NewUnstartedServer() s.Handler = func(c smpptest.Conn, p pdu.Body) { switch p.Header().ID { case pdu.SubmitSMID: r := pdu.NewSubmitSMResp() r.Header().Seq = p.Header().Seq r.Fields().Set(pdufield.MessageID, "foobar") c.Write(r) pf := p.Fields() rd := pf[pdufield.RegisteredDelivery] if rd.Bytes()[0] == 0 { return } r = pdu.NewDeliverSM() f := r.Fields() f.Set(pdufield.SourceAddr, pf[pdufield.SourceAddr]) f.Set(pdufield.DestinationAddr, pf[pdufield.DestinationAddr]) f.Set(pdufield.ShortMessage, pf[pdufield.ShortMessage]) c.Write(r) default: smpptest.EchoHandler(c, p) } } s.Start() defer s.Close() ack := make(chan error) receiver := func(p pdu.Body) { defer close(ack) if p.Header().ID != pdu.DeliverSMID { ack <- fmt.Errorf("unexpected PDU: %s", p.Header().ID) } } tc := &Transceiver{ Addr: s.Addr(), User: smpptest.DefaultUser, Passwd: smpptest.DefaultPasswd, Handler: receiver, } defer tc.Close() conn := <-tc.Bind() switch conn.Status() { case Connected: default: t.Fatal(conn.Error()) } sm, err := tc.Submit(&ShortMessage{ Src: "root", Dst: "foobar", Text: pdutext.Raw("Lorem ipsum"), Register: FinalDeliveryReceipt, }) if err != nil { t.Fatal(err) } msgid := sm.RespID() if msgid == "" { t.Fatalf("pdu does not contain msgid: %#v", sm.Resp()) } if msgid != "foobar" { t.Fatalf("unexpected msgid: want foobar, have %q", msgid) } select { case err := <-ack: if err != nil { t.Fatal(err) } case <-time.After(time.Second): t.Fatal("timeout waiting for ack") } }