status-go/protocol/communities/community_categories_test.go

package communities

import (
	"github.com/status-im/status-go/protocol/protobuf"
)

func (s *CommunitySuite) TestCreateCategory() {
	newCategoryID := "new-category-id"
	newCategoryName := "new-category-name"

	org := s.buildCommunity(&s.identity.PublicKey)
	org.config.PrivateKey = nil
	org.config.ID = nil

	_, err := org.CreateCategory(newCategoryID, newCategoryName, []string{})
	s.Require().Equal(ErrNotAuthorized, err)

	org.config.PrivateKey = s.identity
	org.config.ID = &s.identity.PublicKey

	changes, err := org.CreateCategory(newCategoryID, newCategoryName, []string{})

	description := org.config.CommunityDescription

	s.Require().NoError(err)
	s.Require().NotNil(description.Categories)
	s.Require().NotNil(description.Categories[newCategoryID])
	s.Require().Equal(newCategoryName, description.Categories[newCategoryID].Name)
	s.Require().Equal(newCategoryID, description.Categories[newCategoryID].CategoryId)
	s.Require().Equal(int32(len(description.Categories)-1), description.Categories[newCategoryID].Position)
	s.Require().NotNil(changes)
	s.Require().NotNil(changes.CategoriesAdded[newCategoryID])
	s.Require().Equal(description.Categories[newCategoryID], changes.CategoriesAdded[newCategoryID])
	s.Require().Nil(changes.CategoriesModified[newCategoryID])

	_, err = org.CreateCategory(newCategoryID, newCategoryName, []string{})
	s.Require().Equal(ErrCategoryAlreadyExists, err)

	newCategoryID2 := "new-category-id2"
	newCategoryName2 := "new-category-name2"

	changes, err = org.CreateCategory(newCategoryID2, newCategoryName2, []string{})
	s.Require().NoError(err)
	s.Require().Equal(int32(len(description.Categories)-1), description.Categories[newCategoryID2].Position)
	s.Require().NotNil(changes.CategoriesAdded[newCategoryID2])
	s.Require().Nil(changes.CategoriesModified[newCategoryID2])

	newCategoryID3 := "new-category-id3"
	newCategoryName3 := "new-category-name3"
	_, err = org.CreateCategory(newCategoryID3, newCategoryName3, []string{"some-chat-id"})
	s.Require().Equal(ErrChatNotFound, err)

	newChatID := "new-chat-id"
	identity := &protobuf.ChatIdentity{
		DisplayName: "new-chat-display-name",
		Description: "new-chat-description",
	}
	permissions := &protobuf.CommunityPermissions{
		Access: protobuf.CommunityPermissions_AUTO_ACCEPT,
	}

	_, err = org.CreateChat(newChatID, &protobuf.CommunityChat{
		Identity:    identity,
		Permissions: permissions,
	})
	s.Require().NoError(err)

	changes, err = org.CreateCategory(newCategoryID3, newCategoryName3, []string{newChatID})
	s.Require().NoError(err)
	s.Require().NotNil(changes.ChatsModified[newChatID])
	s.Require().Equal(newCategoryID3, changes.ChatsModified[newChatID].CategoryModified)

	newCategoryID4 := "new-category-id4"
	newCategoryName4 := "new-category-name4"

	_, err = org.CreateCategory(newCategoryID4, newCategoryName4, []string{newChatID})
	s.Require().Equal(ErrChatAlreadyAssigned, err)
}

func (s *CommunitySuite) TestEditCategory() {
	newCategoryID := "new-category-id"
	newCategoryName := "new-category-name"
	editedCategoryName := "edited-category-name"

	org := s.buildCommunity(&s.identity.PublicKey)
	org.config.PrivateKey = s.identity
	_, err := org.CreateCategory(newCategoryID, newCategoryName, []string{testChatID1})
	s.Require().NoError(err)
	org.config.PrivateKey = nil
	org.config.ID = nil

	_, err = org.EditCategory(newCategoryID, editedCategoryName, []string{testChatID1})
	s.Require().Equal(ErrNotAuthorized, err)

	org.config.PrivateKey = s.identity
	org.config.ID = &s.identity.PublicKey

	_, err = org.EditCategory("some-random-category", editedCategoryName, []string{testChatID1})
	s.Require().Equal(ErrCategoryNotFound, err)

	changes, err := org.EditCategory(newCategoryID, editedCategoryName, []string{testChatID1})

	description := org.config.CommunityDescription

	s.Require().NoError(err)
	s.Require().Equal(editedCategoryName, description.Categories[newCategoryID].Name)
	s.Require().NotNil(changes)
	s.Require().NotNil(changes.CategoriesModified[newCategoryID])
	s.Require().Equal(description.Categories[newCategoryID], changes.CategoriesModified[newCategoryID])
	s.Require().Nil(changes.CategoriesAdded[newCategoryID])

	_, err = org.EditCategory(newCategoryID, editedCategoryName, []string{"some-random-chat-id"})
	s.Require().Equal(ErrChatNotFound, err)

	_, err = org.EditCategory(testCategoryID1, testCategoryName1, []string{testChatID1})
	s.Require().Equal(ErrChatAlreadyAssigned, err)

	// Edit by removing the chats

	identity1 := &protobuf.ChatIdentity{
		DisplayName: "new-chat-display-name",
		Description: "new-chat-description",
	}
	permissions := &protobuf.CommunityPermissions{
		Access: protobuf.CommunityPermissions_AUTO_ACCEPT,
	}

	testChatID2 := "test-chat-id-2"
	testChatID3 := "test-chat-id-3"

	_, err = org.CreateChat(testChatID2, &protobuf.CommunityChat{
		Identity:    identity1,
		Permissions: permissions,
	})
	s.Require().NoError(err)
	identity2 := &protobuf.ChatIdentity{
		DisplayName: "identity-2",
		Description: "new-chat-description",
	}

	_, err = org.CreateChat(testChatID3, &protobuf.CommunityChat{
		Identity:    identity2,
		Permissions: permissions,
	})
	s.Require().NoError(err)

	_, err = org.EditCategory(newCategoryID, editedCategoryName, []string{testChatID1, testChatID2, testChatID3})
	s.Require().NoError(err)

	s.Require().Equal(newCategoryID, description.Chats[testChatID1].CategoryId)
	s.Require().Equal(newCategoryID, description.Chats[testChatID2].CategoryId)
	s.Require().Equal(newCategoryID, description.Chats[testChatID3].CategoryId)

	s.Require().Equal(int32(0), description.Chats[testChatID1].Position)
	s.Require().Equal(int32(1), description.Chats[testChatID2].Position)
	s.Require().Equal(int32(2), description.Chats[testChatID3].Position)

	_, err = org.EditCategory(newCategoryID, editedCategoryName, []string{testChatID1, testChatID3})
	s.Require().NoError(err)
	s.Require().Equal("", description.Chats[testChatID2].CategoryId)
	s.Require().Equal(int32(0), description.Chats[testChatID1].Position)
	s.Require().Equal(int32(1), description.Chats[testChatID3].Position)

	_, err = org.EditCategory(newCategoryID, editedCategoryName, []string{testChatID3})
	s.Require().NoError(err)
	s.Require().Equal("", description.Chats[testChatID1].CategoryId)
	s.Require().Equal(int32(0), description.Chats[testChatID3].Position)
}

func (s *CommunitySuite) TestDeleteCategory() {
	org := s.buildCommunity(&s.identity.PublicKey)
	org.config.PrivateKey = s.identity
	permissions := &protobuf.CommunityPermissions{
		Access: protobuf.CommunityPermissions_AUTO_ACCEPT,
	}

	testChatID2 := "test-chat-id-2"
	testChatID3 := "test-chat-id-3"
	newCategoryID := "new-category-id"
	newCategoryName := "new-category-name"

	identity1 := &protobuf.ChatIdentity{
		DisplayName: "display-name-2",
		Description: "new-chat-description",
	}

	_, err := org.CreateChat(testChatID2, &protobuf.CommunityChat{
		Identity:    identity1,
		Permissions: permissions,
	})
	s.Require().NoError(err)

	identity2 := &protobuf.ChatIdentity{
		DisplayName: "display-name-3",
		Description: "new-chat-description",
	}

	_, err = org.CreateChat(testChatID3, &protobuf.CommunityChat{
		Identity:    identity2,
		Permissions: permissions,
	})
	s.Require().NoError(err)

	_, err = org.CreateCategory(newCategoryID, newCategoryName, []string{})
	s.Require().NoError(err)

	description := org.config.CommunityDescription

	_, err = org.EditCategory(newCategoryID, newCategoryName, []string{testChatID2, testChatID1})
	s.Require().NoError(err)
	s.Require().Equal(newCategoryID, description.Chats[testChatID1].CategoryId)
	s.Require().Equal(newCategoryID, description.Chats[testChatID2].CategoryId)

	s.Require().Equal(int32(0), description.Chats[testChatID3].Position)
	s.Require().Equal(int32(0), description.Chats[testChatID2].Position)
	s.Require().Equal(int32(1), description.Chats[testChatID1].Position)

	org.config.PrivateKey = nil
	org.config.ID = nil
	_, err = org.DeleteCategory(testCategoryID1)
	s.Require().Equal(ErrNotAuthorized, err)

	org.config.PrivateKey = s.identity
	org.config.ID = &s.identity.PublicKey
	_, err = org.DeleteCategory("some-category-id")
	s.Require().Equal(ErrCategoryNotFound, err)

	changes, err := org.DeleteCategory(newCategoryID)
	s.Require().NoError(err)
	s.Require().NotNil(changes)

	s.Require().Equal("", description.Chats[testChatID1].CategoryId)
	s.Require().Equal("", description.Chats[testChatID2].CategoryId)
	s.Require().Equal("", description.Chats[testChatID3].CategoryId)
}

func (s *CommunitySuite) TestDeleteChatOrder() {
	org := s.buildCommunity(&s.identity.PublicKey)
	org.config.PrivateKey = s.identity
	permissions := &protobuf.CommunityPermissions{
		Access: protobuf.CommunityPermissions_AUTO_ACCEPT,
	}

	testChatID2 := "test-chat-id-2"
	testChatID3 := "test-chat-id-3"
	newCategoryID := "new-category-id"
	newCategoryName := "new-category-name"

	identity1 := &protobuf.ChatIdentity{
		DisplayName: "identity-1",
		Description: "new-chat-description",
	}

	_, err := org.CreateChat(testChatID2, &protobuf.CommunityChat{
		Identity:    identity1,
		Permissions: permissions,
	})
	s.Require().NoError(err)
	identity2 := &protobuf.ChatIdentity{
		DisplayName: "identity-2",
		Description: "new-chat-description",
	}

	_, err = org.CreateChat(testChatID3, &protobuf.CommunityChat{
		Identity:    identity2,
		Permissions: permissions,
	})
	s.Require().NoError(err)

	_, err = org.CreateCategory(newCategoryID, newCategoryName, []string{testChatID1, testChatID2, testChatID3})
	s.Require().NoError(err)

	changes, err := org.DeleteChat(testChatID2)
	s.Require().NoError(err)
	s.Require().Equal(int32(0), org.Chats()[testChatID1].Position)
	s.Require().Equal(int32(1), org.Chats()[testChatID3].Position)
	s.Require().Len(changes.ChatsRemoved, 1)

	changes, err = org.DeleteChat(testChatID1)
	s.Require().NoError(err)
	s.Require().Equal(int32(0), org.Chats()[testChatID3].Position)
	s.Require().Len(changes.ChatsRemoved, 1)

	_, err = org.DeleteChat(testChatID3)
	s.Require().NoError(err)
}