Merge pull request #826 from gotd/feat-tgtest-allow-private-key

feat(tgtest): export NewPrivateKey

Author: ernado

tgtest/server.go

@@ -2,6 +2,7 @@ package tgtest
 
 import (
 	"context"
+	"crypto/rsa"
 	"io"
 	"net"
 	"time"
@@ -49,6 +50,13 @@ type Server struct {
 	log   *zap.Logger // immutable
 }
 
+// NewPrivateKey creates new private key from RSA private key.
+func NewPrivateKey(k *rsa.PrivateKey) exchange.PrivateKey {
+	return exchange.PrivateKey{
+		RSA: k,
+	}
+}
+
 // NewServer creates new Server.
 func NewServer(key exchange.PrivateKey, handler Handler, opts ServerOptions) *Server {
 	opts.setDefaults()

transport/protocol.go

@@ -54,6 +54,11 @@ func (p Protocol) Codec() Codec {
 	return p.codec()
 }
 
+// CodecNoHeader is Codec without header.
+func (p Protocol) CodecNoHeader() Codec {
+	return codec.NoHeader{Codec: p.codec()}
+}
+
 // Handshake inits given net.Conn as MTProto connection.
 func (p Protocol) Handshake(conn net.Conn) (Conn, error) {
 	connCodec := p.codec()