job_type_test.go

// Copyright 2015 Alex Browne.  All rights reserved.
// Use of this source code is governed by the MIT
// license, which can be found in the LICENSE file.

package jobs

import (
	"testing"
	"time"
)

func TestRegisterType(t *testing.T) {
	testingSetUp()
	defer testingTeardown()
	// Reset job types
	Types = map[string]*Type{}
	// Make sure we can register a job type without error
	testJobName := "testJob1"
	testJobRetries := uint(3)
	Type, err := RegisterType(testJobName, testJobRetries, noOpHandler)
	if err != nil {
		t.Fatalf("Unexpected err registering job type: %s", err.Error())
	}
	// Make sure the name property is correct
	if Type.name != testJobName {
		t.Errorf("Got wrong name for job type. Expected %s but got %s", testJobName, Type.name)
	}
	// Make sure the retries property is correct
	if Type.retries != testJobRetries {
		t.Errorf("Got wrong number of retries for job type. Expected %d but got %d", testJobRetries, Type.retries)
	}
	// Make sure the Type was added to the global map
	if _, found := Types[testJobName]; !found {
		t.Errorf("Type was not added to the global map of job types.")
	}
	// Make sure we cannot register a job type with the same name
	if _, err := RegisterType(testJobName, 0, noOpHandler); err == nil {
		t.Errorf("Expected error when registering job with the same name but got none")
	} else if _, ok := err.(ErrorNameAlreadyRegistered); !ok {
		t.Errorf("Expected ErrorNameAlreadyRegistered but got error of type %T", err)
	}
	// Make sure we can register a job type with a handler function that has an argument
	if _, err := RegisterType("testJobWithArg", 0, func(s string) error { print(s); return nil }); err != nil {
		t.Errorf("Unexpected err registering job type with handler with one argument: %s", err)
	}
	// Make sure we cannot register a job type with an invalid handler
	invalidHandlers := []interface{}{
		"notAFunc",
		func(a, b string) error { return nil },
	}
	for _, handler := range invalidHandlers {
		if _, err := RegisterType("testJobWithInvalidHandler", 0, handler); err == nil {
			t.Errorf("Expected error when registering job with invalid handler type %T %v, but got none.", handler, handler)
		}
	}
}

func TestTypeSchedule(t *testing.T) {
	testingSetUp()
	defer testingTeardown()
	// Register a new job type
	testJobName := "testJob1"
	testJobPriority := 100
	testJobTime := time.Now()
	testJobData := "testData"
	encodedData, err := encode(testJobData)
	if err != nil {
		t.Errorf("Unexpected error encoding data: %s", err)
	}
	encodedTime := testJobTime.UTC().UnixNano()
	Type, err := RegisterType(testJobName, 0, func(string) error { return nil })
	if err != nil {
		t.Fatalf("Unexpected error registering job type: %s", err)
	}
	// Call Schedule
	job, err := Type.Schedule(testJobPriority, testJobTime, testJobData)
	if err != nil {
		t.Errorf("Unexpected error in Type.Schedule(): %s", err)
	}
	// Make sure the job was saved in the database correctly
	if job.id == "" {
		t.Errorf("After Type.Schedule, job.id was empty.")
	}
	expectKeyExists(t, job.Key())
	expectJobFieldEquals(t, job, "priority", testJobPriority, intConverter)
	expectJobFieldEquals(t, job, "time", encodedTime, int64Converter)
	expectJobFieldEquals(t, job, "data", encodedData, bytesConverter)
	expectStatusEquals(t, job, StatusQueued)
	// Make sure we get an error if the data is not the correct type
	if _, err := Type.Schedule(0, time.Now(), 0); err == nil {
		t.Errorf("Expected error when calling Type.Schedule with incorrect data type but got none")
	}
}