-
Notifications
You must be signed in to change notification settings - Fork 35
/
main_mock_test.go
55 lines (43 loc) · 1.78 KB
/
main_mock_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
// Copyright 2018 The go-hep Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package xrootd // import "go-hep.org/x/hep/xrootd"
import (
"context"
"net"
"go-hep.org/x/hep/xrootd/internal/mux"
"go-hep.org/x/hep/xrootd/internal/xrdenc"
"go-hep.org/x/hep/xrootd/xrdproto"
"go-hep.org/x/hep/xrootd/xrdproto/signing"
"golang.org/x/xerrors"
)
var testClientAddrs []string
func testClientWithMockServer(serverFunc func(cancel func(), conn net.Conn), clientFunc func(cancel func(), client *Client)) {
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
server, conn := net.Pipe()
defer server.Close()
defer conn.Close()
client := &Client{cancel: cancel, sessions: make(map[string]*cliSession), maxRedirections: 8}
session := &cliSession{cancel: cancel, ctx: ctx, conn: conn, mux: mux.New(), requests: make(map[xrdproto.StreamID]pendingRequest), client: client, signRequirements: signing.Default()}
client.initialSessionID = "test.org:1234"
client.sessions[client.initialSessionID] = session
defer client.Close()
go serverFunc(func() { client.Close() }, server)
go session.consume()
clientFunc(cancel, client)
}
func unmarshalRequest(data []byte, request xrdproto.Request) (xrdproto.RequestHeader, error) {
var header xrdproto.RequestHeader
rBuffer := xrdenc.NewRBuffer(data)
if err := header.UnmarshalXrd(rBuffer); err != nil {
return xrdproto.RequestHeader{}, err
}
if header.RequestID != request.ReqID() {
return xrdproto.RequestHeader{}, xerrors.Errorf("xrootd: unexpected request id was specified:\nwant = %d\ngot = %d\n", request.ReqID(), header.RequestID)
}
if err := request.UnmarshalXrd(rBuffer); err != nil {
return xrdproto.RequestHeader{}, err
}
return header, nil
}