kalix.javasdk.testkit.impl.TestKitActionContext.scala Maven / Gradle / Ivy
/*
* Copyright 2021 Lightbend 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 kalix.javasdk.testkit.impl;
import kalix.javasdk.Metadata
import kalix.javasdk.action.{ ActionContext, ActionCreationContext }
import kalix.javasdk.impl.InternalContext
import kalix.javasdk.testkit.MockRegistry
/**
* INTERNAL API Used by the generated testkit
*/
final class TestKitActionContext(metadata: Metadata, mockRegistry: MockRegistry = MockRegistry.EMPTY)
extends AbstractTestKitContext(mockRegistry)
with ActionContext
with ActionCreationContext
with InternalContext {
def this() = {
this(Metadata.EMPTY, MockRegistry.EMPTY)
}
def this(metadata: Metadata) = {
this(metadata, MockRegistry.EMPTY)
}
override def metadata() = metadata
override def eventSubject() = metadata.get("ce-subject")
override def getGrpcClient[T](clientClass: Class[T], service: String): T = getComponentGrpcClient(clientClass)
}