forked from mirrors/homebox
66 lines
1.4 KiB
Go
66 lines
1.4 KiB
Go
|
package repo
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMaintenanceEntryRepository_GetLog(t *testing.T) {
|
||
|
|
item := useItems(t, 1)[0]
|
||
|
|
|
||
|
|
// Create 10 maintenance entries for the item
|
||
|
|
created := make([]MaintenanceEntryCreate, 10)
|
||
|
|
|
||
|
|
lastMonth := time.Now().AddDate(0, -1, 0)
|
||
|
|
thisMonth := time.Now()
|
||
|
|
|
||
|
|
for i := 0; i < 10; i++ {
|
||
|
|
dt := lastMonth
|
||
|
|
if i%2 == 0 {
|
||
|
|
dt = thisMonth
|
||
|
|
}
|
||
|
|
|
||
|
|
created[i] = MaintenanceEntryCreate{
|
||
|
|
Date: dt,
|
||
|
|
Name: "Maintenance",
|
||
|
|
Description: "Maintenance description",
|
||
|
|
Cost: 10,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, entry := range created {
|
||
|
|
_, err := tRepos.MaintEntry.Create(context.Background(), item.ID, entry)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to create maintenance entry: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Get the log for the item
|
||
|
|
log, err := tRepos.MaintEntry.GetLog(context.Background(), item.ID)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("failed to get maintenance log: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, item.ID, log.ItemID)
|
||
|
|
assert.Equal(t, 10, len(log.Entries))
|
||
|
|
|
||
|
|
// Calculate the average cost
|
||
|
|
var total float64
|
||
|
|
|
||
|
|
for _, entry := range log.Entries {
|
||
|
|
total += entry.Cost
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, total, log.CostTotal, "total cost should be equal to the sum of all entries")
|
||
|
|
assert.Equal(t, total/2, log.CostAverage, "average cost should be the average of the two months")
|
||
|
|
|
||
|
|
for _, entry := range log.Entries {
|
||
|
|
err := tRepos.MaintEntry.Delete(context.Background(), entry.ID)
|
||
|
|
assert.NoError(t, err)
|
||
|
|
}
|
||
|
|
}
|