github.com/k0marov/go-socnet@v0.0.0-20220715154813-90d07867c782/features/posts/integration_test/posts_test.go

package posts_test

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/k0marov/go-socnet/core/general/core_values"
	"github.com/k0marov/go-socnet/core/general/static_store"
	helpers "github.com/k0marov/go-socnet/core/helpers/http_test_helpers"
	. "github.com/k0marov/go-socnet/core/helpers/test_helpers"
	"log"
	"mime/multipart"
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"strconv"
	"testing"

	"github.com/go-chi/chi/v5"
	"github.com/k0marov/go-socnet/features/posts"
	"github.com/k0marov/go-socnet/features/posts/delivery/http/responses"
	"github.com/k0marov/go-socnet/features/posts/domain/values"
	post_storage "github.com/k0marov/go-socnet/features/posts/store/file_storage"
	"github.com/k0marov/go-socnet/features/profiles"
	profile_entities "github.com/k0marov/go-socnet/features/profiles/domain/entities"
	profile_models "github.com/k0marov/go-socnet/features/profiles/domain/models"
	auth "github.com/k0marov/golang-auth"
	_ "github.com/mattn/go-sqlite3"
)

func TestPosts(t *testing.T) {
	// working directory
	os.Mkdir("tmp_test", 0777)
	os.Chdir("tmp_test")
	defer func() {
		os.Chdir("..")
		os.RemoveAll("tmp_test")
	}()

	// db
	sql := OpenSqliteDB(t)

	r := chi.NewRouter()
	// profiles
	r.Route("/profiles", profiles.NewProfilesRouterImpl(sql))
	fakeRegisterProfile := profiles.NewRegisterCallback(sql)
	// posts
	r.Route("/posts", posts.NewPostsRouterImpl(sql, profiles.NewProfileGetterImpl(sql)))

	// helpers
	createPost := func(t testing.TB, author auth.User, images [][]byte, text string) {
		t.Helper()
		body := bytes.NewBuffer(nil)
		writer := multipart.NewWriter(body)

		writer.WriteField("text", text)
		for i, image := range images {
			fw, _ := writer.CreateFormFile(fmt.Sprintf("image_%d", i+1), RandomString())
			fw.Write(image)
		}
		writer.Close()
		request := helpers.AddAuthDataToRequest(httptest.NewRequest(http.MethodPost, "/posts", body), author)
		request.Header.Set("Content-Type", writer.FormDataContentType())

		response := httptest.NewRecorder()
		r.ServeHTTP(response, request)
		AssertStatusCode(t, response, http.StatusOK)
	}
	getPosts := func(t testing.TB, author core_values.UserId, caller auth.User) []responses.PostResponse {
		t.Helper()
		request := helpers.AddAuthDataToRequest(httptest.NewRequest(http.MethodGet, "/posts/?profile_id="+author, nil), caller)
		response := httptest.NewRecorder()
		r.ServeHTTP(response, request)
		AssertStatusCode(t, response, http.StatusOK)
		var posts responses.PostsResponse
		json.NewDecoder(response.Body).Decode(&posts)
		return posts.Posts
	}
	assertImageCreated := func(t testing.TB, post responses.PostResponse, postImage responses.PostImageResponse, wantImage []byte) {
		t.Helper()
		path := filepath.Join(static_store.StaticDir, post_storage.GetPostDir(post.Id, post.Author.Id), post_storage.ImagePrefix+strconv.Itoa(postImage.Index))
		got := readFile(t, path)
		Assert(t, got, wantImage, "the stored image data")
	}
	deletePost := func(t testing.TB, postId values.PostId, author auth.User) {
		t.Helper()
		request := helpers.AddAuthDataToRequest(httptest.NewRequest(http.MethodDelete, "/posts/"+postId, nil), author)
		response := httptest.NewRecorder()
		r.ServeHTTP(response, request)
		AssertStatusCode(t, response, http.StatusOK)
	}
	assertPostFilesDeleted := func(t testing.TB, postId values.PostId, author core_values.UserId) {
		t.Helper()
		postPath := filepath.Join(static_store.StaticDir, post_storage.GetPostDir(postId, author))
		_, err := os.ReadDir(postPath)
		AssertSomeError(t, err)
	}
	toggleLike := func(t testing.TB, postId values.PostId, caller auth.User) {
		t.Helper()
		request := helpers.AddAuthDataToRequest(httptest.NewRequest(http.MethodPost, "/posts/"+postId+"/toggle-like", nil), caller)
		response := httptest.NewRecorder()
		r.ServeHTTP(response, request)
		AssertStatusCode(t, response, http.StatusOK)
	}

	registerProfile := func(user auth.User) profile_entities.Profile {
		fakeRegisterProfile(user)
		return profile_entities.Profile{
			ProfileModel: profile_models.ProfileModel{
				Id:       user.Id,
				Username: user.Username,
			},
		}
	}

	// create 2 profiles
	user1 := RandomAuthUser()
	user2 := RandomAuthUser()
	registerProfile(user1)
	registerProfile(user2)

	t.Run("creating, reading and deleting posts", func(t *testing.T) {

		// create post (without images) belonging to 2-nd profile
		text1 := "Hello, World!"
		createPost(t, user2, [][]byte{}, text1)

		// create post (with images) belonging to 2-nd profile
		text2 := "Hello, World with Images!"
		image1 := readFixture(t, "test_image.jpg")
		image2 := readFixture(t, "test_image.jpg")
		createPost(t, user2, [][]byte{image1, image2}, text2)

		// assert they were created
		posts := getPosts(t, user2.Id, user2)

		AssertFatal(t, len(posts), 2, "number of created posts")

		log.Print(fmt.Sprintf("first:  %+v, \nsecond: %+v", posts[0], posts[1]))

		Assert(t, posts[0].Text, text1, "the first post's text")
		Assert(t, posts[0].Author.Id, user2.Id, "first post's author")
		AssertFatal(t, len(posts[0].Images), 0, "number of images in first post")

		Assert(t, posts[1].Text, text2, "the second post's text")
		Assert(t, posts[1].Author.Id, user2.Id, "second posts's author")
		AssertFatal(t, len(posts[1].Images), 2, "number of images in second post")
		assertImageCreated(t, posts[1], posts[1].Images[0], image1)
		assertImageCreated(t, posts[1], posts[1].Images[1], image2)
		// delete them
		deletePost(t, posts[0].Id, user2)
		deletePost(t, posts[1].Id, user2)
		// assert they were deleted
		nowPosts := getPosts(t, user2.Id, user2)
		Assert(t, len(nowPosts), 0, "number of posts after deletion")
		assertPostFilesDeleted(t, posts[0].Id, user2.Id)
		assertPostFilesDeleted(t, posts[1].Id, user2.Id)
	})
	t.Run("liking posts", func(t *testing.T) {
		// create a post belonging to 1-st profile
		createPost(t, user1, [][]byte{}, "")
		posts := getPosts(t, user1.Id, user1)

		// like it from 2-nd profile
		toggleLike(t, posts[0].Id, user2)
		// assert it is liked
		posts = getPosts(t, user1.Id, user2)
		Assert(t, posts[0].IsLiked, true, "post is liked")

		// unlike it from 2-nd profile
		toggleLike(t, posts[0].Id, user2)
		// assert it is not liked
		posts = getPosts(t, user1.Id, user2)
		Assert(t, posts[0].IsLiked, false, "post is not liked")
	})
}

func readFixture(t testing.TB, filename string) []byte {
	t.Helper()
	return readFile(t, filepath.Join("..", "testdata", filename)) // ".." since we change the working directory to tmp_test
}

func readFile(t testing.TB, filepath string) []byte {
	t.Helper()
	data, err := os.ReadFile(filepath)
	if err != nil {
		t.Fatalf("error while reading file %s: %v", filepath, err)
	}
	return data
}