hours_test.go

package harvest

import (
	"os"
	"testing"

	"encoding/json"
	"time"
)

var (
	testDataFile = "test/test_timeEntries.json"
)

func TestDailyHours(t *testing.T) {
	var expect float64
	var timeEntries TimeEntries
	expect = 7.5

	body, err := os.ReadFile(testDataFile)
	if err != nil {
		t.Errorf("[ERROR] %v", err)
	}

	err = json.Unmarshal(body, &timeEntries)
	if err != nil {
		t.Errorf("[ERROR] %v", err)
	}

	date, _ := time.Parse("2006-01-02", "2019-04-15")
	result := timeEntries.DailyHours(date)
	if result != expect {
		t.Errorf("DailyHours: expecting %v, got %v!", expect, result)
	} else {
		t.Logf("[PASS] DailyHours: expecting %v, got %v!", expect, result)
	}

	expect = 5.5
	date, _ = time.Parse("2006-01-02", "2019-04-16")
	result = timeEntries.DailyHours(date)
	if result != expect {
		t.Errorf("DailyHours: expecting %v, got %v!", expect, result)
	} else {
		t.Logf("[PASS] DailyHours: expecting %v, got %v!", expect, result)
	}
}

func TestDailyTotalsReturnsExpectedHours(t *testing.T) {
	expectedHours := 5.5
	timeEntries, _ := readTimeEntriesFromFile()
	date, _ := time.Parse("2006-01-02", "2019-04-16")

	actualHours, _, _ := timeEntries.DailyTotals(date)

	if actualHours != expectedHours {
		t.Errorf("DailyTotals.Hours: expecting %v, got %v!", expectedHours, actualHours)
	}
}

func TestDailyTotalsReturnsExpectedSaldo(t *testing.T) {
	expectedSaldo := 0.0
	timeEntries, _ := readTimeEntriesFromFile()
	date, _ := time.Parse("2006-01-02", "2019-04-16")

	_, actualSaldo, _ := timeEntries.DailyTotals(date)

	if actualSaldo != expectedSaldo {
		t.Errorf("Incorrect saldo: expected %v, got %v!", expectedSaldo, actualSaldo)
	}
}

func TestDailyTotalsReturnsExpectedOvertime(t *testing.T) {
	expectedOvertime := -2.0
	timeEntries, _ := readTimeEntriesFromFile()
	date, _ := time.Parse("2006-01-02", "2019-04-16")

	_, _, actualOvertime := timeEntries.DailyTotals(date)

	if actualOvertime != expectedOvertime {
		t.Errorf("Incorrect overtime: expecting %v, got %v!", expectedOvertime, actualOvertime)
	}
}

func TestTotalHours(t *testing.T) {
	var expect float64
	var timeEntries TimeEntries
	expectWeekOne := 47.0
	expectWeekTwo := 39.5
	expect = expectWeekOne + expectWeekTwo

	body, err := os.ReadFile(testDataFile)
	if err != nil {
		t.Errorf("[ERROR] Could not read test data.\n[ERROR] %v", err)
	}

	err = json.Unmarshal(body, &timeEntries)
	if err != nil {
		t.Errorf("[ERROR] Could not unmarshal test data.\n[ERROR] %v", err)
	}

	result := timeEntries.TotalHours()
	if result != expect {
		t.Errorf("TotalHours: expecting %v, got %v!", expect, result)
	} else {
		t.Logf("[PASS] TotalHours: expecting %v, got %v!", expect, result)
	}
}

func readTimeEntriesFromFile() (TimeEntries, error) {
	body, _ := os.ReadFile(testDataFile)

	var timeEntries TimeEntries
	err := json.Unmarshal(body, &timeEntries)

	if err != nil {
		return timeEntries, err
	}

	return timeEntries, err
}