Using log instead of fmt

This commit is contained in:
saxon 2018-07-04 21:43:17 +09:30
parent c861862ff7
commit 0933912f76
1 changed files with 39 additions and 11 deletions

View File

@ -1,7 +1,34 @@
/*
NAME
parser_test.go
DESCRIPTION
See Readme.md
AUTHOR
Saxon Nelson-Milton <saxon@ausocean.org>
LICENSE
parser_test.go is Copyright (C) 2017 the Australian Ocean Lab (AusOcean)
It is free software: you can redistribute it and/or modify them
under the terms of the GNU General Public License as published by the
Free Software Foundation, either version 3 of the License, or (at your
option) any later version.
It is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
for more details.
You should have received a copy of the GNU General Public License
along with revid in gpl.txt. If not, see http://www.gnu.org/licenses.
*/
package parser package parser
import ( import (
"fmt" "log"
"os" "os"
"strconv" "strconv"
"testing" "testing"
@ -13,21 +40,22 @@ const (
) )
func TestH264Parser(t *testing.T) { func TestH264Parser(t *testing.T) {
fmt.Println("Opening input file!") log.SetOutput(os.Stderr)
log.Println("Opening input file!")
inputFile, err := os.Open(h264InputFileName) inputFile, err := os.Open(h264InputFileName)
if err != nil { if err != nil {
t.Errorf("Should not have got error opening file!") t.Errorf("Should not have got error opening file!")
return return
} }
fmt.Println("Getting file stats!") log.Println("Getting file stats!")
stats, err := inputFile.Stat() stats, err := inputFile.Stat()
if err != nil { if err != nil {
t.Errorf("Could not get input file stats!") t.Errorf("Could not get input file stats!")
return return
} }
fmt.Println("Creating space for file data and reading!") log.Println("Creating space for file data and reading!")
data := make([]byte, stats.Size()) data := make([]byte, stats.Size())
_, err = inputFile.Read(data) _, err = inputFile.Read(data)
if err != nil { if err != nil {
@ -36,7 +64,7 @@ func TestH264Parser(t *testing.T) {
} }
// Create 'parser' and start it up // Create 'parser' and start it up
fmt.Println("Creating parser!") log.Println("Creating parser!")
parser := NewH264Parser() parser := NewH264Parser()
parser.SetOutputChan(make(chan []byte, 10000)) parser.SetOutputChan(make(chan []byte, 10000))
parser.Start() parser.Start()
@ -67,34 +95,34 @@ func TestH264Parser(t *testing.T) {
/* /*
func TestMJPEGParser(t *testing.T) { func TestMJPEGParser(t *testing.T) {
fmt.Println("Opening input file!") log.Println("Opening input file!")
// Open the input file // Open the input file
inputFile, err := os.Open(testInputFileName) inputFile, err := os.Open(testInputFileName)
if err != nil { if err != nil {
t.Errorf("Should not have got error opening file!") t.Errorf("Should not have got error opening file!")
} }
fmt.Println("Getting file stats!") log.Println("Getting file stats!")
stats, err := inputFile.Stat() stats, err := inputFile.Stat()
if err != nil { if err != nil {
t.Errorf("Could not get input file stats!") t.Errorf("Could not get input file stats!")
return return
} }
fmt.Println("Creating space for file data!") log.Println("Creating space for file data!")
data := make([]byte, stats.Size()) data := make([]byte, stats.Size())
_, err = inputFile.Read(data) _, err = inputFile.Read(data)
if err != nil { if err != nil {
t.Errorf("Should not have got read error!") t.Errorf("Should not have got read error!")
return return
} }
fmt.Println("Creating parser!") log.Println("Creating parser!")
parser := NewMJPEGParser(len(data) + 1) parser := NewMJPEGParser(len(data) + 1)
parser.SetOutputChan(make(chan []byte, 10000)) parser.SetOutputChan(make(chan []byte, 10000))
parser.Start() parser.Start()
fmt.Printf("len(data): %v\n", len(data)) log.Printf("len(data): %v\n", len(data))
for i := range data { for i := range data {
parser.GetInputChan() <- data[i] parser.GetInputChan() <- data[i]
} }
fmt.Println("Writing jpegs to files!") log.Println("Writing jpegs to files!")
for i := 0; len(parser.GetOutputChan()) > 0; i++ { for i := 0; len(parser.GetOutputChan()) > 0; i++ {
// Open a new output file // Open a new output file
outputFile, err := os.Create("testOutput/image" + strconv.Itoa(i) + ".jpeg") outputFile, err := os.Create("testOutput/image" + strconv.Itoa(i) + ".jpeg")