get_test.go

package device

import (
	"testing"

	"github.com/stretchr/testify/require"
	"github.com/traggo/server/generated/gqlmodel"
	"github.com/traggo/server/model"
	"github.com/traggo/server/test"
	"github.com/traggo/server/test/fake"
)

func TestGQL_Devices(t *testing.T) {
	db := test.InMemoryDB(t)
	defer db.Close()
	db.User(1)
	db.User(2)
	db.Create(&model.Device{
		ID:        1,
		Name:      "Android",
		Token:     "abc",
		UserID:    1,
		CreatedAt: test.Time("2009-06-30T18:30:00Z"),
		ActiveAt:  test.Time("2018-06-30T18:30:00Z"),
		Type:      model.TypeNoExpiry,
	})
	db.Create(&model.Device{
		ID:        2,
		Name:      "Browser",
		Token:     "abcd",
		UserID:    2,
		CreatedAt: test.Time("2004-06-30T18:30:00Z"),
		ActiveAt:  test.Time("2015-06-30T18:30:00Z"),
		Type:      model.TypeNoExpiry,
	})

	resolver := ResolverForDevice{DB: db.DB}
	devices, err := resolver.Devices(fake.User(1))

	require.Nil(t, err)
	expected := []*gqlmodel.Device{
		{
			ID:        1,
			Name:      "Android",
			CreatedAt: test.ModelTimeUTC("2009-06-30T18:30:00Z"),
			ActiveAt:  test.ModelTimeUTC("2018-06-30T18:30:00Z"),
			Type:      gqlmodel.DeviceTypeNoExpiry,
		},
	}
	require.Equal(t, expected, devices)
}