belongs_to_test.go

package pop

import (
	"context"
	"testing"

	"github.com/stretchr/testify/require"
)

func Test_BelongsTo(t *testing.T) {
	if PDB == nil {
		t.Skip("skipping integration tests")
	}
	r := require.New(t)

	q := PDB.BelongsTo(&User{ID: 1})

	m := NewModel(new(Enemy), context.Background())

	sql, _ := q.ToSQL(m)
	r.Equal(ts("SELECT enemies.A FROM enemies AS enemies WHERE user_id = ?"), sql)
}

func Test_BelongsToAs(t *testing.T) {
	if PDB == nil {
		t.Skip("skipping integration tests")
	}
	r := require.New(t)

	q := PDB.BelongsToAs(&User{ID: 1}, "u_id")

	m := NewModel(new(Enemy), context.Background())

	sql, _ := q.ToSQL(m)
	r.Equal(ts("SELECT enemies.A FROM enemies AS enemies WHERE u_id = ?"), sql)
}

func Test_BelongsToThrough(t *testing.T) {
	if PDB == nil {
		t.Skip("skipping integration tests")
	}
	r := require.New(t)

	q := PDB.BelongsToThrough(&User{ID: 1}, &Friend{})
	qs := "SELECT enemies.A FROM enemies AS enemies, good_friends AS good_friends WHERE good_friends.user_id = ? AND enemies.id = good_friends.enemy_id"

	m := NewModel(new(Enemy), context.Background())
	sql, _ := q.ToSQL(m)
	r.Equal(ts(qs), sql)
}