add follower repository
This commit is contained in:
parent
633c0991e9
commit
10ca2bdcd9
|
@ -51,3 +51,9 @@ type InteractionRepository interface {
|
||||||
// ListAllInteractions lists all interactions, sorted by creation date (descending)
|
// ListAllInteractions lists all interactions, sorted by creation date (descending)
|
||||||
ListAllInteractions() ([]model.Interaction, error)
|
ListAllInteractions() ([]model.Interaction, error)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type FollowerRepository interface {
|
||||||
|
Add(follower string) error
|
||||||
|
Remove(follower string) error
|
||||||
|
All() ([]string, error)
|
||||||
|
}
|
||||||
|
|
|
@ -0,0 +1,57 @@
|
||||||
|
package infra
|
||||||
|
|
||||||
|
import (
|
||||||
|
"owl-blogs/app/repository"
|
||||||
|
|
||||||
|
"github.com/jmoiron/sqlx"
|
||||||
|
)
|
||||||
|
|
||||||
|
type sqlFollower struct {
|
||||||
|
Follwer string `db:"follower"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type DefaultFollowerRepo struct {
|
||||||
|
db *sqlx.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewFollowerRepository(db Database) repository.FollowerRepository {
|
||||||
|
sqlxdb := db.Get()
|
||||||
|
|
||||||
|
// Create tables if not exists
|
||||||
|
sqlxdb.MustExec(`
|
||||||
|
CREATE TABLE IF NOT EXISTS followers (
|
||||||
|
follower TEXT PRIMARY KEY
|
||||||
|
);
|
||||||
|
`)
|
||||||
|
|
||||||
|
return &DefaultFollowerRepo{
|
||||||
|
db: sqlxdb,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add implements repository.FollowerRepository.
|
||||||
|
func (d *DefaultFollowerRepo) Add(follower string) error {
|
||||||
|
_, err := d.db.Exec("INSERT INTO followers (follower) VALUES (?) ON CONFLICT DO NOTHING", follower)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove implements repository.FollowerRepository.
|
||||||
|
func (d *DefaultFollowerRepo) Remove(follower string) error {
|
||||||
|
_, err := d.db.Exec("DELETE FROM followers WHERE follower = ?", follower)
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// All implements repository.FollowerRepository.
|
||||||
|
func (d *DefaultFollowerRepo) All() ([]string, error) {
|
||||||
|
var followers []sqlFollower
|
||||||
|
err := d.db.Select(&followers, "SELECT * FROM followers")
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
result := []string{}
|
||||||
|
for _, follower := range followers {
|
||||||
|
result = append(result, follower.Follwer)
|
||||||
|
}
|
||||||
|
return result, nil
|
||||||
|
}
|
|
@ -0,0 +1,91 @@
|
||||||
|
package infra_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"owl-blogs/app/repository"
|
||||||
|
"owl-blogs/infra"
|
||||||
|
"owl-blogs/test"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func setupFollowerRepo() repository.FollowerRepository {
|
||||||
|
db := test.NewMockDb()
|
||||||
|
repo := infra.NewFollowerRepository(db)
|
||||||
|
return repo
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAddFollower(t *testing.T) {
|
||||||
|
repo := setupFollowerRepo()
|
||||||
|
|
||||||
|
err := repo.Add("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err := repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 1)
|
||||||
|
require.Equal(t, followers[0], "foo@example.com")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDoubleAddFollower(t *testing.T) {
|
||||||
|
repo := setupFollowerRepo()
|
||||||
|
|
||||||
|
err := repo.Add("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
err = repo.Add("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err := repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 1)
|
||||||
|
require.Equal(t, followers[0], "foo@example.com")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMultipleAddFollower(t *testing.T) {
|
||||||
|
repo := setupFollowerRepo()
|
||||||
|
|
||||||
|
err := repo.Add("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
err = repo.Add("bar@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
err = repo.Add("baz@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err := repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemoveFollower(t *testing.T) {
|
||||||
|
repo := setupFollowerRepo()
|
||||||
|
|
||||||
|
err := repo.Add("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err := repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 1)
|
||||||
|
|
||||||
|
err = repo.Remove("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err = repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 0)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemoveNonExistingFollower(t *testing.T) {
|
||||||
|
repo := setupFollowerRepo()
|
||||||
|
|
||||||
|
err := repo.Remove("foo@example.com")
|
||||||
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
followers, err := repo.All()
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Len(t, followers, 0)
|
||||||
|
|
||||||
|
}
|
Loading…
Reference in New Issue