Hex Artifact Content

Artifact 91aaac47132120a9ddfd3c0663eb681fad1c2044:


0000: 70 61 63 6b 61 67 65 20 69 6f 75 74 69 6c 0a 0a  package ioutil..
0010: 69 6d 70 6f 72 74 20 28 0a 09 64 69 66 66 20 22  import (..diff "
0020: 30 64 65 76 2e 6f 72 67 2f 64 69 66 66 22 0a 09  0dev.org/diff"..
0030: 22 62 79 74 65 73 22 0a 09 22 65 72 72 6f 72 73  "bytes".."errors
0040: 22 0a 09 22 69 6f 22 0a 09 22 74 65 73 74 69 6e  ".."io".."testin
0050: 67 22 0a 29 0a 0a 66 75 6e 63 20 54 65 73 74 57  g".)..func TestW
0060: 72 69 74 65 72 46 75 6e 63 28 74 20 2a 74 65 73  riterFunc(t *tes
0070: 74 69 6e 67 2e 54 29 20 7b 0a 09 76 61 72 20 28  ting.T) {..var (
0080: 0a 09 09 69 6e 70 75 74 20 20 5b 5d 62 79 74 65  ...input  []byte
0090: 20 3d 20 5b 5d 62 79 74 65 7b 30 2c 20 31 2c 20   = []byte{0, 1, 
00a0: 32 2c 20 33 2c 20 34 2c 20 35 2c 20 36 2c 20 37  2, 3, 4, 5, 6, 7
00b0: 7d 0a 09 09 6f 75 74 70 75 74 20 5b 5d 62 79 74  }...output []byt
00c0: 65 0a 0a 09 09 72 65 61 64 65 72 20 2a 62 79 74  e....reader *byt
00d0: 65 73 2e 52 65 61 64 65 72 20 3d 20 62 79 74 65  es.Reader = byte
00e0: 73 2e 4e 65 77 52 65 61 64 65 72 28 69 6e 70 75  s.NewReader(inpu
00f0: 74 29 0a 09 09 62 75 66 66 65 72 20 62 79 74 65  t)...buffer byte
0100: 73 2e 42 75 66 66 65 72 0a 09 29 0a 0a 09 72 65  s.Buffer..)...re
0110: 61 64 65 72 2e 57 72 69 74 65 54 6f 28 57 72 69  ader.WriteTo(Wri
0120: 74 65 72 46 75 6e 63 28 62 75 66 66 65 72 2e 57  terFunc(buffer.W
0130: 72 69 74 65 29 29 0a 09 6f 75 74 70 75 74 20 3d  rite))..output =
0140: 20 62 75 66 66 65 72 2e 42 79 74 65 73 28 29 0a   buffer.Bytes().
0150: 0a 09 2f 2f 20 44 69 66 66 20 74 68 65 20 72 65  ..// Diff the re
0160: 73 75 6c 74 20 61 67 61 69 6e 73 74 20 74 68 65  sult against the
0170: 20 69 6e 69 74 69 61 6c 20 69 6e 70 75 74 0a 09   initial input..
0180: 64 65 6c 74 61 20 3a 3d 20 64 69 66 66 2e 44 69  delta := diff.Di
0190: 66 66 28 64 69 66 66 2e 44 7b 6c 65 6e 28 69 6e  ff(diff.D{len(in
01a0: 70 75 74 29 2c 20 6c 65 6e 28 6f 75 74 70 75 74  put), len(output
01b0: 29 2c 0a 09 09 66 75 6e 63 28 69 2c 20 6a 20 69  ),...func(i, j i
01c0: 6e 74 29 20 62 6f 6f 6c 20 7b 20 72 65 74 75 72  nt) bool { retur
01d0: 6e 20 69 6e 70 75 74 5b 69 5d 20 3d 3d 20 6f 75  n input[i] == ou
01e0: 74 70 75 74 5b 6a 5d 20 7d 7d 29 0a 09 69 66 20  tput[j] }})..if 
01f0: 6c 65 6e 28 64 65 6c 74 61 2e 41 64 64 65 64 29  len(delta.Added)
0200: 20 3e 20 30 20 7c 7c 20 6c 65 6e 28 64 65 6c 74   > 0 || len(delt
0210: 61 2e 52 65 6d 6f 76 65 64 29 20 3e 20 30 20 7b  a.Removed) > 0 {
0220: 0a 09 09 74 2e 45 72 72 6f 72 28 22 44 69 66 66  ...t.Error("Diff
0230: 65 72 65 6e 63 65 73 20 64 65 74 65 63 74 65 64  erences detected
0240: 20 22 2c 20 64 65 6c 74 61 29 0a 09 7d 0a 7d 0a   ", delta)..}.}.
0250: 0a 66 75 6e 63 20 54 65 73 74 52 65 61 64 65 72  .func TestReader
0260: 46 75 6e 63 28 74 20 2a 74 65 73 74 69 6e 67 2e  Func(t *testing.
0270: 54 29 20 7b 0a 09 76 61 72 20 28 0a 09 09 69 6e  T) {..var (...in
0280: 70 75 74 20 20 5b 5d 62 79 74 65 20 3d 20 5b 5d  put  []byte = []
0290: 62 79 74 65 7b 30 2c 20 31 2c 20 32 2c 20 33 2c  byte{0, 1, 2, 3,
02a0: 20 34 2c 20 35 2c 20 36 2c 20 37 7d 0a 09 09 6f   4, 5, 6, 7}...o
02b0: 75 74 70 75 74 20 5b 5d 62 79 74 65 0a 0a 09 09  utput []byte....
02c0: 72 65 61 64 65 72 20 2a 62 79 74 65 73 2e 52 65  reader *bytes.Re
02d0: 61 64 65 72 20 3d 20 62 79 74 65 73 2e 4e 65 77  ader = bytes.New
02e0: 52 65 61 64 65 72 28 69 6e 70 75 74 29 0a 09 09  Reader(input)...
02f0: 62 75 66 66 65 72 20 62 79 74 65 73 2e 42 75 66  buffer bytes.Buf
0300: 66 65 72 0a 09 29 0a 0a 09 62 75 66 66 65 72 2e  fer..)...buffer.
0310: 52 65 61 64 46 72 6f 6d 28 52 65 61 64 65 72 46  ReadFrom(ReaderF
0320: 75 6e 63 28 72 65 61 64 65 72 2e 52 65 61 64 29  unc(reader.Read)
0330: 29 0a 09 6f 75 74 70 75 74 20 3d 20 62 75 66 66  )..output = buff
0340: 65 72 2e 42 79 74 65 73 28 29 0a 0a 09 2f 2f 20  er.Bytes()...// 
0350: 44 69 66 66 20 74 68 65 20 72 65 73 75 6c 74 20  Diff the result 
0360: 61 67 61 69 6e 73 74 20 74 68 65 20 69 6e 69 74  against the init
0370: 69 61 6c 20 69 6e 70 75 74 0a 09 64 65 6c 74 61  ial input..delta
0380: 20 3a 3d 20 64 69 66 66 2e 44 69 66 66 28 64 69   := diff.Diff(di
0390: 66 66 2e 44 7b 6c 65 6e 28 69 6e 70 75 74 29 2c  ff.D{len(input),
03a0: 20 6c 65 6e 28 6f 75 74 70 75 74 29 2c 0a 09 09   len(output),...
03b0: 66 75 6e 63 28 69 2c 20 6a 20 69 6e 74 29 20 62  func(i, j int) b
03c0: 6f 6f 6c 20 7b 20 72 65 74 75 72 6e 20 69 6e 70  ool { return inp
03d0: 75 74 5b 69 5d 20 3d 3d 20 6f 75 74 70 75 74 5b  ut[i] == output[
03e0: 6a 5d 20 7d 7d 29 0a 09 69 66 20 6c 65 6e 28 64  j] }})..if len(d
03f0: 65 6c 74 61 2e 41 64 64 65 64 29 20 3e 20 30 20  elta.Added) > 0 
0400: 7c 7c 20 6c 65 6e 28 64 65 6c 74 61 2e 52 65 6d  || len(delta.Rem
0410: 6f 76 65 64 29 20 3e 20 30 20 7b 0a 09 09 74 2e  oved) > 0 {...t.
0420: 45 72 72 6f 72 28 22 44 69 66 66 65 72 65 6e 63  Error("Differenc
0430: 65 73 20 64 65 74 65 63 74 65 64 20 22 2c 20 64  es detected ", d
0440: 65 6c 74 61 29 0a 09 7d 0a 7d 0a 0a 66 75 6e 63  elta)..}.}..func
0450: 20 54 65 73 74 53 69 7a 65 64 57 72 69 74 65 72   TestSizedWriter
0460: 28 74 20 2a 74 65 73 74 69 6e 67 2e 54 29 20 7b  (t *testing.T) {
0470: 0a 09 76 61 72 20 28 0a 09 09 62 75 66 66 65 72  ..var (...buffer
0480: 20 62 79 74 65 73 2e 42 75 66 66 65 72 0a 09 09   bytes.Buffer...
0490: 77 72 69 74 65 72 20 69 6f 2e 57 72 69 74 65 72  writer io.Writer
04a0: 20 3d 20 53 69 7a 65 64 57 72 69 74 65 72 28 26   = SizedWriter(&
04b0: 62 75 66 66 65 72 2c 20 34 29 0a 09 29 0a 0a 09  buffer, 4)..)...
04c0: 63 6f 75 6e 74 2c 20 65 72 72 20 3a 3d 20 77 72  count, err := wr
04d0: 69 74 65 72 2e 57 72 69 74 65 28 5b 5d 62 79 74  iter.Write([]byt
04e0: 65 28 22 31 32 22 29 29 0a 09 69 66 20 63 6f 75  e("12"))..if cou
04f0: 6e 74 20 21 3d 20 32 20 7b 0a 09 09 74 2e 45 72  nt != 2 {...t.Er
0500: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0510: 77 72 69 74 65 20 63 6f 75 6e 74 20 66 72 6f 6d  write count from
0520: 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20 63   SizedWriter", c
0530: 6f 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72 72  ount)..}..if err
0540: 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 74 2e 45 72   != nil {...t.Er
0550: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0560: 65 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64  error from Sized
0570: 57 72 69 74 65 72 22 2c 20 65 72 72 29 0a 09 7d  Writer", err)..}
0580: 0a 0a 09 63 6f 75 6e 74 2c 20 65 72 72 20 3d 20  ...count, err = 
0590: 77 72 69 74 65 72 2e 57 72 69 74 65 28 5b 5d 62  writer.Write([]b
05a0: 79 74 65 28 22 33 34 35 36 22 29 29 0a 09 69 66  yte("3456"))..if
05b0: 20 63 6f 75 6e 74 20 21 3d 20 32 20 7b 0a 09 09   count != 2 {...
05c0: 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70 65 63  t.Error("Unexpec
05d0: 74 65 64 20 77 72 69 74 65 20 63 6f 75 6e 74 20  ted write count 
05e0: 66 72 6f 6d 20 53 69 7a 65 64 57 72 69 74 65 72  from SizedWriter
05f0: 22 2c 20 63 6f 75 6e 74 29 0a 09 7d 0a 09 69 66  ", count)..}..if
0600: 20 65 72 72 20 21 3d 20 6e 69 6c 20 7b 0a 09 09   err != nil {...
0610: 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70 65 63  t.Error("Unexpec
0620: 74 65 64 20 65 72 72 6f 72 20 66 72 6f 6d 20 53  ted error from S
0630: 69 7a 65 64 57 72 69 74 65 72 22 2c 20 65 72 72  izedWriter", err
0640: 29 0a 09 7d 0a 09 69 66 20 62 75 66 66 65 72 2e  )..}..if buffer.
0650: 53 74 72 69 6e 67 28 29 20 21 3d 20 22 31 32 33  String() != "123
0660: 34 22 20 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22  4" {...t.Error("
0670: 55 6e 65 78 70 65 63 74 65 64 20 76 61 6c 75 65  Unexpected value
0680: 20 69 6e 20 77 72 61 70 70 65 64 20 77 72 69 74   in wrapped writ
0690: 65 72 22 2c 20 62 75 66 66 65 72 2e 53 74 72 69  er", buffer.Stri
06a0: 6e 67 28 29 29 0a 09 7d 0a 0a 09 2f 2f 20 46 6c  ng())..}...// Fl
06b0: 75 73 68 20 74 68 65 20 62 75 66 66 65 72 0a 09  ush the buffer..
06c0: 63 6f 75 6e 74 2c 20 65 72 72 20 3d 20 77 72 69  count, err = wri
06d0: 74 65 72 2e 57 72 69 74 65 28 6e 69 6c 29 0a 09  ter.Write(nil)..
06e0: 69 66 20 63 6f 75 6e 74 20 21 3d 20 30 20 7b 0a  if count != 0 {.
06f0: 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70  ..t.Error("Unexp
0700: 65 63 74 65 64 20 77 72 69 74 65 20 63 6f 75 6e  ected write coun
0710: 74 20 66 72 6f 6d 20 53 69 7a 65 64 57 72 69 74  t from SizedWrit
0720: 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09 7d 0a 09  er", count)..}..
0730: 69 66 20 65 72 72 20 21 3d 20 6e 69 6c 20 7b 0a  if err != nil {.
0740: 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70  ..t.Error("Unexp
0750: 65 63 74 65 64 20 65 72 72 6f 72 20 66 72 6f 6d  ected error from
0760: 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20 65   SizedWriter", e
0770: 72 72 29 0a 09 7d 0a 09 69 66 20 62 75 66 66 65  rr)..}..if buffe
0780: 72 2e 53 74 72 69 6e 67 28 29 20 21 3d 20 22 31  r.String() != "1
0790: 32 33 34 35 36 22 20 7b 0a 09 09 74 2e 45 72 72  23456" {...t.Err
07a0: 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20 76  or("Unexpected v
07b0: 61 6c 75 65 20 69 6e 20 77 72 61 70 70 65 64 20  alue in wrapped 
07c0: 77 72 69 74 65 72 22 2c 20 62 75 66 66 65 72 2e  writer", buffer.
07d0: 53 74 72 69 6e 67 28 29 29 0a 09 7d 0a 0a 09 63  String())..}...c
07e0: 6f 75 6e 74 2c 20 65 72 72 20 3d 20 77 72 69 74  ount, err = writ
07f0: 65 72 2e 57 72 69 74 65 28 5b 5d 62 79 74 65 28  er.Write([]byte(
0800: 22 37 38 39 30 22 29 29 0a 09 69 66 20 63 6f 75  "7890"))..if cou
0810: 6e 74 20 21 3d 20 34 20 7b 0a 09 09 74 2e 45 72  nt != 4 {...t.Er
0820: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0830: 77 72 69 74 65 20 63 6f 75 6e 74 20 66 72 6f 6d  write count from
0840: 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20 63   SizedWriter", c
0850: 6f 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72 72  ount)..}..if err
0860: 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 74 2e 45 72   != nil {...t.Er
0870: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0880: 65 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64  error from Sized
0890: 57 72 69 74 65 72 22 2c 20 65 72 72 29 0a 09 7d  Writer", err)..}
08a0: 0a 09 69 66 20 62 75 66 66 65 72 2e 53 74 72 69  ..if buffer.Stri
08b0: 6e 67 28 29 20 21 3d 20 22 31 32 33 34 35 36 37  ng() != "1234567
08c0: 38 39 30 22 20 7b 0a 09 09 74 2e 45 72 72 6f 72  890" {...t.Error
08d0: 28 22 55 6e 65 78 70 65 63 74 65 64 20 76 61 6c  ("Unexpected val
08e0: 75 65 20 69 6e 20 77 72 61 70 70 65 64 20 77 72  ue in wrapped wr
08f0: 69 74 65 72 22 2c 20 62 75 66 66 65 72 2e 53 74  iter", buffer.St
0900: 72 69 6e 67 28 29 29 0a 09 7d 0a 7d 0a 0a 66 75  ring())..}.}..fu
0910: 6e 63 20 54 65 73 74 53 69 7a 65 57 72 69 74 65  nc TestSizeWrite
0920: 72 4c 61 72 67 65 72 28 74 20 2a 74 65 73 74 69  rLarger(t *testi
0930: 6e 67 2e 54 29 20 7b 0a 09 76 61 72 20 28 0a 09  ng.T) {..var (..
0940: 09 69 6e 70 75 74 20 20 5b 5d 62 79 74 65 20 3d  .input  []byte =
0950: 20 5b 5d 62 79 74 65 28 22 30 31 32 33 34 35 36   []byte("0123456
0960: 37 38 39 41 42 22 29 0a 09 09 62 75 66 66 65 72  789AB")...buffer
0970: 20 62 79 74 65 73 2e 42 75 66 66 65 72 0a 09 09   bytes.Buffer...
0980: 77 72 69 74 65 72 20 3d 20 53 69 7a 65 64 57 72  writer = SizedWr
0990: 69 74 65 72 28 26 62 75 66 66 65 72 2c 20 38 29  iter(&buffer, 8)
09a0: 0a 09 29 0a 0a 09 63 6f 75 6e 74 2c 20 65 72 72  ..)...count, err
09b0: 20 3a 3d 20 77 72 69 74 65 72 2e 57 72 69 74 65   := writer.Write
09c0: 28 69 6e 70 75 74 29 0a 09 69 66 20 63 6f 75 6e  (input)..if coun
09d0: 74 20 21 3d 20 31 32 20 7b 0a 09 09 74 2e 45 72  t != 12 {...t.Er
09e0: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
09f0: 77 72 69 74 65 20 63 6f 75 6e 74 20 66 72 6f 6d  write count from
0a00: 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20 63   SizedWriter", c
0a10: 6f 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72 72  ount)..}..if err
0a20: 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 74 2e 45 72   != nil {...t.Er
0a30: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0a40: 65 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64  error from Sized
0a50: 57 72 69 74 65 72 22 2c 20 65 72 72 29 0a 09 7d  Writer", err)..}
0a60: 0a 09 69 66 20 62 75 66 66 65 72 2e 53 74 72 69  ..if buffer.Stri
0a70: 6e 67 28 29 20 21 3d 20 22 30 31 32 33 34 35 36  ng() != "0123456
0a80: 37 22 20 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22  7" {...t.Error("
0a90: 55 6e 65 78 70 65 63 74 65 64 20 76 61 6c 75 65  Unexpected value
0aa0: 20 69 6e 20 77 72 61 70 70 65 64 20 77 72 69 74   in wrapped writ
0ab0: 65 72 22 2c 20 62 75 66 66 65 72 2e 53 74 72 69  er", buffer.Stri
0ac0: 6e 67 28 29 29 0a 09 7d 0a 0a 09 63 6f 75 6e 74  ng())..}...count
0ad0: 2c 20 65 72 72 20 3d 20 77 72 69 74 65 72 2e 57  , err = writer.W
0ae0: 72 69 74 65 28 6e 69 6c 29 0a 09 69 66 20 63 6f  rite(nil)..if co
0af0: 75 6e 74 20 21 3d 20 30 20 7b 0a 09 09 74 2e 45  unt != 0 {...t.E
0b00: 72 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64  rror("Unexpected
0b10: 20 77 72 69 74 65 20 63 6f 75 6e 74 20 66 72 6f   write count fro
0b20: 6d 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20  m SizedWriter", 
0b30: 63 6f 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72  count)..}..if er
0b40: 72 20 21 3d 20 6e 69 6c 20 7b 0a 09 09 74 2e 45  r != nil {...t.E
0b50: 72 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64  rror("Unexpected
0b60: 20 65 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65   error from Size
0b70: 64 57 72 69 74 65 72 22 2c 20 65 72 72 29 0a 09  dWriter", err)..
0b80: 7d 0a 09 69 66 20 62 75 66 66 65 72 2e 53 74 72  }..if buffer.Str
0b90: 69 6e 67 28 29 20 21 3d 20 22 30 31 32 33 34 35  ing() != "012345
0ba0: 36 37 38 39 41 42 22 20 7b 0a 09 09 74 2e 45 72  6789AB" {...t.Er
0bb0: 72 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20  ror("Unexpected 
0bc0: 76 61 6c 75 65 20 69 6e 20 77 72 61 70 70 65 64  value in wrapped
0bd0: 20 77 72 69 74 65 72 22 2c 20 62 75 66 66 65 72   writer", buffer
0be0: 2e 53 74 72 69 6e 67 28 29 29 0a 09 7d 0a 7d 0a  .String())..}.}.
0bf0: 0a 66 75 6e 63 20 54 65 73 74 53 69 7a 65 64 57  .func TestSizedW
0c00: 72 69 74 65 72 45 72 72 6f 72 31 28 74 20 2a 74  riterError1(t *t
0c10: 65 73 74 69 6e 67 2e 54 29 20 7b 0a 09 76 61 72  esting.T) {..var
0c20: 20 28 0a 09 09 65 72 72 6f 72 57 72 69 74 65 72   (...errorWriter
0c30: 20 69 6f 2e 57 72 69 74 65 72 20 3d 20 57 72 69   io.Writer = Wri
0c40: 74 65 72 46 75 6e 63 28 66 75 6e 63 28 5b 5d 62  terFunc(func([]b
0c50: 79 74 65 29 20 28 69 6e 74 2c 20 65 72 72 6f 72  yte) (int, error
0c60: 29 20 7b 0a 09 09 09 72 65 74 75 72 6e 20 31 2c  ) {....return 1,
0c70: 20 65 72 72 6f 72 73 2e 4e 65 77 28 22 49 6e 76   errors.New("Inv
0c80: 61 6c 69 64 20 77 72 69 74 65 22 29 0a 09 09 7d  alid write")...}
0c90: 29 0a 09 09 77 72 69 74 65 72 20 69 6f 2e 57 72  )...writer io.Wr
0ca0: 69 74 65 72 20 3d 20 53 69 7a 65 64 57 72 69 74  iter = SizedWrit
0cb0: 65 72 28 65 72 72 6f 72 57 72 69 74 65 72 2c 20  er(errorWriter, 
0cc0: 32 29 0a 09 29 0a 0a 09 63 6f 75 6e 74 2c 20 65  2)..)...count, e
0cd0: 72 72 20 3a 3d 20 77 72 69 74 65 72 2e 57 72 69  rr := writer.Wri
0ce0: 74 65 28 5b 5d 62 79 74 65 28 22 31 22 29 29 0a  te([]byte("1")).
0cf0: 09 69 66 20 63 6f 75 6e 74 20 21 3d 20 31 20 7b  .if count != 1 {
0d00: 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78  ...t.Error("Unex
0d10: 70 65 63 74 65 64 20 77 72 69 74 65 20 63 6f 75  pected write cou
0d20: 6e 74 20 66 72 6f 6d 20 53 69 7a 65 64 57 72 69  nt from SizedWri
0d30: 74 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09 7d 0a  ter", count)..}.
0d40: 09 69 66 20 65 72 72 20 21 3d 20 6e 69 6c 20 7b  .if err != nil {
0d50: 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78  ...t.Error("Unex
0d60: 70 65 63 74 65 64 20 65 72 72 6f 72 20 66 72 6f  pected error fro
0d70: 6d 20 53 69 7a 65 64 57 72 69 74 65 72 22 2c 20  m SizedWriter", 
0d80: 65 72 72 29 0a 09 7d 0a 0a 09 63 6f 75 6e 74 2c  err)..}...count,
0d90: 20 65 72 72 20 3d 20 77 72 69 74 65 72 2e 57 72   err = writer.Wr
0da0: 69 74 65 28 5b 5d 62 79 74 65 28 22 32 22 29 29  ite([]byte("2"))
0db0: 0a 09 69 66 20 63 6f 75 6e 74 20 21 3d 20 31 20  ..if count != 1 
0dc0: 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65  {...t.Error("Une
0dd0: 78 70 65 63 74 65 64 20 77 72 69 74 65 20 63 6f  xpected write co
0de0: 75 6e 74 20 66 72 6f 6d 20 53 69 7a 65 64 57 72  unt from SizedWr
0df0: 69 74 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09 7d  iter", count)..}
0e00: 0a 09 69 66 20 65 72 72 20 3d 3d 20 6e 69 6c 20  ..if err == nil 
0e10: 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65  {...t.Error("Une
0e20: 78 70 65 63 74 65 64 20 6c 61 63 6b 20 6f 66 20  xpected lack of 
0e30: 65 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64  error from Sized
0e40: 57 72 69 74 65 72 22 29 0a 09 7d 0a 7d 0a 0a 66  Writer")..}.}..f
0e50: 75 6e 63 20 54 65 73 74 53 69 7a 65 64 57 72 69  unc TestSizedWri
0e60: 74 65 72 45 72 72 6f 72 32 28 74 20 2a 74 65 73  terError2(t *tes
0e70: 74 69 6e 67 2e 54 29 20 7b 0a 09 76 61 72 20 28  ting.T) {..var (
0e80: 0a 09 09 65 72 72 6f 72 57 72 69 74 65 72 20 69  ...errorWriter i
0e90: 6f 2e 57 72 69 74 65 72 20 3d 20 57 72 69 74 65  o.Writer = Write
0ea0: 72 46 75 6e 63 28 66 75 6e 63 28 5b 5d 62 79 74  rFunc(func([]byt
0eb0: 65 29 20 28 69 6e 74 2c 20 65 72 72 6f 72 29 20  e) (int, error) 
0ec0: 7b 0a 09 09 09 72 65 74 75 72 6e 20 31 2c 20 65  {....return 1, e
0ed0: 72 72 6f 72 73 2e 4e 65 77 28 22 49 6e 76 61 6c  rrors.New("Inval
0ee0: 69 64 20 77 72 69 74 65 22 29 0a 09 09 7d 29 0a  id write")...}).
0ef0: 09 09 77 72 69 74 65 72 20 69 6f 2e 57 72 69 74  ..writer io.Writ
0f00: 65 72 20 3d 20 53 69 7a 65 64 57 72 69 74 65 72  er = SizedWriter
0f10: 28 65 72 72 6f 72 57 72 69 74 65 72 2c 20 31 29  (errorWriter, 1)
0f20: 0a 09 29 0a 0a 09 63 6f 75 6e 74 2c 20 65 72 72  ..)...count, err
0f30: 20 3a 3d 20 77 72 69 74 65 72 2e 57 72 69 74 65   := writer.Write
0f40: 28 5b 5d 62 79 74 65 28 22 31 32 22 29 29 0a 09  ([]byte("12"))..
0f50: 69 66 20 63 6f 75 6e 74 20 21 3d 20 31 20 7b 0a  if count != 1 {.
0f60: 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70  ..t.Error("Unexp
0f70: 65 63 74 65 64 20 77 72 69 74 65 20 63 6f 75 6e  ected write coun
0f80: 74 20 66 72 6f 6d 20 53 69 7a 65 64 57 72 69 74  t from SizedWrit
0f90: 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09 7d 0a 09  er", count)..}..
0fa0: 69 66 20 65 72 72 20 3d 3d 20 6e 69 6c 20 7b 0a  if err == nil {.
0fb0: 09 09 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70  ..t.Error("Unexp
0fc0: 65 63 74 65 64 20 6c 61 63 6b 20 6f 66 20 65 72  ected lack of er
0fd0: 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64 57 72  ror from SizedWr
0fe0: 69 74 65 72 22 29 0a 09 7d 0a 7d 0a 0a 66 75 6e  iter")..}.}..fun
0ff0: 63 20 54 65 73 74 53 69 7a 65 64 52 65 61 64 65  c TestSizedReade
1000: 72 28 74 20 2a 74 65 73 74 69 6e 67 2e 54 29 20  r(t *testing.T) 
1010: 7b 0a 09 76 61 72 20 28 0a 09 09 69 6e 70 75 74  {..var (...input
1020: 20 20 5b 5d 62 79 74 65 20 3d 20 5b 5d 62 79 74    []byte = []byt
1030: 65 7b 30 2c 20 31 2c 20 32 2c 20 33 2c 20 34 2c  e{0, 1, 2, 3, 4,
1040: 20 35 2c 20 36 2c 20 37 7d 0a 09 09 6f 75 74 70   5, 6, 7}...outp
1050: 75 74 20 5b 5d 62 79 74 65 20 3d 20 6d 61 6b 65  ut []byte = make
1060: 28 5b 5d 62 79 74 65 2c 20 31 36 29 0a 0a 09 09  ([]byte, 16)....
1070: 72 65 61 64 65 72 20 2a 62 79 74 65 73 2e 52 65  reader *bytes.Re
1080: 61 64 65 72 20 3d 20 62 79 74 65 73 2e 4e 65 77  ader = bytes.New
1090: 52 65 61 64 65 72 28 69 6e 70 75 74 29 0a 09 09  Reader(input)...
10a0: 6d 69 6e 20 20 20 20 69 6f 2e 52 65 61 64 65 72  min    io.Reader
10b0: 20 20 20 20 20 3d 20 53 69 7a 65 64 52 65 61 64       = SizedRead
10c0: 65 72 28 72 65 61 64 65 72 2c 20 34 29 0a 09 29  er(reader, 4)..)
10d0: 0a 0a 09 2f 2f 20 45 78 70 65 63 74 69 6e 67 20  ...// Expecting 
10e0: 61 20 72 65 61 64 20 63 6f 75 6e 74 20 6f 66 20  a read count of 
10f0: 32 0a 09 63 6f 75 6e 74 2c 20 65 72 72 20 3a 3d  2..count, err :=
1100: 20 6d 69 6e 2e 52 65 61 64 28 6f 75 74 70 75 74   min.Read(output
1110: 5b 3a 32 5d 29 0a 09 69 66 20 63 6f 75 6e 74 20  [:2])..if count 
1120: 21 3d 20 32 20 7b 0a 09 09 74 2e 45 72 72 6f 72  != 2 {...t.Error
1130: 28 22 49 6e 76 61 6c 69 64 20 72 65 61 64 20 63  ("Invalid read c
1140: 6f 75 6e 74 20 66 72 6f 6d 20 53 69 7a 65 64 52  ount from SizedR
1150: 65 61 64 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09  eader", count)..
1160: 7d 0a 09 69 66 20 65 72 72 20 21 3d 20 6e 69 6c  }..if err != nil
1170: 20 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e   {...t.Error("Un
1180: 65 78 70 65 63 74 65 64 20 65 72 72 6f 72 20 66  expected error f
1190: 72 6f 6d 20 53 69 7a 65 64 52 65 61 64 65 72 22  rom SizedReader"
11a0: 2c 20 65 72 72 29 0a 09 7d 0a 0a 09 2f 2f 20 45  , err)..}...// E
11b0: 78 70 65 63 74 69 6e 67 20 61 20 72 65 61 64 20  xpecting a read 
11c0: 63 6f 75 6e 74 20 6f 66 20 32 20 61 73 20 69 74  count of 2 as it
11d0: 20 73 68 6f 75 6c 64 20 68 61 76 65 20 32 20 62   should have 2 b
11e0: 79 74 65 73 20 69 6e 20 69 74 73 20 62 75 66 66  ytes in its buff
11f0: 65 72 0a 09 63 6f 75 6e 74 2c 20 65 72 72 20 3d  er..count, err =
1200: 20 6d 69 6e 2e 52 65 61 64 28 6f 75 74 70 75 74   min.Read(output
1210: 5b 3a 33 5d 29 0a 09 69 66 20 63 6f 75 6e 74 20  [:3])..if count 
1220: 21 3d 20 32 20 7b 0a 09 09 74 2e 45 72 72 6f 72  != 2 {...t.Error
1230: 28 22 49 6e 76 61 6c 69 64 20 72 65 61 64 20 63  ("Invalid read c
1240: 6f 75 6e 74 20 66 72 6f 6d 20 53 69 7a 65 64 52  ount from SizedR
1250: 65 61 64 65 72 22 2c 20 63 6f 75 6e 74 29 0a 09  eader", count)..
1260: 7d 0a 09 69 66 20 65 72 72 20 21 3d 20 6e 69 6c  }..if err != nil
1270: 20 7b 0a 09 09 74 2e 45 72 72 6f 72 28 22 55 6e   {...t.Error("Un
1280: 65 78 70 65 63 74 65 64 20 65 72 72 6f 72 20 66  expected error f
1290: 72 6f 6d 20 53 69 7a 65 64 52 65 61 64 65 72 22  rom SizedReader"
12a0: 2c 20 65 72 72 29 0a 09 7d 0a 0a 09 2f 2f 20 45  , err)..}...// E
12b0: 78 70 65 63 74 69 6e 67 20 61 20 72 65 61 64 20  xpecting a read 
12c0: 63 6f 75 6e 74 20 6f 66 20 34 20 61 73 20 74 68  count of 4 as th
12d0: 65 20 62 75 66 66 65 72 20 73 68 6f 75 6c 64 20  e buffer should 
12e0: 62 65 20 65 6d 70 74 79 0a 09 63 6f 75 6e 74 2c  be empty..count,
12f0: 20 65 72 72 20 3d 20 6d 69 6e 2e 52 65 61 64 28   err = min.Read(
1300: 6f 75 74 70 75 74 5b 3a 34 5d 29 0a 09 69 66 20  output[:4])..if 
1310: 63 6f 75 6e 74 20 21 3d 20 34 20 7b 0a 09 09 74  count != 4 {...t
1320: 2e 45 72 72 6f 72 28 22 49 6e 76 61 6c 69 64 20  .Error("Invalid 
1330: 72 65 61 64 20 63 6f 75 6e 74 20 66 72 6f 6d 20  read count from 
1340: 53 69 7a 65 64 52 65 61 64 65 72 22 2c 20 63 6f  SizedReader", co
1350: 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72 72 20  unt)..}..if err 
1360: 21 3d 20 6e 69 6c 20 7b 0a 09 09 74 2e 45 72 72  != nil {...t.Err
1370: 6f 72 28 22 55 6e 65 78 70 65 63 74 65 64 20 65  or("Unexpected e
1380: 72 72 6f 72 20 66 72 6f 6d 20 53 69 7a 65 64 52  rror from SizedR
1390: 65 61 64 65 72 22 2c 20 65 72 72 29 0a 09 7d 0a  eader", err)..}.
13a0: 0a 09 2f 2f 20 45 78 70 65 63 74 69 6e 67 20 61  ..// Expecting a
13b0: 20 72 65 61 64 20 63 6f 75 6e 74 20 6f 66 20 30   read count of 0
13c0: 20 77 69 74 68 20 61 6e 20 45 4f 46 20 61 73 20   with an EOF as 
13d0: 74 68 65 20 62 75 66 66 65 72 20 73 68 6f 75 6c  the buffer shoul
13e0: 64 20 62 65 20 65 6d 70 74 79 0a 09 63 6f 75 6e  d be empty..coun
13f0: 74 2c 20 65 72 72 20 3d 20 6d 69 6e 2e 52 65 61  t, err = min.Rea
1400: 64 28 6f 75 74 70 75 74 5b 3a 31 5d 29 0a 09 69  d(output[:1])..i
1410: 66 20 63 6f 75 6e 74 20 21 3d 20 30 20 7b 0a 09  f count != 0 {..
1420: 09 74 2e 45 72 72 6f 72 28 22 49 6e 76 61 6c 69  .t.Error("Invali
1430: 64 20 72 65 61 64 20 63 6f 75 6e 74 20 66 72 6f  d read count fro
1440: 6d 20 53 69 7a 65 64 52 65 61 64 65 72 22 2c 20  m SizedReader", 
1450: 63 6f 75 6e 74 29 0a 09 7d 0a 09 69 66 20 65 72  count)..}..if er
1460: 72 20 21 3d 20 69 6f 2e 45 4f 46 20 7b 0a 09 09  r != io.EOF {...
1470: 74 2e 45 72 72 6f 72 28 22 55 6e 65 78 70 65 63  t.Error("Unexpec
1480: 74 65 64 20 65 72 72 6f 72 20 66 72 6f 6d 20 53  ted error from S
1490: 69 7a 65 64 52 65 61 64 65 72 22 2c 20 65 72 72  izedReader", err
14a0: 29 0a 09 7d 0a 7d 0a                             )..}.}.