File: shared_test.go

package info (click to toggle)
golang-github-streadway-amqp 0.0~git20200716.e6b33f4-3
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, sid, trixie
  • size: 560 kB
  • sloc: xml: 495; sh: 180; makefile: 2
file content (71 lines) | stat: -rw-r--r-- 1,575 bytes parent folder | download | duplicates (2)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
// Copyright (c) 2012, Sean Treadway, SoundCloud Ltd.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
// Source code and contact info at http://github.com/streadway/amqp

package amqp

import (
	"encoding/hex"
	"io"
	"testing"
)

type pipe struct {
	r *io.PipeReader
	w *io.PipeWriter
}

func (p pipe) Read(b []byte) (int, error) {
	return p.r.Read(b)
}

func (p pipe) Write(b []byte) (int, error) {
	return p.w.Write(b)
}

func (p pipe) Close() error {
	p.r.Close()
	p.w.Close()
	return nil
}

type logIO struct {
	t      *testing.T
	prefix string
	proxy  io.ReadWriteCloser
}

func (log *logIO) Read(p []byte) (n int, err error) {
	log.t.Logf("%s reading %d\n", log.prefix, len(p))
	n, err = log.proxy.Read(p)
	if err != nil {
		log.t.Logf("%s read %x: %v\n", log.prefix, p[0:n], err)
	} else {
		log.t.Logf("%s read:\n%s\n", log.prefix, hex.Dump(p[0:n]))
		//fmt.Printf("%s read:\n%s\n", log.prefix, hex.Dump(p[0:n]))
	}
	return
}

func (log *logIO) Write(p []byte) (n int, err error) {
	log.t.Logf("%s writing %d\n", log.prefix, len(p))
	n, err = log.proxy.Write(p)
	if err != nil {
		log.t.Logf("%s write %d, %x: %v\n", log.prefix, len(p), p[0:n], err)
	} else {
		log.t.Logf("%s write %d:\n%s", log.prefix, len(p), hex.Dump(p[0:n]))
		//fmt.Printf("%s write %d:\n%s", log.prefix, len(p), hex.Dump(p[0:n]))
	}
	return
}

func (log *logIO) Close() (err error) {
	err = log.proxy.Close()
	if err != nil {
		log.t.Logf("%s close : %v\n", log.prefix, err)
	} else {
		log.t.Logf("%s close\n", log.prefix)
	}
	return
}