Test some more errors

Author: markus-wa

demoinfocs_test.go

@@ -2,6 +2,7 @@ package demoinfocs_test
 
 import (
 	"bytes"
+	"crypto/rand"
 	"fmt"
 	"io/ioutil"
 	"os"
@@ -258,6 +259,33 @@ func TestCancelParseToEnd(t *testing.T) {
 	}
 }
 
+func TestInvalidFileType(t *testing.T) {
+	invalidDemoData := make([]byte, 2048)
+	rand.Read(invalidDemoData)
+
+	p := dem.NewParser(bytes.NewBuffer(invalidDemoData))
+
+	_, err := p.ParseHeader()
+	if err != dem.ErrInvalidFileType {
+		t.Fatal("Invalid demo but error was not ErrInvalidFileType:", err)
+	}
+}
+
+func TestHeaderNotParsed(t *testing.T) {
+	f, err := os.Open(defaultDemPath)
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer f.Close()
+
+	p := dem.NewParser(f)
+
+	err = p.ParseToEnd()
+	if err != dem.ErrHeaderNotParsed {
+		t.Fatal("Tried to parse tick before header but error was not ErrHeaderNotParsed:", err)
+	}
+}
+
 func TestConcurrent(t *testing.T) {
 	var i int64
 	runner := func() {