// Copyright 2018 Istio Authors
//
// 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 handler

import (
	"sync"
	"testing"
	"time"

	"istio.io/pkg/log"
	"istio.io/pkg/pool"
)

func TestEnv(t *testing.T) {
	for i := 0; i < 2; i++ {
		gp := pool.NewGoroutinePool(128, i == 0)
		gp.AddWorkers(32)

		// set up the ambient logger so newEnv picks it up
		o := log.DefaultOptions()
		_ = log.Configure(o)

		e := NewEnv(0, "Foo", gp)
		log := e.Logger()
		log.Infof("Test%s", "ing")
		log.Warningf("Test%s", "ing")
		err := log.Errorf("Test%s", "ing")
		if err == nil {
			t.Error("Expected an error but got nil")
		}
		log.Debugf("Test%s", "ing")

		if !log.InfoEnabled() {
			t.Error("Expected true for InfoEnabled check")
		}
		if !log.ErrorEnabled() {
			t.Error("Expected true for ErrorEnabled check")
		}
		if !log.WarnEnabled() {
			t.Error("Expected true for WarnEnabled check")
		}
		if log.DebugEnabled() {
			t.Error("Expected false for DebugEnabled check")
		}

		wg := sync.WaitGroup{}
		wg.Add(1)
		e.ScheduleWork(func() {
			if !e.(env).hasStrayWorkers() {
				t.Error("hasStrayWorkers() => false; wanted true")
			}
			wg.Done()
		})

		wg.Add(1)
		e.ScheduleDaemon(func() {
			if !e.(env).hasStrayWorkers() {
				t.Error("hasStrayWorkers() => false; wanted true")
			}
			wg.Done()
		})

		e.ScheduleWork(func() {
			panic("bye!")
		})

		e.ScheduleDaemon(func() {
			panic("bye!")
		})

		wg.Wait()

		// hack to give time for the panic to 'take hold' if it doesn't get recovered properly
		time.Sleep(200 * time.Millisecond)

		if e.(env).hasStrayWorkers() {
			t.Error("hasStrayWorkers() => true; wanted false")
		}

		_ = gp.Close()
	}
}
