influx-cli/internal/backup_restore/manifest_test.go

package backup_restore_test

import (
	"fmt"
	"testing"
	"time"

	"github.com/influxdata/influx-cli/v2/api"
	br "github.com/influxdata/influx-cli/v2/internal/backup_restore"
	"github.com/stretchr/testify/require"
)

func TestConvertBucketManifest(t *testing.T) {
	t.Parallel()

	now := time.Now()

	manifest := api.BucketMetadataManifest{
		OrganizationID:         "123",
		OrganizationName:       "org",
		BucketID:               "456",
		BucketName:             "bucket",
		DefaultRetentionPolicy: "foo",
		RetentionPolicies: []api.RetentionPolicyManifest{
			{
				Name:               "foo",
				ReplicaN:           1,
				Duration:           100,
				ShardGroupDuration: 10,
				ShardGroups: []api.ShardGroupManifest{
					{
						Id:          1,
						StartTime:   now,
						EndTime:     now,
						TruncatedAt: &now,
						Shards: []api.ShardManifest{
							{
								Id:          10,
								ShardOwners: []api.ShardOwner{{NodeID: 1}},
							},
							{
								Id:          20,
								ShardOwners: []api.ShardOwner{{NodeID: 2}, {NodeID: 3}},
							},
						},
					},
					{
						Id:        2,
						StartTime: now,
						EndTime:   now,
						DeletedAt: &now,
						Shards: []api.ShardManifest{
							{
								Id: 30,
							},
						},
					},
				},
				Subscriptions: []api.SubscriptionManifest{},
			},
			{
				Name:               "bar",
				ReplicaN:           3,
				Duration:           9999,
				ShardGroupDuration: 1,
				ShardGroups: []api.ShardGroupManifest{
					{
						Id:        3,
						StartTime: now,
						EndTime:   now,
						Shards:    []api.ShardManifest{},
					},
				},
				Subscriptions: []api.SubscriptionManifest{
					{
						Name:         "test",
						Mode:         "on",
						Destinations: []string{"here", "there", "everywhere"},
					},
					{
						Name:         "test2",
						Mode:         "off",
						Destinations: []string{},
					},
				},
			},
		},
	}

	fakeGetShard := func(id int64) (*br.ManifestFileEntry, error) {
		if id == 20 {
			return nil, nil
		}
		return &br.ManifestFileEntry{
			FileName:    fmt.Sprintf("%d.gz", id),
			Size:        id * 100,
			Compression: br.GzipCompression,
		}, nil
	}

	converted, err := br.ConvertBucketManifest(manifest, fakeGetShard)
	require.NoError(t, err)

	expected := br.ManifestBucketEntry{
		OrganizationID:         "123",
		OrganizationName:       "org",
		BucketID:               "456",
		BucketName:             "bucket",
		DefaultRetentionPolicy: "foo",
		RetentionPolicies: []br.ManifestRetentionPolicy{
			{
				Name:               "foo",
				ReplicaN:           1,
				Duration:           100,
				ShardGroupDuration: 10,
				ShardGroups: []br.ManifestShardGroup{
					{
						ID:          1,
						StartTime:   now,
						EndTime:     now,
						TruncatedAt: &now,
						Shards: []br.ManifestShardEntry{
							{
								ID:          10,
								ShardOwners: []br.ShardOwnerEntry{{NodeID: 1}},
								ManifestFileEntry: br.ManifestFileEntry{
									FileName:    "10.gz",
									Size:        1000,
									Compression: br.GzipCompression,
								},
							},
						},
					},
					{
						ID:        2,
						StartTime: now,
						EndTime:   now,
						DeletedAt: &now,
						Shards: []br.ManifestShardEntry{
							{
								ID:          30,
								ShardOwners: []br.ShardOwnerEntry{},
								ManifestFileEntry: br.ManifestFileEntry{
									FileName:    "30.gz",
									Size:        3000,
									Compression: br.GzipCompression,
								},
							},
						},
					},
				},
				Subscriptions: []br.ManifestSubscription{},
			},
			{
				Name:               "bar",
				ReplicaN:           3,
				Duration:           9999,
				ShardGroupDuration: 1,
				ShardGroups: []br.ManifestShardGroup{
					{
						ID:        3,
						StartTime: now,
						EndTime:   now,
						Shards:    []br.ManifestShardEntry{},
					},
				},
				Subscriptions: []br.ManifestSubscription{
					{
						Name:         "test",
						Mode:         "on",
						Destinations: []string{"here", "there", "everywhere"},
					},
					{
						Name:         "test2",
						Mode:         "off",
						Destinations: []string{},
					},
				},
			},
		},
	}

	require.Equal(t, expected, converted)
}