package github

import (
	"testing"

	. "github.com/onsi/gomega"
)

func TestPRLocation(t *testing.T) {
	g := NewGomegaWithT(t)

	owner := "tidbcloud"
	repo := "devops"
	pr := int64(173)

	loc := NewPRLocation(owner, repo, pr)
	g.Expect(string(loc)).To(Equal("tidbcloud/devops#173"))

	parsedOwner, parsedRepo, parsedPRNr := loc.Parse()
	g.Expect(parsedOwner).To(Equal(owner))
	g.Expect(parsedRepo).To(Equal(repo))
	g.Expect(parsedPRNr).To(Equal(pr))
}

func TestRepoLocation(t *testing.T) {
	g := NewGomegaWithT(t)

	owner := "tidbcloud"
	repo := "devops"

	loc := NewRepoLocation(owner, repo)
	g.Expect(string(loc)).To(Equal("tidbcloud/devops"))

	parsedOwner, parsedRepo := loc.Parse()
	g.Expect(parsedOwner).To(Equal(owner))
	g.Expect(parsedRepo).To(Equal(repo))
}