minio/cmd/bucket-notification-handlers_test.go

/*
 * Minio Cloud Storage, (C) 2016, 2017 Minio, 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 cmd

import (
	"bufio"
	"bytes"
	"encoding/json"
	"encoding/xml"
	"errors"
	"io"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"os"
	"reflect"
	"sync"
	"testing"

	"time"

	"github.com/minio/minio/pkg/auth"
)

// Implement a dummy flush writer.
type flushWriter struct {
	io.Writer
}

// Flush writer is a dummy writer compatible with http.Flusher and http.ResponseWriter.
func (f *flushWriter) Flush()                            {}
func (f *flushWriter) Write(b []byte) (n int, err error) { return f.Writer.Write(b) }
func (f *flushWriter) Header() http.Header               { return http.Header{} }
func (f *flushWriter) WriteHeader(code int)              {}

func newFlushWriter(writer io.Writer) http.ResponseWriter {
	return &flushWriter{writer}
}

// Tests write notification code.
func TestWriteNotification(t *testing.T) {
	// Initialize a new test config.
	root, err := newTestConfig(globalMinioDefaultRegion)
	if err != nil {
		t.Fatalf("Unable to initialize test config %s", err)
	}
	defer os.RemoveAll(root)

	var buffer bytes.Buffer
	// Collection of test cases for each event writer.
	testCases := []struct {
		writer http.ResponseWriter
		event  map[string][]NotificationEvent
		err    error
	}{
		// Invalid input argument with writer `nil` - Test - 1
		{
			writer: nil,
			event:  nil,
			err:    errInvalidArgument,
		},
		// Invalid input argument with event `nil` - Test - 2
		{
			writer: newFlushWriter(ioutil.Discard),
			event:  nil,
			err:    errInvalidArgument,
		},
		// Unmarshal and write, validate last 5 bytes. - Test - 3
		{
			writer: newFlushWriter(&buffer),
			event: map[string][]NotificationEvent{
				"Records": {newNotificationEvent(eventData{
					Type:   ObjectCreatedPut,
					Bucket: "testbucket",
					ObjInfo: ObjectInfo{
						Name: "key",
					},
					ReqParams: map[string]string{
						"ip": "10.1.10.1",
					}}),
				},
			},
			err: nil,
		},
	}
	// Validates all the testcases for writing notification.
	for _, testCase := range testCases {
		err := writeNotification(testCase.writer, testCase.event)
		if err != testCase.err {
			t.Errorf("Unable to write notification %s", err)
		}
		// Validates if the ending string has 'crlf'
		if err == nil && !bytes.HasSuffix(buffer.Bytes(), crlf) {
			buf := buffer.Bytes()[buffer.Len()-5 : 0]
			t.Errorf("Invalid suffix found from the writer last 5 bytes %s, expected `\r\n`", string(buf))
		}
		// Not printing 'buf' on purpose, validates look for string '10.1.10.1'.
		if err == nil && !bytes.Contains(buffer.Bytes(), []byte("10.1.10.1")) {
			// Enable when debugging)
			// fmt.Println(string(buffer.Bytes()))
			t.Errorf("Requested content couldn't be found, expected `10.1.10.1`")
		}
	}
}

// testResponseWriter implements `http.ResponseWriter` that buffers
// response body in a `bytes.Buffer` and returns error after `failCount`
// calls to `Write` method
type testResponseWriter struct {
	mu        sync.Mutex
	failCount int
	buf       *bytes.Buffer
	m         http.Header
}

func newTestResponseWriter(failAt int) *testResponseWriter {
	return &testResponseWriter{
		buf:       new(bytes.Buffer),
		m:         make(http.Header),
		failCount: failAt,
	}
}

func (trw *testResponseWriter) Flush() {
}

func (trw *testResponseWriter) Write(p []byte) (int, error) {
	trw.mu.Lock()
	defer trw.mu.Unlock()

	if trw.failCount == 0 {
		return 0, errors.New("Custom error")
	}
	trw.failCount--

	return trw.buf.Write(p)
}

func (trw *testResponseWriter) Header() http.Header {
	return trw.m
}

func (trw *testResponseWriter) WriteHeader(i int) {
}

func TestListenChan(t *testing.T) {
	// Initialize a new test config.
	root, err := newTestConfig(globalMinioDefaultRegion)
	if err != nil {
		t.Fatalf("Unable to initialize test config %s", err)
	}
	defer os.RemoveAll(root)

	// Create a listen channel to manage notifications
	nListenCh := newListenChan()

	// Construct notification events to be passed on the events channel.
	var events []NotificationEvent
	evTypes := []EventName{
		ObjectCreatedPut,
		ObjectCreatedPost,
		ObjectCreatedCopy,
		ObjectCreatedCompleteMultipartUpload,
	}

	for _, evType := range evTypes {
		events = append(events, newNotificationEvent(eventData{
			Type: evType,
		}))
	}

	// Send notification events one-by-one
	go func() {
		for _, event := range events {
			nListenCh.sendNotificationEvent([]NotificationEvent{event})
		}
	}()

	// Create a http.ResponseWriter that fails after len(events)
	// number of times
	trw := newTestResponseWriter(len(events))

	// Wait for all (4) notification events to be received
	nListenCh.waitForListener(trw)

	// Used to read JSON-formatted event stream line-by-line
	scanner := bufio.NewScanner(trw.buf)
	var records map[string][]NotificationEvent
	for i := 0; scanner.Scan(); i++ {
		err = json.Unmarshal(scanner.Bytes(), &records)
		if err != nil {
			t.Fatalf("Failed to unmarshal json %v", err)
		}

		nEvent := records["Records"][0]
		if nEvent.EventName != evTypes[i].String() {
			t.Errorf("notification event name mismatch, expected %s but got %s", evTypes[i], nEvent.EventName)
		}
	}
}

func TestSendNotificationEvent(t *testing.T) {
	// This test verifies that sendNotificationEvent function
	// returns once listenChan.doneCh is closed

	l := newListenChan()
	testCh := make(chan struct{})
	timeout := 5 * time.Second

	go func() {
		// Send one empty notification event on listenChan
		events := []NotificationEvent{NotificationEvent{}}
		l.sendNotificationEvent(events)
		testCh <- struct{}{}
	}()

	// close l.doneCh to signal client exiting from
	// ListenBucketNotification API call
	close(l.doneCh)

	select {
	case <-time.After(timeout):
		t.Fatalf("sendNotificationEvent didn't return after %v seconds", timeout)
	case <-testCh:
		// If we reach this case, sendNotificationEvent
		// returned on closing l.doneCh
	}
}

func TestGetBucketNotificationHandler(t *testing.T) {
	ExecObjectLayerAPITest(t, testGetBucketNotificationHandler, []string{
		"GetBucketNotification",
	})
}

func testGetBucketNotificationHandler(obj ObjectLayer, instanceType, bucketName string, apiRouter http.Handler,
	credentials auth.Credentials, t *testing.T) {
	// declare sample configs
	filterRules := []filterRule{
		{
			Name:  "prefix",
			Value: "minio",
		},
		{
			Name:  "suffix",
			Value: "*.jpg",
		},
	}
	sampleSvcCfg := ServiceConfig{
		[]string{"s3:ObjectRemoved:*", "s3:ObjectCreated:*"},
		filterStruct{
			keyFilter{filterRules},
		},
		"1",
	}
	sampleNotifCfg := notificationConfig{
		QueueConfigs: []queueConfig{
			{
				ServiceConfig: sampleSvcCfg,
				QueueARN:      "testqARN",
			},
		},
	}
	rec := httptest.NewRecorder()
	req, err := newTestSignedRequestV4("GET", getGetBucketNotificationURL("", bucketName),
		0, nil, credentials.AccessKey, credentials.SecretKey)
	if err != nil {
		t.Fatalf("%s: Failed to create HTTP testRequest for ListenBucketNotification: <ERROR> %v", instanceType, err)
	}
	apiRouter.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("Unexpected http response %d", rec.Code)
	}
	if err = persistNotificationConfig(bucketName, &sampleNotifCfg, obj); err != nil {
		t.Fatalf("Unable to save notification config %s", err)
	}
	rec = httptest.NewRecorder()
	req, err = newTestSignedRequestV4("GET", getGetBucketNotificationURL("", bucketName),
		0, nil, credentials.AccessKey, credentials.SecretKey)
	if err != nil {
		t.Fatalf("%s: Failed to create HTTP testRequest for ListenBucketNotification: <ERROR> %v", instanceType, err)
	}
	apiRouter.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("Unexpected http response %d", rec.Code)
	}
	notificationBytes, err := ioutil.ReadAll(rec.Body)
	if err != nil {
		t.Fatalf("Unexpected error %s", err)
	}
	nConfig := notificationConfig{}
	if err = xml.Unmarshal(notificationBytes, &nConfig); err != nil {
		t.Fatalf("Unexpected XML received %s", err)
	}
	if sampleNotifCfg.QueueConfigs[0].QueueARN != nConfig.QueueConfigs[0].QueueARN {
		t.Fatalf("Uexpected notification configs expected %#v, got %#v", sampleNotifCfg, nConfig)
	}
	if !reflect.DeepEqual(sampleNotifCfg.QueueConfigs[0].Events, nConfig.QueueConfigs[0].Events) {
		t.Fatalf("Uexpected notification configs expected %#v, got %#v", sampleNotifCfg, nConfig)
	}
}

func TestPutBucketNotificationHandler(t *testing.T) {
	ExecObjectLayerAPITest(t, testPutBucketNotificationHandler, []string{
		"PutBucketNotification",
	})
}

func testPutBucketNotificationHandler(obj ObjectLayer, instanceType,
	bucketName string, apiRouter http.Handler, credentials auth.Credentials,
	t *testing.T) {

	// declare sample configs
	filterRules := []filterRule{
		{
			Name:  "prefix",
			Value: "minio",
		},
		{
			Name:  "suffix",
			Value: "*.jpg",
		},
	}
	sampleSvcCfg := ServiceConfig{
		[]string{"s3:ObjectRemoved:*", "s3:ObjectCreated:*"},
		filterStruct{
			keyFilter{filterRules},
		},
		"1",
	}
	sampleNotifCfg := notificationConfig{
		QueueConfigs: []queueConfig{
			{
				ServiceConfig: sampleSvcCfg,
				QueueARN:      "testqARN",
			},
		},
	}

	{
		sampleNotifCfg.LambdaConfigs = []lambdaConfig{
			{
				sampleSvcCfg, "testLARN",
			},
		}
		xmlBytes, err := xml.Marshal(sampleNotifCfg)
		if err != nil {
			t.Fatalf("%s: Unexpected err: %#v", instanceType, err)
		}
		rec := httptest.NewRecorder()
		req, err := newTestSignedRequestV4("PUT",
			getPutBucketNotificationURL("", bucketName),
			int64(len(xmlBytes)), bytes.NewReader(xmlBytes),
			credentials.AccessKey, credentials.SecretKey)
		if err != nil {
			t.Fatalf("%s: Failed to create HTTP testRequest for PutBucketNotification: <ERROR> %v",
				instanceType, err)
		}
		apiRouter.ServeHTTP(rec, req)
		if rec.Code != http.StatusBadRequest {
			t.Fatalf("Unexpected http response %d", rec.Code)
		}
	}

	{
		sampleNotifCfg.LambdaConfigs = nil
		sampleNotifCfg.TopicConfigs = []topicConfig{
			{
				sampleSvcCfg, "testTARN",
			},
		}
		xmlBytes, err := xml.Marshal(sampleNotifCfg)
		if err != nil {
			t.Fatalf("%s: Unexpected err: %#v", instanceType, err)
		}
		rec := httptest.NewRecorder()
		req, err := newTestSignedRequestV4("PUT",
			getPutBucketNotificationURL("", bucketName),
			int64(len(xmlBytes)), bytes.NewReader(xmlBytes),
			credentials.AccessKey, credentials.SecretKey)
		if err != nil {
			t.Fatalf("%s: Failed to create HTTP testRequest for PutBucketNotification: <ERROR> %v",
				instanceType, err)
		}
		apiRouter.ServeHTTP(rec, req)
		if rec.Code != http.StatusBadRequest {
			t.Fatalf("Unexpected http response %d", rec.Code)
		}
	}
}

func TestListenBucketNotificationNilHandler(t *testing.T) {
	ExecObjectLayerAPITest(t, testListenBucketNotificationNilHandler, []string{
		"ListenBucketNotification",
		"PutObject",
	})
}

func testListenBucketNotificationNilHandler(obj ObjectLayer, instanceType, bucketName string, apiRouter http.Handler,
	credentials auth.Credentials, t *testing.T) {
	// get random bucket name.
	randBucket := getRandomBucketName()

	// Nil Object layer
	nilAPIRouter := initTestAPIEndPoints(nil, []string{
		"ListenBucketNotification",
	})
	testRec := httptest.NewRecorder()
	testReq, tErr := newTestSignedRequestV4("GET",
		getListenBucketNotificationURL("", randBucket, []string{},
			[]string{"*.jpg"}, []string{
				"s3:ObjectCreated:*",
				"s3:ObjectRemoved:*",
				"s3:ObjectAccessed:*",
			}), 0, nil, credentials.AccessKey, credentials.SecretKey)
	if tErr != nil {
		t.Fatalf("%s: Failed to create HTTP testRequest for ListenBucketNotification: <ERROR> %v", instanceType, tErr)
	}
	nilAPIRouter.ServeHTTP(testRec, testReq)
	if testRec.Code != http.StatusServiceUnavailable {
		t.Fatalf("Test 1: %s: expected HTTP code %d, but received %d: <ERROR> %v",
			instanceType, http.StatusServiceUnavailable, testRec.Code, tErr)
	}
}

func testRemoveNotificationConfig(obj ObjectLayer, instanceType,
	bucketName string, apiRouter http.Handler, credentials auth.Credentials,
	t *testing.T) {

	invalidBucket := "Invalid\\Bucket"
	// get random bucket name.
	randBucket := bucketName

	nCfg := notificationConfig{
		QueueConfigs: []queueConfig{
			{
				ServiceConfig: ServiceConfig{
					Events: []string{"s3:ObjectRemoved:*",
						"s3:ObjectCreated:*"},
				},
				QueueARN: "testqARN",
			},
		},
	}
	if err := persistNotificationConfig(randBucket, &nCfg, obj); err != nil {
		t.Fatalf("Unexpected error: %#v", err)
	}

	testCases := []struct {
		bucketName  string
		expectedErr error
	}{
		{invalidBucket, BucketNameInvalid{Bucket: invalidBucket}},
		{randBucket, nil},
	}
	for i, test := range testCases {
		tErr := removeNotificationConfig(test.bucketName, obj)
		if tErr != test.expectedErr {
			t.Errorf("Test %d: %s expected error %v, but received %v", i+1, instanceType, test.expectedErr, tErr)
		}
	}
}

func TestRemoveNotificationConfig(t *testing.T) {
	ExecObjectLayerAPITest(t, testRemoveNotificationConfig, []string{
		"PutBucketNotification",
		"ListenBucketNotification",
	})
}