Add test data generator

Author: antonmedv

test/mock/mock.go

@@ -35,6 +35,7 @@ type Env struct {
 	ArrayOfFoo         []*Foo
 	MapOfFoo           map[string]Foo
 	MapOfAny           map[string]interface{}
+	MapIntAny          map[int]string
 	FuncParam          func(_ bool, _ int, _ string) bool
 	FuncParamAny       func(_ interface{}) bool
 	FuncTooManyReturns func() (int, int, error)

test/playground/data.go

@@ -0,0 +1,119 @@
+package playground
+
+import (
+	"fmt"
+	"time"
+)
+
+func ExampleData() interface{} {
+	profileJohn := UserProfile{
+		Birthday:  time.Date(1990, 1, 1, 0, 0, 0, 0, time.UTC),
+		Biography: "A passionate writer about Go.",
+		Website:   "https://johndoe.com",
+	}
+
+	profileJane := UserProfile{
+		Birthday:  time.Date(1985, 2, 15, 0, 0, 0, 0, time.UTC),
+		Biography: "A web developer and writer.",
+		Website:   "https://jane.com",
+	}
+
+	authorJohn := Author{
+		ID:        1,
+		FirstName: "John",
+		LastName:  "Doe",
+		Email:     "[email protected]",
+		Profile:   profileJohn,
+	}
+
+	authorJane := Author{
+		ID:        2,
+		FirstName: "Jane",
+		LastName:  "Smith",
+		Email:     "[email protected]",
+		Profile:   profileJane,
+	}
+
+	posts := []Post{
+		{
+			ID:          1,
+			Title:       "Understanding Golang",
+			Content:     "Go is an open-source programming language...",
+			PublishDate: time.Now().AddDate(0, -1, 0),
+			Author:      authorJohn,
+			Comments:    generateComments(3),
+			Tags:        []string{"Go", "Programming"},
+			Likes:       100,
+		},
+		{
+			ID:          2,
+			Title:       "Exploring Python",
+			Content:     "Python is versatile...",
+			PublishDate: time.Now().AddDate(0, -2, 0),
+			Author:      authorJane,
+			Comments:    generateComments(4),
+			Tags:        []string{"Python", "Development"},
+			Likes:       150,
+		},
+		{
+			ID:          3,
+			Title:       "Web Development Basics",
+			Content:     "The world of web development...",
+			PublishDate: time.Now().AddDate(0, -3, 0),
+			Author:      authorJane,
+			Comments:    generateComments(5),
+			Tags:        []string{"Web", "HTML", "CSS"},
+			Likes:       125,
+		},
+		{
+			ID:          4,
+			Title:       "Machine Learning in a Nutshell",
+			Content:     "ML is revolutionizing industries...",
+			PublishDate: time.Now().AddDate(0, -5, 0),
+			Author:      authorJohn,
+			Comments:    generateComments(6),
+			Tags:        []string{"ML", "AI"},
+			Likes:       200,
+		},
+		{
+			ID:          5,
+			Title:       "JavaScript: The Good Parts",
+			Content:     "JavaScript powers the web...",
+			PublishDate: time.Now().AddDate(0, -4, 0),
+			Author:      authorJane,
+			Comments:    generateComments(3),
+			Tags:        []string{"JavaScript", "Web"},
+			Likes:       170,
+		},
+	}
+
+	blog := Blog{
+		Posts:      make([]Post, len(posts)),
+		Authors:    map[int]Author{authorJohn.ID: authorJohn, authorJane.ID: authorJane},
+		TotalViews: 10000,
+		TotalPosts: len(posts),
+		TotalLikes: 0,
+	}
+
+	for i, post := range posts {
+		blog.Posts[i] = post
+		blog.TotalLikes += post.Likes
+	}
+
+	return blog
+}
+
+func generateComments(count int) []Comment {
+	var comments []Comment
+	for i := 1; i <= count; i++ {
+		comment := Comment{
+			ID:          i,
+			AuthorName:  fmt.Sprintf("Commenter %d", i),
+			Content:     fmt.Sprintf("This is comment %d!", i),
+			CommentDate: time.Now().AddDate(0, 0, -i),
+			Upvotes:     i * 5,
+		}
+		comments = append(comments, comment)
+	}
+	return comments
+}

test/playground/env.go

@@ -10,6 +10,10 @@ type UserProfile struct {
 	Website   string
 }
 
+func (u UserProfile) Age() int {
+	return time.Now().Year() - u.Birthday.Year()
+}
+
 type Author struct {
 	ID        int
 	FirstName string
@@ -18,6 +22,14 @@ type Author struct {
 	Profile   UserProfile
 }
 
+func (a Author) FullName() string {
+	return a.FirstName + " " + a.LastName
+}
+
+func (a Author) IsAdmin() bool {
+	return a.ID == 1
+}
+
 type Post struct {
 	ID          int
 	Title       string
@@ -29,6 +41,35 @@ type Post struct {
 	Likes       int
 }
 
+func (p Post) Published() bool {
+	return !p.PublishDate.IsZero()
+}
+
+func (p Post) After(date time.Time) bool {
+	return p.PublishDate.After(date)
+}
+
+func (p Post) Before(date time.Time) bool {
+	return p.PublishDate.Before(date)
+}
+
+func (p Post) Compare(other Post) int {
+	if p.PublishDate.Before(other.PublishDate) {
+		return -1
+	} else if p.PublishDate.After(other.PublishDate) {
+		return 1
+	}
+	return 0
+}
+
+func (p Post) Equal(other Post) bool {
+	return p.Compare(other) == 0
+}
+
+func (p Post) IsZero() bool {
+	return p.ID == 0 && p.Title == "" && p.Content == "" && p.PublishDate.IsZero()
+}
+
 type Comment struct {
 	ID          int
 	AuthorName  string
@@ -37,10 +78,94 @@ type Comment struct {
 	Upvotes     int
 }
 
+func (c Comment) Upvoted() bool {
+	return c.Upvotes > 0
+}
+
+func (c Comment) AuthorEmail() string {
+	return c.AuthorName + "@example.com"
+}
+
 type Blog struct {
 	Posts      []Post
 	Authors    map[int]Author
 	TotalViews int
 	TotalPosts int
 	TotalLikes int
 }
+
+func (b Blog) RecentPosts() []Post {
+	var posts []Post
+	for _, post := range b.Posts {
+		if post.Published() {
+			posts = append(posts, post)
+		}
+	}
+	return posts
+}
+
+func (b Blog) PopularPosts() []Post {
+	var posts []Post
+	for _, post := range b.Posts {
+		if post.Likes > 150 {
+			posts = append(posts, post)
+		}
+	}
+	return posts
+}
+
+func (b Blog) TotalUpvotes() int {
+	var upvotes int
+	for _, post := range b.Posts {
+		for _, comment := range post.Comments {
+			upvotes += comment.Upvotes
+		}
+	}
+	return upvotes
+}
+
+func (b Blog) TotalComments() int {
+	var comments int
+	for _, post := range b.Posts {
+		comments += len(post.Comments)
+	}
+	return comments
+}
+
+func (Blog) Add(a, b float64) float64 {
+	return a + b
+}
+
+func (Blog) Sub(a, b float64) float64 {
+	return a - b
+}
+
+func (Blog) Title(post Post) string {
+	return post.Title
+}
+
+func (Blog) HasTag(post Post, tag string) bool {
+	for _, t := range post.Tags {
+		if t == tag {
+			return true
+		}
+	}
+	return false
+}
+
+func (Blog) IsAdmin(author Author) bool {
+	return author.IsAdmin()
+}
+
+func (Blog) IsZero(post Post) bool {
+	return post.IsZero()
+}
+
+func (Blog) WithID(posts []Post, id int) Post {
+	for _, post := range posts {
+		if post.ID == id {
+			return post
+		}
+	}
+	return Post{}
+}