more test coverage

zplgfa_test.go

@@ -1,7 +1,14 @@
 package zplgfa
 
 import (
+	"image"
+	_ "image/gif"
+	_ "image/jpeg"
+	_ "image/png"
+	"log"
+	"os"
 	"testing"
+	"strings"
 )
 
 func Test_CompressASCII(t *testing.T) {
@@ -9,3 +16,44 @@ func Test_CompressASCII(t *testing.T) {
 		t.Fatalf("CompressASCII failed")
 	}
 }
+
+func Test_ConvertToZPL(t *testing.T) {
+	// open file
+	file, err := os.Open("./test.png")
+	if err != nil {
+		log.Printf("Warning: could not open the file: %s\n", err)
+		return
+	}
+
+	defer file.Close()
+
+	// load image head information
+	config, format, err := image.DecodeConfig(file)
+	if err != nil {
+		log.Printf("Warning: image not compatible, format: %s, config: %v, error: %s\n", format, config, err)
+	}
+
+	// reset file pointer to the beginning of the file
+	file.Seek(0, 0)
+
+	// load and decode image
+	img, _, err := image.Decode(file)
+	if err != nil {
+		log.Printf("Warning: could not decode the file, %s\n", err)
+		return
+	}
+
+	// flatten image
+	flat := FlattenImage(img)
+
+	// convert image to zpl compatible type
+	gfimg := ConvertToZPL(flat, CompressedASCII)
+	
+	// remove whitespace - only for the test
+	gfimg = strings.Replace(gfimg, " ", "", -1)
+	gfimg = strings.Replace(gfimg, "\n", "", -1)
+
+	if gfimg != "^XA,^FS^FO0,0^GFA,32,51,3,,::01C000::,001C00::,1DDC00::,::^FS,^XZ" {
+		t.Fatalf("ConvertToZPL failed")
+	}
+}