gosax_test.go

// Tests.
//
// Eli Bendersky [https://eli.thegreenplace.net]
// This code is in the public domain.
package gosax

import (
	"bytes"
	"strings"
	"testing"
)

func TestInit(*testing.T) {
	// Test that nothing crashed in init()
}

func TestBasic(t *testing.T) {
	var plantId string
	var numOrigins int
	var startDoc bool
	var endDoc bool

	scb := SaxCallbacks{
		StartDocument: func() {
			startDoc = true
		},
		EndDocument: func() {
			endDoc = true
		},
		StartElement: func(name string, attrs []string) {
			if name == "plant" {
				if len(attrs) < 2 {
					t.Errorf("want len(attrs) at least 2, got %v", len(attrs))
				}
				if attrs[0] != "id" {
					t.Errorf("want 'id' attr, got %v", attrs[0])
				}
				plantId = attrs[1]
			} else if name == "origin" {
				numOrigins++
			}
		},
		EndElement: func(name string) {
		},
	}

	err := ParseFile("testfiles/fruit.xml", scb)
	if err != nil {
		panic(err)
	}

	if plantId != "27" {
		t.Errorf("want plant id %v, got %v", 27, plantId)
	}

	if numOrigins != 2 {
		t.Errorf("want num origins 2, got %v", numOrigins)
	}
	if !startDoc {
		t.Errorf("want doc start, found none")
	}

	if !endDoc {
		t.Errorf("want doc end, found none")
	}
}

func TestCharacters(t *testing.T) {
	m := make(map[string]bool)
	scb := SaxCallbacks{
		Characters: func(contents string) {
			m[contents] = true
		},
	}

	err := ParseFile("testfiles/fruit.xml", scb)
	if err != nil {
		panic(err)
	}

	chars := []string{"Coffee", "Ethiopia", "Brazil"}
	for _, c := range chars {
		if _, ok := m[c]; !ok {
			t.Errorf("expected to find %v characters", c)
		}
	}
}

func TestError(t *testing.T) {
	scb := SaxCallbacks{}
	err := ParseFile("testfiles/badfile.xml", scb)
	if err == nil {
		t.Errorf("want non-nil error")
	}
	if !strings.Contains(err.Error(), "Start tag expected") {
		t.Errorf("want start tag error, got '%v'", err)
	}
}

func TestMemParse(t *testing.T) {
	m := make(map[string]bool)
	scb := SaxCallbacks{
		Characters: func(contents string) {
			m[contents] = true
		},
	}
	buf := bytes.Buffer{}
	buf.WriteString(`<?xml version="1.0" encoding="UTF-8"?>
	<plant id="27">
	<name>Coffee</name>
	<origin>Ethiopia</origin>
	<origin>Brazil</origin>
	</plant>`)
	err := ParseMem(buf, scb)
	if err != nil {
		panic(err)
	}

	chars := []string{"Coffee", "Ethiopia", "Brazil"}
	for _, c := range chars {
		if _, ok := m[c]; !ok {
			t.Errorf("expected to find %v characters", c)
		}
	}
}