Function TestControlSessionDialerDialV2

client/control_session_test.go:126–233

(t *testing.T)

}

func TestControlSessionDialerDialV2(t *testing.T) {
	clientRaw, serverRaw := net.Pipe()
	defer serverRaw.Close()

	connector := &testConnector{conn: &trackingConn{Conn: clientRaw}}
	serverErrCh := make(chan error, 1)
	go func() {
		magic := make([]byte, len(wire.MagicV2))
		if _, err := io.ReadFull(serverRaw, magic); err != nil {
			serverErrCh <- err
			return
		}
		if string(magic) != wire.MagicV2 {
			serverErrCh <- fmt.Errorf("unexpected magic: %q", string(magic))
			return
		}

		wireConn := wire.NewConn(serverRaw)
		clientHelloFrame, err := wireConn.ReadFrame()
		if err != nil {
			serverErrCh <- err
			return
		}
		if clientHelloFrame.Type != wire.FrameTypeClientHello {
			serverErrCh <- fmt.Errorf("unexpected frame type %d, want %d", clientHelloFrame.Type, wire.FrameTypeClientHello)
			return
		}
		var hello wire.ClientHello
		if err := wireConn.UnmarshalFrame(clientHelloFrame, &hello); err != nil {
			serverErrCh <- err
			return
		}
		if err := wire.ValidateClientHello(hello); err != nil {
			serverErrCh <- err
			return
		}

		rw := msg.NewV2ReadWriterWithConn(wireConn)
		var loginMsg msg.Login
		if err := rw.ReadMsgInto(&loginMsg); err != nil {
			serverErrCh <- err
			return
		}
		if loginMsg.User != "test-user" {
			serverErrCh <- fmt.Errorf("unexpected user: %s", loginMsg.User)
			return
		}
		serverHello, err := wire.NewServerHello(hello)
		if err != nil {
			serverErrCh <- err
			return
		}
		serverHelloFrame, err := wire.NewJSONFrame(wire.FrameTypeServerHello, serverHello)
		if err != nil {
			serverErrCh <- err
			return
		}
		cryptoContext := wire.NewCryptoContext(