/*
 * Copyright 2015-2016 IBM Corporation
 *
 * 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 common

import scala.concurrent.Await
import scala.concurrent.ExecutionContext
import scala.concurrent.duration.DurationInt

import akka.actor.ActorSystem
import akka.http.scaladsl.Http

import org.scalatest.BeforeAndAfterAll
import org.scalatest.Suite

/**
 * Helper trait to provide an implicit actor system and execution context
 *  to tests, and properly shut it down at the end.
 *
 *  If you use this trait and override afterAll(), make sure to also call super.afterAll().
 */
trait WskActorSystem extends BeforeAndAfterAll {
    self: Suite =>

    implicit val actorSystem: ActorSystem = ActorSystem()

    implicit def executionContext: ExecutionContext = actorSystem.dispatcher

    override def afterAll() {
        try {
            Await.result(Http().shutdownAllConnectionPools(), 30.seconds)
        } finally {
            actorSystem.terminate()
            Await.result(actorSystem.whenTerminated, 30.seconds)
        }
        super.afterAll()
    }
}