// Copyright © 2018 Enrico Stahn <enrico.stahn@gmail.com>
// 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 phpfpm

import (
	"encoding/json"
	"testing"

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

func TestCountProcessState(t *testing.T) {
	processes := []PoolProcess{
		{State: PoolProcessRequestIdle},
		{State: PoolProcessRequestRunning},
		{State: PoolProcessRequestReadingHeaders},
		{State: PoolProcessRequestInfo},
		{State: PoolProcessRequestFinishing},
		{State: PoolProcessRequestEnding},
	}

	active, idle := CountProcessState(processes)

	assert.Equal(t, int64(2), active, "active processes")
	assert.Equal(t, int64(1), idle, "idle processes")
}

// https://github.com/hipages/php-fpm_exporter/issues/10
func TestCannotUnmarshalNumberIssue10(t *testing.T) {
	pool := Pool{}
	content := []byte(`{
	   "pool":"www",
	   "process manager":"dynamic",
	   "start time":1519474655,
	   "start since":302035,
	   "accepted conn":44144,
	   "listen queue":0,
	   "max listen queue":1,
	   "listen queue len":128,
	   "idle processes":1,
	   "active processes":1,
	   "total processes":2,
	   "max active processes":2,
	   "max children reached":0,
	   "slow requests":0,
	   "processes":[
		  {
			 "pid":23,
			 "state":"Idle",
			 "start time":1519474655,
			 "start since":302035,
			 "requests":22071,
			 "request duration":295,
			 "request method":"GET",
			 "request uri":"/status?json&full",
			 "content length":0,
			 "user":"-",
			 "script":"-",
			 "last request cpu":0.00,
			 "last request memory":2097152
		  },
		  {
			 "pid":24,
			 "state":"Running",
			 "start time":1519474655,
			 "start since":302035,
			 "requests":22073,
			 "request duration":18446744073709550774,
			 "request method":"GET",
			 "request uri":"/status?json&full",
			 "content length":0,
			 "user":"-",
			 "script":"-",
			 "last request cpu":0.00,
			 "last request memory":0
		  }
	   ]
    }`)

	err := json.Unmarshal(content, &pool)

	assert.Nil(t, err, "successfully unmarshal on invalid 'request duration'")
	assert.Equal(t, int(pool.Processes[0].RequestDuration), 295, "request duration set to 0 because it couldn't be deserialized")
	assert.Equal(t, int(pool.Processes[1].RequestDuration), 0, "request duration set to 0 because it couldn't be deserialized")
}

// https://github.com/hipages/php-fpm_exporter/issues/24
func TestInvalidCharacterIssue24(t *testing.T) {
	// todo: Implement fcgi client dependency injection to allow testing of Pool.Update
}

func TestJsonResponseFixer(t *testing.T) {
	pool := Pool{}
	content := []byte(`{"pool":"www","process manager":"dynamic","start time":1528367006,"start since":15073840,"accepted conn":1577112,"listen queue":0,"max listen queue":0,"listen queue len":0,"idle processes":16,"active processes":1,"total processes":17,"max active processes":15,"max children reached":0,"slow requests":0, "processes":[{"pid":15873,"state":"Idle","start time":1543354120,"start since":86726,"requests":853,"request duration":5721,"request method":"GET","request uri":"/vbseo.php?ALTERNATE_TEMPLATES=|%20echo%20"Content-Type:%20text%2Fhtml"%3Becho%20""%20%3B%20id%00","content length":0,"user":"-","script":"/www/forum.example.com/vbseo.php","last request cpu":349.59,"last request memory":786432},{"pid":123,"state":"Idle","start time":1543354120,"start since":86726,"requests":853,"request duration":5721,"request method":"GET","request uri":"123/vbseo.php?ALTERNATE_TEMPLATES=|%20echo%20"Content-Type:%20text%2Fhtml"%3Becho%20""%20%3B%20id%00","content length":0,"user":"-","script":"/www/forum.example.com/vbseo.php","last request cpu":349.59,"last request memory":786432}]}`)

	content = JSONResponseFixer(content)

	err := json.Unmarshal(content, &pool)

	assert.Nil(t, err, "successfully unmarshal on invalid 'request uri'")
	assert.Equal(t, pool.Processes[0].RequestURI, `/vbseo.php?ALTERNATE_TEMPLATES=|%20echo%20"Content-Type:%20text%2Fhtml"%3Becho%20""%20%3B%20id%00`, "request uri couldn't be deserialized")
	assert.Equal(t, pool.Processes[1].RequestURI, `123/vbseo.php?ALTERNATE_TEMPLATES=|%20echo%20"Content-Type:%20text%2Fhtml"%3Becho%20""%20%3B%20id%00`, "request uri couldn't be deserialized")
}

func TestParseURL(t *testing.T) {
	var uris = []struct {
		in  string
		out []string
		err error
	}{
		{"tcp://127.0.0.1:9000/status", []string{"tcp", "127.0.0.1:9000", "/status"}, nil},
		{"tcp://127.0.0.1", []string{"tcp", "127.0.0.1", ""}, nil},
		{"unix:///tmp/php.sock;/status", []string{"unix", "/tmp/php.sock", "/status"}, nil},
		{"unix:///tmp/php.sock", []string{"unix", "/tmp/php.sock", ""}, nil},
	}

	for _, u := range uris {
		scheme, address, path, err := parseURL(u.in)
		assert.Equal(t, u.err, err)
		assert.Equal(t, u.out, []string{scheme, address, path})
	}
}