async_tasks_test.go

package stream_chat

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func TestClient_DeleteChannels(t *testing.T) {
	c := initClient(t)
	ch := initChannel(t, c)
	ctx := context.Background()
	user := randomUser(t, c)
	msg := &Message{Text: "test message"}

	_, err := ch.SendMessage(ctx, msg, user.ID, MessageSkipPush)
	require.NoError(t, err, "send message")

	// should fail without CIDs in parameter
	_, err = c.DeleteChannels(ctx, []string{}, true)
	require.Error(t, err)

	resp1, err := c.DeleteChannels(ctx, []string{ch.CID}, true)
	require.NoError(t, err)
	require.NotEmpty(t, resp1.TaskID)

	for i := 0; i < 10; i++ {
		resp2, err := c.GetTask(ctx, resp1.TaskID)
		require.NoError(t, err)
		require.Equal(t, resp1.TaskID, resp2.TaskID)

		if resp2.Status == TaskStatusCompleted {
			require.Equal(t, map[string]interface{}{"status": "ok"}, resp2.Result[ch.CID])
			return
		}

		time.Sleep(time.Second)
	}
}

func TestClient_DeleteUsers(t *testing.T) {
	c := initClient(t)
	ch := initChannel(t, c)
	ctx := context.Background()
	user := randomUser(t, c)

	msg := &Message{Text: "test message"}

	_, err := ch.SendMessage(ctx, msg, user.ID, MessageSkipPush)
	require.NoError(t, err, "send message")

	// should fail without userIDs in parameter
	_, err = c.DeleteUsers(ctx, []string{}, DeleteUserOptions{
		User:     SoftDelete,
		Messages: HardDelete,
	})
	require.Error(t, err)

	resp1, err := c.DeleteUsers(ctx, []string{user.ID}, DeleteUserOptions{
		User:     SoftDelete,
		Messages: HardDelete,
	})
	require.NoError(t, err)
	require.NotEmpty(t, resp1.TaskID)

	for i := 0; i < 10; i++ {
		resp2, err := c.GetTask(ctx, resp1.TaskID)
		require.NoError(t, err)
		require.Equal(t, resp1.TaskID, resp2.TaskID)

		if resp2.Status == TaskStatusCompleted {
			require.Equal(t, map[string]interface{}{"status": "ok"}, resp2.Result[user.ID])
			return
		}

		time.Sleep(time.Second)
	}

	require.True(t, false, "task did not succeed")
}

func TestClient_ExportChannels(t *testing.T) {
	c := initClient(t)
	ch1 := initChannel(t, c)
	ch2 := initChannel(t, c)
	ctx := context.Background()

	t.Run("Return error if there are 0 channels", func(t *testing.T) {
		_, err := c.ExportChannels(ctx, nil, nil)
		require.Error(t, err)
	})

	t.Run("Return error if exportable channel structs are incorrect", func(t *testing.T) {
		expChannels := []*ExportableChannel{
			{Type: "", ID: ch1.ID},
		}
		_, err := c.ExportChannels(ctx, expChannels, nil)
		require.Error(t, err)
	})

	t.Run("Export channels with no error", func(t *testing.T) {
		expChannels := []*ExportableChannel{
			{Type: ch1.Type, ID: ch1.ID},
			{Type: ch2.Type, ID: ch2.ID},
		}

		resp1, err := c.ExportChannels(ctx, expChannels, nil)
		require.NoError(t, err)
		require.NotEmpty(t, resp1.TaskID)

		for i := 0; i < 10; i++ {
			task, err := c.GetExportChannelsTask(ctx, resp1.TaskID)
			require.NoError(t, err)
			require.Equal(t, resp1.TaskID, task.TaskID)
			require.NotEmpty(t, task.Status)

			if task.Status == TaskStatusCompleted {
				break
			}

			time.Sleep(time.Second)
		}
	})
}

func TestClient_ExportUsers(t *testing.T) {
	c := initClient(t)
	ch1 := initChannel(t, c)
	ctx := context.Background()

	t.Run("Return error if there are 0 user IDs", func(t *testing.T) {
		_, err := c.ExportUsers(ctx, nil)
		require.Error(t, err)
	})

	t.Run("Export users with no error", func(t *testing.T) {
		resp, err := c.ExportUsers(ctx, []string{ch1.CreatedBy.ID})
		require.NoError(t, err)
		require.NotEmpty(t, resp.TaskID)

		for i := 0; i < 10; i++ {
			task, err := c.GetTask(ctx, resp.TaskID)
			require.NoError(t, err)
			require.Equal(t, resp.TaskID, task.TaskID)
			require.NotEmpty(t, task.Status)

			if task.Status == TaskStatusCompleted {
				require.Contains(t, task.Result["url"], "/exports/users/")
				break
			}

			time.Sleep(time.Second)
		}
	})
}