mirror of
https://github.com/hay-kot/homebox.git
synced 2024-12-20 22:06:31 +00:00
bd321af29f
* new PR tasks * add homebox to know words * formatting * bump deps * generate db models * ts errors * drop id * fix accessor * drop unused time field * change CI * add expected error * add type check * resolve serveral type errors * hoise in CI
85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// get the previous month from the current month, accounts for errors when run
|
|
// near the beginning or end of the month/year
|
|
func getPrevMonth(now time.Time) time.Time {
|
|
t := now.AddDate(0, -1, 0)
|
|
|
|
// avoid infinite loop
|
|
max := 15
|
|
for t.Month() == now.Month() {
|
|
println("month is the same")
|
|
t = t.AddDate(0, 0, -1)
|
|
println(t.String())
|
|
|
|
max--
|
|
if max == 0 {
|
|
panic("max exceeded")
|
|
}
|
|
}
|
|
|
|
return t
|
|
}
|
|
|
|
func TestMaintenanceEntryRepository_GetLog(t *testing.T) {
|
|
item := useItems(t, 1)[0]
|
|
|
|
// Create 10 maintenance entries for the item
|
|
created := make([]MaintenanceEntryCreate, 10)
|
|
|
|
thisMonth := time.Now()
|
|
lastMonth := getPrevMonth(thisMonth)
|
|
|
|
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)
|
|
}
|
|
}
|