1 files changed, 389 insertions, 0 deletions
diff --git a/libgo/go/compress/flate/deflate_test.go b/libgo/go/compress/flate/deflate_test.go
new file mode 100644
index 000000000..3db955609
--- /dev/null
+++ b/libgo/go/compress/flate/deflate_test.go
@@ -0,0 +1,389 @@
+// Copyright 2009 The Go 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 flate
+
+import (
+	"bytes"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"os"
+	"sync"
+	"testing"
+)
+
+type deflateTest struct {
+	in    []byte
+	level int
+	out   []byte
+}
+
+type deflateInflateTest struct {
+	in []byte
+}
+
+type reverseBitsTest struct {
+	in       uint16
+	bitCount uint8
+	out      uint16
+}
+
+var deflateTests = []*deflateTest{
+	&deflateTest{[]byte{}, 0, []byte{1, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11}, -1, []byte{18, 4, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11}, DefaultCompression, []byte{18, 4, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11}, 4, []byte{18, 4, 4, 0, 0, 255, 255}},
+
+	&deflateTest{[]byte{0x11}, 0, []byte{0, 1, 0, 254, 255, 17, 1, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x12}, 0, []byte{0, 2, 0, 253, 255, 17, 18, 1, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 0,
+		[]byte{0, 8, 0, 247, 255, 17, 17, 17, 17, 17, 17, 17, 17, 1, 0, 0, 255, 255},
+	},
+	&deflateTest{[]byte{}, 1, []byte{1, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11}, 1, []byte{18, 4, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x12}, 1, []byte{18, 20, 2, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 1, []byte{18, 132, 2, 64, 0, 0, 0, 255, 255}},
+	&deflateTest{[]byte{}, 9, []byte{1, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11}, 9, []byte{18, 4, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x12}, 9, []byte{18, 20, 2, 4, 0, 0, 255, 255}},
+	&deflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}, 9, []byte{18, 132, 2, 64, 0, 0, 0, 255, 255}},
+}
+
+var deflateInflateTests = []*deflateInflateTest{
+	&deflateInflateTest{[]byte{}},
+	&deflateInflateTest{[]byte{0x11}},
+	&deflateInflateTest{[]byte{0x11, 0x12}},
+	&deflateInflateTest{[]byte{0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11, 0x11}},
+	&deflateInflateTest{[]byte{0x11, 0x10, 0x13, 0x41, 0x21, 0x21, 0x41, 0x13, 0x87, 0x78, 0x13}},
+	&deflateInflateTest{getLargeDataChunk()},
+}
+
+var reverseBitsTests = []*reverseBitsTest{
+	&reverseBitsTest{1, 1, 1},
+	&reverseBitsTest{1, 2, 2},
+	&reverseBitsTest{1, 3, 4},
+	&reverseBitsTest{1, 4, 8},
+	&reverseBitsTest{1, 5, 16},
+	&reverseBitsTest{17, 5, 17},
+	&reverseBitsTest{257, 9, 257},
+	&reverseBitsTest{29, 5, 23},
+}
+
+func getLargeDataChunk() []byte {
+	result := make([]byte, 100000)
+	for i := range result {
+		result[i] = byte(int64(i) * int64(i) & 0xFF)
+	}
+	return result
+}
+
+func TestDeflate(t *testing.T) {
+	for _, h := range deflateTests {
+		buffer := bytes.NewBuffer(nil)
+		w := NewWriter(buffer, h.level)
+		w.Write(h.in)
+		w.Close()
+		if bytes.Compare(buffer.Bytes(), h.out) != 0 {
+			t.Errorf("buffer is wrong; level = %v, buffer.Bytes() = %v, expected output = %v",
+				h.level, buffer.Bytes(), h.out)
+		}
+	}
+}
+
+type syncBuffer struct {
+	buf    bytes.Buffer
+	mu     sync.RWMutex
+	closed bool
+	ready  chan bool
+}
+
+func newSyncBuffer() *syncBuffer {
+	return &syncBuffer{ready: make(chan bool, 1)}
+}
+
+func (b *syncBuffer) Read(p []byte) (n int, err os.Error) {
+	for {
+		b.mu.RLock()
+		n, err = b.buf.Read(p)
+		b.mu.RUnlock()
+		if n > 0 || b.closed {
+			return
+		}
+		<-b.ready
+	}
+	panic("unreachable")
+}
+
+func (b *syncBuffer) Write(p []byte) (n int, err os.Error) {
+	n, err = b.buf.Write(p)
+	_ = b.ready <- true
+	return
+}
+
+func (b *syncBuffer) WriteMode() {
+	b.mu.Lock()
+}
+
+func (b *syncBuffer) ReadMode() {
+	b.mu.Unlock()
+	_ = b.ready <- true
+}
+
+func (b *syncBuffer) Close() os.Error {
+	b.closed = true
+	_ = b.ready <- true
+	return nil
+}
+
+func testSync(t *testing.T, level int, input []byte, name string) {
+	if len(input) == 0 {
+		return
+	}
+
+	t.Logf("--testSync %d, %d, %s", level, len(input), name)
+	buf := newSyncBuffer()
+	buf1 := new(bytes.Buffer)
+	buf.WriteMode()
+	w := NewWriter(io.MultiWriter(buf, buf1), level)
+	r := NewReader(buf)
+
+	// Write half the input and read back.
+	for i := 0; i < 2; i++ {
+		var lo, hi int
+		if i == 0 {
+			lo, hi = 0, (len(input)+1)/2
+		} else {
+			lo, hi = (len(input)+1)/2, len(input)
+		}
+		t.Logf("#%d: write %d-%d", i, lo, hi)
+		if _, err := w.Write(input[lo:hi]); err != nil {
+			t.Errorf("testSync: write: %v", err)
+			return
+		}
+		if i == 0 {
+			if err := w.Flush(); err != nil {
+				t.Errorf("testSync: flush: %v", err)
+				return
+			}
+		} else {
+			if err := w.Close(); err != nil {
+				t.Errorf("testSync: close: %v", err)
+			}
+		}
+		buf.ReadMode()
+		out := make([]byte, hi-lo+1)
+		m, err := io.ReadAtLeast(r, out, hi-lo)
+		t.Logf("#%d: read %d", i, m)
+		if m != hi-lo || err != nil {
+			t.Errorf("testSync/%d (%d, %d, %s): read %d: %d, %v (%d left)", i, level, len(input), name, hi-lo, m, err, buf.buf.Len())
+			return
+		}
+		if !bytes.Equal(input[lo:hi], out[:hi-lo]) {
+			t.Errorf("testSync/%d: read wrong bytes: %x vs %x", i, input[lo:hi], out[:hi-lo])
+			return
+		}
+		if i == 0 && buf.buf.Len() != 0 {
+			t.Errorf("testSync/%d (%d, %d, %s): extra data after %d", i, level, len(input), name, hi-lo)
+		}
+		buf.WriteMode()
+	}
+	buf.ReadMode()
+	out := make([]byte, 10)
+	if n, err := r.Read(out); n > 0 || err != os.EOF {
+		t.Errorf("testSync (%d, %d, %s): final Read: %d, %v (hex: %x)", level, len(input), name, n, err, out[0:n])
+	}
+	if buf.buf.Len() != 0 {
+		t.Errorf("testSync (%d, %d, %s): extra data at end", level, len(input), name)
+	}
+	r.Close()
+
+	// stream should work for ordinary reader too
+	r = NewReader(buf1)
+	out, err := ioutil.ReadAll(r)
+	if err != nil {
+		t.Errorf("testSync: read: %s", err)
+		return
+	}
+	r.Close()
+	if !bytes.Equal(input, out) {
+		t.Errorf("testSync: decompress(compress(data)) != data: level=%d input=%s", level, name)
+	}
+}
+
+
+func testToFromWithLevel(t *testing.T, level int, input []byte, name string) os.Error {
+	buffer := bytes.NewBuffer(nil)
+	w := NewWriter(buffer, level)
+	w.Write(input)
+	w.Close()
+	r := NewReader(buffer)
+	out, err := ioutil.ReadAll(r)
+	if err != nil {
+		t.Errorf("read: %s", err)
+		return err
+	}
+	r.Close()
+	if !bytes.Equal(input, out) {
+		t.Errorf("decompress(compress(data)) != data: level=%d input=%s", level, name)
+	}
+
+	testSync(t, level, input, name)
+	return nil
+}
+
+func testToFrom(t *testing.T, input []byte, name string) {
+	for i := 0; i < 10; i++ {
+		testToFromWithLevel(t, i, input, name)
+	}
+}
+
+func TestDeflateInflate(t *testing.T) {
+	for i, h := range deflateInflateTests {
+		testToFrom(t, h.in, fmt.Sprintf("#%d", i))
+	}
+}
+
+func TestReverseBits(t *testing.T) {
+	for _, h := range reverseBitsTests {
+		if v := reverseBits(h.in, h.bitCount); v != h.out {
+			t.Errorf("reverseBits(%v,%v) = %v, want %v",
+				h.in, h.bitCount, v, h.out)
+		}
+	}
+}
+
+func TestDeflateInflateString(t *testing.T) {
+	gold := bytes.NewBufferString(getEdata()).Bytes()
+	testToFromWithLevel(t, 1, gold, "2.718281828...")
+}
+
+func getEdata() string {
+	return "2.718281828459045235360287471352662497757247093699959574966967627724076630353547" +
+		"59457138217852516642742746639193200305992181741359662904357290033429526059563073" +
+		"81323286279434907632338298807531952510190115738341879307021540891499348841675092" +
+		"44761460668082264800168477411853742345442437107539077744992069551702761838606261" +
+		"33138458300075204493382656029760673711320070932870912744374704723069697720931014" +
+		"16928368190255151086574637721112523897844250569536967707854499699679468644549059" +
+		"87931636889230098793127736178215424999229576351482208269895193668033182528869398" +
+		"49646510582093923982948879332036250944311730123819706841614039701983767932068328" +
+		"23764648042953118023287825098194558153017567173613320698112509961818815930416903" +
+		"51598888519345807273866738589422879228499892086805825749279610484198444363463244" +
+		"96848756023362482704197862320900216099023530436994184914631409343173814364054625" +
+		"31520961836908887070167683964243781405927145635490613031072085103837505101157477" +
+		"04171898610687396965521267154688957035035402123407849819334321068170121005627880" +
+		"23519303322474501585390473041995777709350366041699732972508868769664035557071622" +
+		"68447162560798826517871341951246652010305921236677194325278675398558944896970964" +
+		"09754591856956380236370162112047742722836489613422516445078182442352948636372141" +
+		"74023889344124796357437026375529444833799801612549227850925778256209262264832627" +
+		"79333865664816277251640191059004916449982893150566047258027786318641551956532442" +
+		"58698294695930801915298721172556347546396447910145904090586298496791287406870504" +
+		"89585867174798546677575732056812884592054133405392200011378630094556068816674001" +
+		"69842055804033637953764520304024322566135278369511778838638744396625322498506549" +
+		"95886234281899707733276171783928034946501434558897071942586398772754710962953741" +
+		"52111513683506275260232648472870392076431005958411661205452970302364725492966693" +
+		"81151373227536450988890313602057248176585118063036442812314965507047510254465011" +
+		"72721155519486685080036853228183152196003735625279449515828418829478761085263981" +
+		"39559900673764829224437528718462457803619298197139914756448826260390338144182326" +
+		"25150974827987779964373089970388867782271383605772978824125611907176639465070633" +
+		"04527954661855096666185664709711344474016070462621568071748187784437143698821855" +
+		"96709591025968620023537185887485696522000503117343920732113908032936344797273559" +
+		"55277349071783793421637012050054513263835440001863239914907054797780566978533580" +
+		"48966906295119432473099587655236812859041383241160722602998330535370876138939639" +
+		"17795745401613722361878936526053815584158718692553860616477983402543512843961294" +
+		"60352913325942794904337299085731580290958631382683291477116396337092400316894586" +
+		"36060645845925126994655724839186564209752685082307544254599376917041977780085362" +
+		"73094171016343490769642372229435236612557250881477922315197477806056967253801718" +
+		"07763603462459278778465850656050780844211529697521890874019660906651803516501792" +
+		"50461950136658543663271254963990854914420001457476081930221206602433009641270489" +
+		"43903971771951806990869986066365832322787093765022601492910115171776359446020232" +
+		"49300280401867723910288097866605651183260043688508817157238669842242201024950551" +
+		"88169480322100251542649463981287367765892768816359831247788652014117411091360116" +
+		"49950766290779436460058519419985601626479076153210387275571269925182756879893027" +
+		"61761146162549356495903798045838182323368612016243736569846703785853305275833337" +
+		"93990752166069238053369887956513728559388349989470741618155012539706464817194670" +
+		"83481972144888987906765037959036696724949925452790337296361626589760394985767413" +
+		"97359441023744329709355477982629614591442936451428617158587339746791897571211956" +
+		"18738578364475844842355558105002561149239151889309946342841393608038309166281881" +
+		"15037152849670597416256282360921680751501777253874025642534708790891372917228286" +
+		"11515915683725241630772254406337875931059826760944203261924285317018781772960235" +
+		"41306067213604600038966109364709514141718577701418060644363681546444005331608778" +
+		"31431744408119494229755993140118886833148328027065538330046932901157441475631399" +
+		"97221703804617092894579096271662260740718749975359212756084414737823303270330168" +
+		"23719364800217328573493594756433412994302485023573221459784328264142168487872167" +
+		"33670106150942434569844018733128101079451272237378861260581656680537143961278887" +
+		"32527373890392890506865324138062796025930387727697783792868409325365880733988457" +
+		"21874602100531148335132385004782716937621800490479559795929059165547050577751430" +
+		"81751126989851884087185640260353055837378324229241856256442550226721559802740126" +
+		"17971928047139600689163828665277009752767069777036439260224372841840883251848770" +
+		"47263844037953016690546593746161932384036389313136432713768884102681121989127522" +
+		"30562567562547017250863497653672886059667527408686274079128565769963137897530346" +
+		"60616669804218267724560530660773899624218340859882071864682623215080288286359746" +
+		"83965435885668550377313129658797581050121491620765676995065971534476347032085321" +
+		"56036748286083786568030730626576334697742956346437167093971930608769634953288468" +
+		"33613038829431040800296873869117066666146800015121143442256023874474325250769387" +
+		"07777519329994213727721125884360871583483562696166198057252661220679754062106208" +
+		"06498829184543953015299820925030054982570433905535701686531205264956148572492573" +
+		"86206917403695213533732531666345466588597286659451136441370331393672118569553952" +
+		"10845840724432383558606310680696492485123263269951460359603729725319836842336390" +
+		"46321367101161928217111502828016044880588023820319814930963695967358327420249882" +
+		"45684941273860566491352526706046234450549227581151709314921879592718001940968866" +
+		"98683703730220047531433818109270803001720593553052070070607223399946399057131158" +
+		"70996357773590271962850611465148375262095653467132900259943976631145459026858989" +
+		"79115837093419370441155121920117164880566945938131183843765620627846310490346293" +
+		"95002945834116482411496975832601180073169943739350696629571241027323913874175492" +
+		"30718624545432220395527352952402459038057445028922468862853365422138157221311632" +
+		"88112052146489805180092024719391710555390113943316681515828843687606961102505171" +
+		"00739276238555338627255353883096067164466237092264680967125406186950214317621166" +
+		"81400975952814939072226011126811531083873176173232352636058381731510345957365382" +
+		"23534992935822836851007810884634349983518404451704270189381994243410090575376257" +
+		"76757111809008816418331920196262341628816652137471732547772778348877436651882875" +
+		"21566857195063719365653903894493664217640031215278702223664636357555035655769488" +
+		"86549500270853923617105502131147413744106134445544192101336172996285694899193369" +
+		"18472947858072915608851039678195942983318648075608367955149663644896559294818785" +
+		"17840387733262470519450504198477420141839477312028158868457072905440575106012852" +
+		"58056594703046836344592652552137008068752009593453607316226118728173928074623094" +
+		"68536782310609792159936001994623799343421068781349734695924646975250624695861690" +
+		"91785739765951993929939955675427146549104568607020990126068187049841780791739240" +
+		"71945996323060254707901774527513186809982284730860766536866855516467702911336827" +
+		"56310722334672611370549079536583453863719623585631261838715677411873852772292259" +
+		"47433737856955384562468010139057278710165129666367644518724656537304024436841408" +
+		"14488732957847348490003019477888020460324660842875351848364959195082888323206522" +
+		"12810419044804724794929134228495197002260131043006241071797150279343326340799596" +
+		"05314460532304885289729176598760166678119379323724538572096075822771784833616135" +
+		"82612896226118129455927462767137794487586753657544861407611931125958512655759734" +
+		"57301533364263076798544338576171533346232527057200530398828949903425956623297578" +
+		"24887350292591668258944568946559926584547626945287805165017206747854178879822768" +
+		"06536650641910973434528878338621726156269582654478205672987756426325321594294418" +
+		"03994321700009054265076309558846589517170914760743713689331946909098190450129030" +
+		"70995662266203031826493657336984195557769637876249188528656866076005660256054457" +
+		"11337286840205574416030837052312242587223438854123179481388550075689381124935386" +
+		"31863528708379984569261998179452336408742959118074745341955142035172618420084550" +
+		"91708456823682008977394558426792142734775608796442792027083121501564063413416171" +
+		"66448069815483764491573900121217041547872591998943825364950514771379399147205219" +
+		"52907939613762110723849429061635760459623125350606853765142311534966568371511660" +
+		"42207963944666211632551577290709784731562782775987881364919512574833287937715714" +
+		"59091064841642678309949723674420175862269402159407924480541255360431317992696739" +
+		"15754241929660731239376354213923061787675395871143610408940996608947141834069836" +
+		"29936753626215452472984642137528910798843813060955526227208375186298370667872244" +
+		"30195793793786072107254277289071732854874374355781966511716618330881129120245204" +
+		"04868220007234403502544820283425418788465360259150644527165770004452109773558589" +
+		"76226554849416217149895323834216001140629507184904277892585527430352213968356790" +
+		"18076406042138307308774460170842688272261177180842664333651780002171903449234264" +
+		"26629226145600433738386833555534345300426481847398921562708609565062934040526494" +
+		"32442614456659212912256488935696550091543064261342526684725949143142393988454324" +
+		"86327461842846655985332312210466259890141712103446084271616619001257195870793217" +
+		"56969854401339762209674945418540711844643394699016269835160784892451405894094639" +
+		"52678073545797003070511636825194877011897640028276484141605872061841852971891540" +
+		"19688253289309149665345753571427318482016384644832499037886069008072709327673127" +
+		"58196656394114896171683298045513972950668760474091542042842999354102582911350224" +
+		"16907694316685742425225090269390348148564513030699251995904363840284292674125734" +
+		"22447765584177886171737265462085498294498946787350929581652632072258992368768457" +
+		"01782303809656788311228930580914057261086588484587310165815116753332767488701482" +
+		"91674197015125597825727074064318086014281490241467804723275976842696339357735429" +
+		"30186739439716388611764209004068663398856841681003872389214483176070116684503887" +
+		"21236436704331409115573328018297798873659091665961240202177855885487617616198937" +
+		"07943800566633648843650891448055710397652146960276625835990519870423001794655367" +
+		"9"
+}