reader_test.go

// Copyright 2016 The Mellium Contributors.
// Use of this source code is governed by the BSD 2-clause
// license that can be found in the LICENSE file.

package xmlstream

import (
	"encoding/xml"
	"fmt"
	"io"
	"strings"
	"testing"
)

var innerReaderTests = [...]struct {
	R    io.Reader
	Read string
	Err  error
}{
	0: {
		R:    strings.NewReader(``),
		Read: ``,
		Err:  nil,
	},
	1: {
		R:    strings.NewReader(`<test></test>`),
		Read: ``,
		Err:  nil,
	},
	2: {
		R:    strings.NewReader(`<test><inner/></test>`),
		Read: `<inner/>`,
		Err:  nil,
	},
	3: {
		R:    strings.NewReader(`<test>Inner</test>`),
		Read: `Inner`,
		Err:  nil,
	},
	4: {
		R:    strings.NewReader(`<test>Inner</oops>`),
		Read: `Inner`,
		Err:  unexpectedEndError{xml.Name{Local: "oops"}},
	},
	5: {
		R:    strings.NewReader(`<stream xmlns="stream">Test</stream>`),
		Read: `Test`,
		Err:  nil,
	},
	6: {
		R:    strings.NewReader(`<stream:stream><stream:features></stream:stream>`),
		Read: `<stream:features>`,
		Err:  nil,
	},
	7: {
		R:    strings.NewReader(`<stream:stream><stream:features> <stream:stream>`),
		Read: `<stream:features>`,
		Err:  errNotEnd,
	},
	8: {
		R:    strings.NewReader(`<stream:stream><stream:features><stream:stream>`),
		Read: `<stream:features`,
		Err:  errNotEnd,
	},
	9: {
		R:    strings.NewReader(`</stream:stream>`),
		Read: ``,
		Err:  errNotStart,
	},
	10: {
		R:    strings.NewReader(`<!-- Test -->`),
		Read: ``,
		Err:  errNotStart,
	},
	11: {
		R:    strings.NewReader(`What is dis junk?`),
		Read: ``,
		Err:  errNotStart,
	},
	12: {
		R:    strings.NewReader(`<test/>`),
		Read: ``,
		Err:  nil,
	},
	13: {
		R:    strings.NewReader(`<test:test/>`),
		Read: ``,
		Err:  nil,
	},
	14: {
		R:    strings.NewReader(`<stream:stream></stream:oops>`),
		Read: ``,
		Err:  unexpectedEndError{xml.Name{Local: "oops", Space: "stream"}},
	},
	15: {
		R:    strings.NewReader(`<stream:stream></oops:stream>`),
		Read: ``,
		Err:  unexpectedEndError{xml.Name{Local: "stream", Space: "oops"}},
	},
	16: {
		R:    strings.NewReader(`<stream:stream>Inner</stream:oops>`),
		Read: `Inn`,
		Err:  errNotEnd,
	},
	17: {
		R:    strings.NewReader(`<stream:stream></stream:oooooooooooo>`),
		Read: `</stre`,
		Err:  errNotEnd,
	},
	18: {
		R:    strings.NewReader(`<stream:stream></oooooooooooo:stream>`),
		Read: `</oooo`,
		Err:  errNotEnd,
	},
}

func TestInnerReader(t *testing.T) {
	for i, tc := range innerReaderTests {
		t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
			ir := InnerReader(tc.R)
			if ir == nil {
				t.Fatal("InnerReader returned nil reader")
			}
			b, err := io.ReadAll(ir)
			if err != tc.Err {
				t.Errorf("Unxpected error: want=`%v`, got=`%v`", tc.Err, err)
			}
			if string(b) != tc.Read {
				t.Errorf("Unexpected value read: want=`%s`, got=`%s`", tc.Read, b)
			}
		})
	}
}