Lab6 - Puppy Storer - alfredxiao

06_puppy/alfredxiao/main.go

@@ -0,0 +1,20 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"os"
+)
+
+var out io.Writer = os.Stdout
+
+func main() {
+	store1 := NewSyncStore()
+	_ = store1.CreatePuppy(Puppy{
+		ID:     "1",
+		Colour: "Red",
+	})
+
+	puppy, _ := store1.ReadPuppy("1")
+	fmt.Fprint(out, puppy.Colour)
+}

06_puppy/alfredxiao/main_test.go

@@ -0,0 +1,16 @@
+package main
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMainOutput(t *testing.T) {
+	var buf bytes.Buffer
+	out = &buf
+	main()
+
+	assert.Equal(t, "Red", buf.String())
+}

06_puppy/alfredxiao/mapstore.go

@@ -0,0 +1,52 @@
+package main
+
+import (
+	"fmt"
+)
+
+type mapStore struct {
+	data map[string]Puppy
+}
+
+func (s *mapStore) CreatePuppy(p Puppy) error {
+	_, ok := s.data[p.ID]
+	if ok {
+		return fmt.Errorf("puppy with ID[%s] already exists", p.ID)
+	}
+
+	s.data[p.ID] = p
+	return nil
+}
+
+func (s *mapStore) ReadPuppy(id string) (Puppy, error) {
+	p, ok := s.data[id]
+	if !ok {
+		return Puppy{}, fmt.Errorf("puppy with ID[%s] does not exists", p.ID)
+	}
+
+	return p, nil
+}
+
+func (s *mapStore) UpdatePuppy(p Puppy) error {
+	_, ok := s.data[p.ID]
+	if !ok {
+		return fmt.Errorf("puppy with ID[%s] does not exists", p.ID)
+	}
+	s.data[p.ID] = p
+	return nil
+}
+
+func (s *mapStore) DeletePuppy(id string) (bool, error) {
+	p, ok := s.data[id]
+	if !ok {
+		return false, fmt.Errorf("puppy with ID[%s] does not exists", p.ID)
+	}
+	delete(s.data, id)
+	return true, nil
+}
+
+func NewMapStore() Storer {
+	return &mapStore{
+		data: make(map[string]Puppy),
+	}
+}

06_puppy/alfredxiao/store_test.go

@@ -0,0 +1,82 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+)
+
+type storerSuite struct {
+	suite.Suite
+	store        Storer
+	storeFactory func() Storer
+}
+
+func (s *storerSuite) SetupTest() {
+	s.store = s.storeFactory()
+}
+
+func (s *storerSuite) TestCreatePuppyHappyCase() {
+	err := s.store.CreatePuppy(Puppy{ID: "1", Colour: "Black"})
+	s.NoError(err, "Happy case puppy creation")
+
+	p, _ := s.store.ReadPuppy("1")
+	s.Equal("Black", p.Colour)
+}
+
+func (s *storerSuite) TestCreatePuppyAlreadyExists() {
+	_ = s.store.CreatePuppy(Puppy{ID: "2"})
+	err := s.store.CreatePuppy(Puppy{ID: "2", Colour: "Red"})
+	s.Error(err, "Puppy creation fails if ID already exists")
+}
+
+func (s *storerSuite) TestReadPuppyHappyCase() {
+	_ = s.store.CreatePuppy(Puppy{ID: "3", Colour: "Blue"})
+	p, err := s.store.ReadPuppy("3")
+	require.NoError(s.T(), err)
+	s.Equal("Blue", p.Colour)
+}
+
+func (s *storerSuite) TestReadPuppyNonExisting() {
+	_, err := s.store.ReadPuppy("4")
+	s.Error(err)
+}
+
+func (s *storerSuite) TestUpdatePuppyHappyCase() {
+	_ = s.store.CreatePuppy(Puppy{ID: "5", Colour: "Brown"})
+	err := s.store.UpdatePuppy(Puppy{ID: "5", Colour: "Green"})
+	require.NoError(s.T(), err)
+	p, _ := s.store.ReadPuppy("5")
+	s.Equal("Green", p.Colour)
+}
+
+func (s *storerSuite) TestUpdatePuppyNonExisting() {
+	err := s.store.UpdatePuppy(Puppy{ID: "6"})
+	s.Error(err)
+}
+
+func (s *storerSuite) TestDeletePuppyHappyCase() {
+	_ = s.store.CreatePuppy(Puppy{ID: "7", Colour: "Brown"})
+	deleted, err := s.store.DeletePuppy("7")
+	require.NoError(s.T(), err)
+	require.Equal(s.T(), true, deleted)
+
+	_, err = s.store.ReadPuppy("7")
+	s.Error(err, "Puppy gone after deletion")
+}
+
+func (s *storerSuite) TestDeletePuppyNonExisting() {
+	deleted, err := s.store.DeletePuppy("8")
+	require.Error(s.T(), err)
+	s.Equal(false, deleted)
+}
+
+func TestStorers(t *testing.T) {
+	suite.Run(t, &storerSuite{
+		storeFactory: NewMapStore,
+	})
+	suite.Run(t, &storerSuite{
+		storeFactory: NewSyncStore,
+	})
+}

06_puppy/alfredxiao/syncstore.go

@@ -0,0 +1,51 @@
+package main
+
+import (
+	"fmt"
+	"sync"
+)
+
+type syncStore struct {
+	data sync.Map
+}
+
+func (s *syncStore) CreatePuppy(p Puppy) error {
+	_, ok := s.data.Load(p.ID)
+	if ok {
+		return fmt.Errorf("puppy with ID[%s] already exists", p.ID)
+	}
+
+	s.data.Store(p.ID, p)
+	return nil
+}
+
+func (s *syncStore) ReadPuppy(id string) (Puppy, error) {
+	p, ok := s.data.Load(id)
+	if !ok {
+		return Puppy{}, fmt.Errorf("puppy with ID[%s] does not exists", id)
+	}
+
+	return p.(Puppy), nil
+}
+
+func (s *syncStore) UpdatePuppy(p Puppy) error {
+	_, ok := s.data.Load(p.ID)
+	if !ok {
+		return fmt.Errorf("puppy with ID[%s] does not exists", p.ID)
+	}
+	s.data.Store(p.ID, p)
+	return nil
+}
+
+func (s *syncStore) DeletePuppy(id string) (bool, error) {
+	_, ok := s.data.Load(id)
+	if !ok {
+		return false, fmt.Errorf("puppy with ID[%s] does not exists", id)
+	}
+	s.data.Delete(id)
+	return true, nil
+}
+
+func NewSyncStore() Storer {
+	return &syncStore{}
+}

06_puppy/alfredxiao/types.go

@@ -0,0 +1,15 @@
+package main
+
+type Puppy struct {
+	ID     string
+	Breed  string
+	Colour string
+	Value  string
+}
+
+type Storer interface {
+	CreatePuppy(p Puppy) error
+	ReadPuppy(ID string) (Puppy, error)
+	UpdatePuppy(p Puppy) error
+	DeletePuppy(ID string) (bool, error)
+}