cam2ip/camera/camera_test.go

package camera

import (
	"fmt"
	"image/jpeg"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"
	"time"
)

func TestCamera(t *testing.T) {
	camera, err := NewCamera(1)
	if err != nil {
		t.Fatal(err)
	}

	defer camera.Close()

	tmpdir, err := ioutil.TempDir(os.TempDir(), "cam2ip")
	if err != nil {
		t.Error(err)
	}

	defer os.RemoveAll(tmpdir)

	var width, height float64 = 640, 480
	camera.SetProperty(PropFrameWidth, width)
	camera.SetProperty(PropFrameHeight, height)

	if camera.GetProperty(PropFrameWidth) != width {
		t.Error("FrameWidth not correct")
	}

	if camera.GetProperty(PropFrameHeight) != height {
		t.Error("FrameHeight not correct")
	}

	var i int
	var n int = 10

	timeout := time.After(time.Duration(n) * time.Second)

	for {
		select {
		case <-timeout:
			//fmt.Printf("Fps: %d\n", i/n)
			return
		default:
			i += 1

			img, err := camera.Read()
			if err != nil {
				t.Error(err)
			}

			file, err := os.Create(filepath.Join(tmpdir, fmt.Sprintf("%03d.jpg", i)))
			if err != nil {
				t.Error(err)
			}

			err = jpeg.Encode(file, img, &jpeg.Options{Quality: 75})
			if err != nil {
				t.Error(err)
			}

			err = file.Close()
			if err != nil {
				t.Error(err)
			}
		}
	}
}