79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package timespan
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/traggo/server/generated/gqlmodel"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/traggo/server/model"
|
|
"github.com/traggo/server/test"
|
|
"github.com/traggo/server/test/fake"
|
|
)
|
|
|
|
func TestRemoveTimeSpan_succeeds_removesTimeSpan(t *testing.T) {
|
|
db := test.InMemoryDB(t)
|
|
defer db.Close()
|
|
user := db.User(3)
|
|
ts := user.TimeSpan("2019-06-11T18:00:00Z", "2019-06-11T18:00:00Z")
|
|
ts.Tag("hello", "world")
|
|
|
|
resolver := ResolverForTimeSpan{DB: db.DB}
|
|
actual, err := resolver.RemoveTimeSpan(fake.User(3), ts.TimeSpan.ID)
|
|
require.NoError(t, err)
|
|
expected := &gqlmodel.TimeSpan{
|
|
ID: ts.TimeSpan.ID,
|
|
Start: test.ModelTime("2019-06-11T18:00:00Z"),
|
|
End: test.ModelTimeP("2019-06-11T18:00:00Z"),
|
|
Tags: []*gqlmodel.TimeSpanTag{
|
|
{Key: "hello", Value: "world"},
|
|
},
|
|
}
|
|
require.Equal(t, expected, actual)
|
|
assertTimeSpanCount(t, db, 0)
|
|
}
|
|
|
|
func TestRemoveTimeSpan_succeeds_removesTags(t *testing.T) {
|
|
db := test.InMemoryDB(t)
|
|
defer db.Close()
|
|
user := db.User(3)
|
|
ts := user.TimeSpan("2019-06-11T18:00:00Z", "2019-06-11T18:00:00Z")
|
|
ts.Tag("hello", "world")
|
|
|
|
resolver := ResolverForTimeSpan{DB: db.DB}
|
|
_, err := resolver.RemoveTimeSpan(fake.User(3), ts.TimeSpan.ID)
|
|
require.NoError(t, err)
|
|
|
|
ts.AssertHasTag("hello", "world", false)
|
|
}
|
|
|
|
func TestRemoveTimeSpan_fails_notExistingTimeSpan(t *testing.T) {
|
|
db := test.InMemoryDB(t)
|
|
defer db.Close()
|
|
db.User(3)
|
|
|
|
resolver := ResolverForTimeSpan{DB: db.DB}
|
|
_, err := resolver.RemoveTimeSpan(fake.User(3), 5)
|
|
require.EqualError(t, err, "timespan with id 5 does not exist")
|
|
}
|
|
|
|
func TestRemoveTimeSpan_fails_noPermission(t *testing.T) {
|
|
db := test.InMemoryDB(t)
|
|
defer db.Close()
|
|
db.User(3)
|
|
db.User(5)
|
|
db.Create(&model.TimeSpan{
|
|
StartUserTime: test.Time("2019-06-11T18:00:00Z"),
|
|
StartUTC: test.Time("2019-06-11T18:00:00Z"),
|
|
EndUserTime: nil,
|
|
EndUTC: nil,
|
|
OffsetUTC: 0,
|
|
ID: 1,
|
|
UserID: 3,
|
|
})
|
|
|
|
resolver := ResolverForTimeSpan{DB: db.DB}
|
|
_, err := resolver.RemoveTimeSpan(fake.User(5), 1)
|
|
require.EqualError(t, err, "timespan with id 1 does not exist")
|
|
}
|