import (
"github.com/cloudfoundry-incubator/notifications/db"
"github.com/cloudfoundry-incubator/notifications/testing/helpers"
"github.com/cloudfoundry-incubator/notifications/v1/models"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("UnsubscribesRepo", func() {
var repo models.UnsubscribesRepo
var conn *db.Connection
BeforeEach(func() {
repo = models.NewUnsubscribesRepo()
database := db.NewDatabase(sqlDB, db.Config{})
helpers.TruncateTables(database)
conn = database.Connection().(*db.Connection)
})
Describe("Get/Set", func() {
It("returns false for unsubscribes that have not been set", func() {
isUnsubscribed, err := repo.Get(conn, "user-id", "client-id", "kind-id")
Expect(err).NotTo(HaveOccurred())
Expect(isUnsubscribed).To(BeFalse())
})
It("returns true for unsubscribes that have been set", func() {
err := repo.Set(conn, "user-id", "client-id", "kind-id", true)
func (m *Mother) UnsubscribesRepo() v1models.UnsubscribesRepo {
return v1models.NewUnsubscribesRepo()
}