gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

add more tests for Dial in the h2quic client

Browse Source
This commit is contained in:
Marten Seemann
2017-05-12 21:56:04 +08:00
parent 9fad63ff50
commit 33a9b5aae4
2 changed files with 42 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
h2quic/client.go
View File

@@ -24,7 +24,8 @@ import (
type Client struct {
mutex sync.RWMutex
config *quic.Config
dialAddr func(hostname string, config *quic.Config) (quic.Session, error)
config *quic.Config
t *QuicRoundTripper
@@ -46,6 +47,7 @@ var _ h2quicClient = &Client{}
func NewClient(t *QuicRoundTripper, tlsConfig *tls.Config, hostname string) *Client {
return &Client{
t: t,
dialAddr: quic.DialAddr,
hostname: authorityAddr("https", hostname),
responses: make(map[protocol.StreamID]chan *http.Response),
encryptionLevel: protocol.EncryptionUnencrypted,
@@ -60,7 +62,7 @@ func NewClient(t *QuicRoundTripper, tlsConfig *tls.Config, hostname string) *Cli
// Dial dials the connection
func (c *Client) Dial() error {
var err error
c.session, err = quic.DialAddr(c.hostname, c.config)
c.session, err = c.dialAddr(c.hostname, c.config)
if err != nil {
return err
}