// This file is part of MinIO Console Server
// Copyright (c) 2021 MinIO, Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package restapi

import (
	"crypto/sha256"
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/go-openapi/swag"
	"github.com/minio/console/models"
	asrt "github.com/stretchr/testify/assert"
)

func TestLogSearch(t *testing.T) {
	responseItem := []map[string]interface{}{
		{
			"time":                    "2006-01-02T15:04:05Z",
			"api_time":                "GetConfigKV",
			"bucket":                  "",
			"object":                  "",
			"time_to_response_ns":     float64(452546530),
			"remote_host":             "10.116.1.94",
			"request_id":              "16595A4E30CCFE79",
			"user_agent":              "MinIO (linux; amd64) madmin-go/0.0.1",
			"response_status":         "OK",
			"response_status_code":    200,
			"request_content_length":  nil,
			"response_content_length": nil,
		}, {
			"time":                    "2006-01-02T15:04:05Z",
			"api_time":                "AssumeRole",
			"bucket":                  "",
			"object":                  "",
			"time_to_response_ns":     float64(307423794),
			"remote_host":             "127.0.0.1",
			"request_id":              "16595A4DA906FBA9",
			"user_agent":              "Go-http-client/1.1",
			"response_status":         "OK",
			"response_status_code":    200,
			"request_content_length":  nil,
			"response_content_length": nil,
		},
	}

	assert := asrt.New(t)
	type args struct {
		apiResponse     string
		apiResponseCode int
	}

	response, _ := json.Marshal(responseItem)

	successfulResponse := &models.LogSearchResponse{
		Results: responseItem,
	}

	tests := []struct {
		name             string
		args             args
		expectedResponse *models.LogSearchResponse
		expectedError    *models.Error
	}{
		{
			name: "200 Success response",
			args: args{
				apiResponse:     string(response),
				apiResponseCode: 200,
			},
			expectedResponse: successfulResponse,
			expectedError:    nil,
		},
		{
			name: "500 unsuccessful response",
			args: args{
				apiResponse:     "Some random error",
				apiResponseCode: 500,
			},
			expectedResponse: nil,
			expectedError: &models.Error{
				Code:    500,
				Message: swag.String(fmt.Sprintf("error retrieving logs: %s", http.StatusText(500))),
			},
		},
	}

	for _, tt := range tests {
		tt := tt
		t.Run(tt.name, func(t *testing.T) {
			testRequest := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.WriteHeader(tt.args.apiResponseCode)
				fmt.Fprintln(w, tt.args.apiResponse)
			}))
			defer testRequest.Close()

			resp, err := logSearch(testRequest.URL)

			if tt.expectedError != nil {
				assert.Equal(tt.expectedError.Code, err.Code, fmt.Sprintf("logSearch() error code: `%v`, wantErr: `%v`", err.Code, tt.expectedError))
				assert.Equal(tt.expectedError.Message, err.Message, fmt.Sprintf("logSearch() error message: `%v`, wantErr: `%v`", err.Message, tt.expectedError))
			} else {
				assert.Nil(err, fmt.Sprintf("logSearch() error: %v, wantErr: %v", err, tt.expectedError))
				buf1, err1 := tt.expectedResponse.MarshalBinary()
				buf2, err2 := resp.MarshalBinary()
				if err1 != err2 {
					t.Errorf("logSearch() resp: %v, expectedResponse: %v", resp, tt.expectedResponse)
					return
				}
				h := sha256.New()
				h.Write(buf1)
				checkSum1 := fmt.Sprintf("%x\n", h.Sum(nil))
				h.Reset()
				h.Write(buf2)
				checkSum2 := fmt.Sprintf("%x\n", h.Sum(nil))
				if checkSum1 != checkSum2 {
					t.Errorf("logSearch() resp: %v, expectedResponse: %v", resp, tt.expectedResponse)
				}
			}
		})
	}
}