zio.aws.mwaa.MwaaMock.scala Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of zio-aws-mwaa_2.13 Show documentation
Low-level AWS wrapper for ZIO
The newest version!
package zio.aws.mwaa
import zio.aws.mwaa.model.UpdateEnvironmentResponse.ReadOnly
import zio.mock.{Proxy, Mock}
import zio.aws.mwaa.model.{
  GetEnvironmentRequest,
  ListEnvironmentsRequest,
  UpdateEnvironmentRequest,
  ListTagsForResourceRequest,
  CreateCliTokenRequest,
  UntagResourceRequest,
  CreateEnvironmentRequest,
  DeleteEnvironmentRequest,
  CreateWebLoginTokenRequest,
  TagResourceRequest,
  PublishMetricsRequest
}
import zio.aws.mwaa.model.primitives.EnvironmentName
import zio.aws.core.AwsError
import zio.aws.core.aspects.AwsCallAspect
import zio.{IO, ZEnvironment, URLayer}
import zio.stream.ZStream
import software.amazon.awssdk.services.mwaa.MwaaAsyncClient
object MwaaMock extends Mock[Mwaa] {
  object DeleteEnvironment
      extends Effect[
        DeleteEnvironmentRequest,
        AwsError,
        zio.aws.mwaa.model.DeleteEnvironmentResponse.ReadOnly
      ]
  object PublishMetrics
      extends Effect[
        PublishMetricsRequest,
        AwsError,
        zio.aws.mwaa.model.PublishMetricsResponse.ReadOnly
      ]
  object CreateEnvironment
      extends Effect[
        CreateEnvironmentRequest,
        AwsError,
        zio.aws.mwaa.model.CreateEnvironmentResponse.ReadOnly
      ]
  object GetEnvironment
      extends Effect[
        GetEnvironmentRequest,
        AwsError,
        zio.aws.mwaa.model.GetEnvironmentResponse.ReadOnly
      ]
  object ListEnvironments
      extends Stream[ListEnvironmentsRequest, AwsError, EnvironmentName]
  object ListEnvironmentsPaginated
      extends Effect[
        ListEnvironmentsRequest,
        AwsError,
        zio.aws.mwaa.model.ListEnvironmentsResponse.ReadOnly
      ]
  object UntagResource
      extends Effect[
        UntagResourceRequest,
        AwsError,
        zio.aws.mwaa.model.UntagResourceResponse.ReadOnly
      ]
  object ListTagsForResource
      extends Effect[
        ListTagsForResourceRequest,
        AwsError,
        zio.aws.mwaa.model.ListTagsForResourceResponse.ReadOnly
      ]
  object TagResource
      extends Effect[
        TagResourceRequest,
        AwsError,
        zio.aws.mwaa.model.TagResourceResponse.ReadOnly
      ]
  object UpdateEnvironment
      extends Effect[UpdateEnvironmentRequest, AwsError, ReadOnly]
  object CreateCliToken
      extends Effect[
        CreateCliTokenRequest,
        AwsError,
        zio.aws.mwaa.model.CreateCliTokenResponse.ReadOnly
      ]
  object CreateWebLoginToken
      extends Effect[
        CreateWebLoginTokenRequest,
        AwsError,
        zio.aws.mwaa.model.CreateWebLoginTokenResponse.ReadOnly
      ]
  val compose: URLayer[Proxy, Mwaa] = zio.ZLayer {
    zio.ZIO.service[Proxy].flatMap { proxy =>
      withRuntime[Proxy, Mwaa] { rts =>
        zio.ZIO.succeed {
          new Mwaa {
            val api: MwaaAsyncClient = null
            def withAspect[R1](
                newAspect: AwsCallAspect[R1],
                r: ZEnvironment[R1]
            ): Mwaa = this
            def deleteEnvironment(request: DeleteEnvironmentRequest): IO[
              AwsError,
              zio.aws.mwaa.model.DeleteEnvironmentResponse.ReadOnly
            ] = proxy(DeleteEnvironment, request)
            def publishMetrics(request: PublishMetricsRequest): IO[
              AwsError,
              zio.aws.mwaa.model.PublishMetricsResponse.ReadOnly
            ] = proxy(PublishMetrics, request)
            def createEnvironment(request: CreateEnvironmentRequest): IO[
              AwsError,
              zio.aws.mwaa.model.CreateEnvironmentResponse.ReadOnly
            ] = proxy(CreateEnvironment, request)
            def getEnvironment(request: GetEnvironmentRequest): IO[
              AwsError,
              zio.aws.mwaa.model.GetEnvironmentResponse.ReadOnly
            ] = proxy(GetEnvironment, request)
            def listEnvironments(
                request: ListEnvironmentsRequest
            ): ZStream[Any, AwsError, EnvironmentName] =
              zio.Unsafe.unsafe { implicit u =>
                rts.unsafe
                  .run {
                    proxy(ListEnvironments, request)
                  }
                  .getOrThrowFiberFailure()
              }
            def listEnvironmentsPaginated(request: ListEnvironmentsRequest): IO[
              AwsError,
              zio.aws.mwaa.model.ListEnvironmentsResponse.ReadOnly
            ] = proxy(ListEnvironmentsPaginated, request)
            def untagResource(
                request: UntagResourceRequest
            ): IO[AwsError, zio.aws.mwaa.model.UntagResourceResponse.ReadOnly] =
              proxy(UntagResource, request)
            def listTagsForResource(request: ListTagsForResourceRequest): IO[
              AwsError,
              zio.aws.mwaa.model.ListTagsForResourceResponse.ReadOnly
            ] = proxy(ListTagsForResource, request)
            def tagResource(
                request: TagResourceRequest
            ): IO[AwsError, zio.aws.mwaa.model.TagResourceResponse.ReadOnly] =
              proxy(TagResource, request)
            def updateEnvironment(
                request: UpdateEnvironmentRequest
            ): IO[AwsError, ReadOnly] = proxy(UpdateEnvironment, request)
            def createCliToken(request: CreateCliTokenRequest): IO[
              AwsError,
              zio.aws.mwaa.model.CreateCliTokenResponse.ReadOnly
            ] = proxy(CreateCliToken, request)
            def createWebLoginToken(request: CreateWebLoginTokenRequest): IO[
              AwsError,
              zio.aws.mwaa.model.CreateWebLoginTokenResponse.ReadOnly
            ] = proxy(CreateWebLoginToken, request)
          }
        }
      }
    }
  }
}