option_test.go

// Copyright 2019 SEQSENSE, Inc.
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package s3sync

import (
	"testing"

	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/session"
	"github.com/aws/aws-sdk-go/service/s3/s3manager"
)

func TestWithParallel(t *testing.T) {
	sess := session.New(&aws.Config{
		Credentials: credentials.AnonymousCredentials,
		Region:      aws.String("dummy"),
	})

	m := New(sess, WithParallel(2))
	if m.nJobs != 2 {
		t.Fatal("Manager.nJobs must be configured by WithParallel option")
	}
}

func TestWithACL(t *testing.T) {
	sess := session.New(&aws.Config{
		Credentials: credentials.AnonymousCredentials,
		Region:      aws.String("dummy"),
	})

	t.Run("Nil", func(t *testing.T) {
		m := New(sess)
		if m.acl != nil {
			t.Fatal("Manager.acl must be nil if initialized with WithACL")
		}
	})
	t.Run("WithACL", func(t *testing.T) {
		m := New(sess, WithACL("test"))
		if *m.acl != "test" {
			t.Fatal("Manager.acl must be configured by WithParallel option")
		}
	})
}

func TestUploaderDownloaderOptions(t *testing.T) {
	sess := session.New(&aws.Config{
		Credentials: credentials.AnonymousCredentials,
		Region:      aws.String("dummy"),
	})

	t.Run("Uploader", func(t *testing.T) {
		m := New(sess, WithUploaderOptions(
			func(u *s3manager.Uploader) {},
		))
		if len(m.uploaderOpts) != 1 {
			t.Fatal("Manager.uploaderOpts must have a option")
		}
	})
	t.Run("Downloader", func(t *testing.T) {
		m := New(sess, WithDownloaderOptions(
			func(d *s3manager.Downloader) {},
		))
		if len(m.downloaderOpts) != 1 {
			t.Fatal("Manager.downloaderOpts must have a option")
		}
	})
}