41 lines
741 B
Go
41 lines
741 B
Go
package fixed
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestZonesForDay(t *testing.T) {
|
|
zones := Zones{
|
|
{Days: nil},
|
|
{Days: []Day{Monday, Tuesday, Wednesday}},
|
|
{Days: []Day{Wednesday, Thursday, Friday}},
|
|
}
|
|
|
|
assert.Len(t, zones.ForDay(Sunday), 1)
|
|
assert.Len(t, zones.ForDay(Monday), 2)
|
|
assert.Len(t, zones.ForDay(Wednesday), 3)
|
|
assert.Len(t, zones.ForDay(Thursday), 2)
|
|
}
|
|
|
|
func TestZonesTimeTableMarkers(t *testing.T) {
|
|
zones := Zones{
|
|
{Hours: TimeRange{
|
|
From: HourMin{1, 0},
|
|
To: HourMin{2, 0},
|
|
}},
|
|
{Hours: TimeRange{
|
|
From: HourMin{3, 30},
|
|
To: HourMin{4, 30},
|
|
}},
|
|
}
|
|
|
|
assert.Equal(t, []HourMin{
|
|
{0, 0},
|
|
{1, 0},
|
|
{2, 0},
|
|
{3, 30},
|
|
{4, 30},
|
|
}, zones.TimeTableMarkers())
|
|
}
|