• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• skunk-master
  • .github
    • workflows
      • build.yml
      • release.yml
  • LICENSE
  • project
    • build.properties
    • project
      • plugins.sbt
    • plugins.sbt
  • .gitattributes
  • CHANGELOG.md
  • README.md
  • world
    • server.key
    • README.md
    • server.crt
    • Dockerfile
  • build.sbt
  • .gitignore
  • modules
    • circe
      • src
        • main
          • scala
            • JsonCodecs.scala
    • example
      • src
        • main
          • scala
            • Minimal1.scala
            • NatchezHttp4sModule.scala
            • Channel.scala
            • Error.scala
            • Math1.scala
            • Values.scala
            • Main.scala
            • Join.scala
            • Http4s.scala
            • Math2.scala
            • Minimal3.scala
            • Minimal2.scala
            • Transaction.scala
    • refined
      • src
        • main
          • scala
            • codec
              • NumericCodecs.scala
    • tests
      • src
        • main
          • scala
            • ffstest
              • FFramework.scala
        • test
          • scala
            • CommandTest.scala
            • StartupTest.scala
            • codec
              • BinaryCodecTest.scala
              • EnumCodecTest.scala
              • NumericCodecTest.scala
              • JsonCodecTest.scala
              • TextCodecTest.scala
              • BooleanCodecTest.scala
              • UuidCodecTest.scala
              • TemporalCodecTest.scala
              • CodecTest.scala
            • ExtendedQueryErrorTest.scala
            • SkunkTest.scala
            • SslTest.scala
            • TransactionStatusTest.scala
            • ListTest.scala
            • ErrorResponseTest.scala
            • PoolTest.scala
            • QuickQueryErrorTest.scala
    • core
      • src
        • main
          • scala-2.12
            • utll
              • package.scala
          • scala
            • Decoder.scala
            • PreparedCommand.scala
            • Statement.scala
            • Channel.scala
            • codec
              • BooleanCodec.scala
              • UuidCodec.scala
              • BinaryCodecs.scala
              • AllCodecs.scala
              • EnumCodec.scala
              • TemporalCodecs.scala
              • NumericCodecs.scala
              • TextCodecs.scala
            • Cursor.scala
            • Command.scala
            • Session.scala
            • package.scala
            • net
              • SSLNegotiation.scala
              • package.scala
              • Protocol.scala
              • AbstractMessageSocket.scala
              • message
                • AuthenticationRequest.scala
                • ParseComplete.scala
                • PortalSuspended.scala
                • AuthenticationKerberosV5.scala
                • Terminate.scala
                • ErrorResponse.scala
                • AuthenticationSSPI.scala
                • AuthenticationSASLContinue.scala
                • package.scala
                • NoticeResponse.scala
                • FrontendMessage.scala
                • AuthenticationSASL.scala
                • Flush.scala
                • Execute.scala
                • AuthenticationCleartextPassword.scala
                • RowDescription.scala
                • Sync.scala
                • StartupMessage.scala
                • AuthenticationMD5Password.scala
                • AuthenticationSASLFinal.scala
                • BackendMessage.scala
                • CommandComplete.scala
                • ParameterDescription.scala
                • Describe.scala
                • Close.scala
                • ParameterStatus.scala
                • NotificationResponse.scala
                • Parse.scala
                • RowData.scala
                • Bind.scala
                • BindComplete.scala
                • UnknownMessage.scala
                • CloseComplete.scala
                • AuthenticationSCMCredential.scala
                • Query.scala
                • AuthenticationOk.scala
                • NoData.scala
                • PasswordMessage.scala
                • ReadyForQuery.scala
                • AuthenticationGSS.scala
                • BackendKeyData.scala
                • AuthenticationGSSContinue.scala
              • MessageSocket.scala
              • BufferedMessageSocket.scala
              • protocol
                • Unroll.scala
                • Exchange.scala
                • package.scala
                • Execute.scala
                • Startup.scala
                • Describe.scala
                • Close.scala
                • Parse.scala
                • Bind.scala
                • Query.scala
                • Prepare.scala
              • BitVectorSocket.scala
            • SSL.scala
            • PreparedQuery.scala
            • Encoder.scala
            • util
              • Text.scala
              • Twiddler.scala
              • Pretty.scala
              • Recycler.scala
              • Namer.scala
              • Pool.scala
              • Typer.scala
            • Void.scala
            • syntax
              • ListOps.scala
              • IdOps.scala
              • AllOps.scala
              • StringContextOps.scala
            • exception
              • NoDataException.scala
              • UnknownTypeException.scala
              • StartupException.scala
              • UnknownOidException.scala
              • package.scala
              • DecodeException.scala
              • SkunkException.scala
              • ProtocolError.scala
              • UnexpectedColumnsException.scala
              • ColumnAlignmentException.scala
              • PostgresErrorException.scala
            • data
              • TypedRowDescription.scala
              • Completion.scala
              • Identifier.scala
              • Notification.scala
              • Type.scala
              • TransactionStatus.scala
            • SqlState.scala
            • Fragment.scala
            • Query.scala
            • Codec.scala
            • Transaction.scala
          • scala-2.13
            • utll
              • package.scala
    • docs
      • src
        • main
          • paradox
            • tutorial
              • Setup.md
              • Channels.md
              • Transactions.md
              • Command.md
              • Tracing.md
              • Index.md
              • Query.md
            • index.md
            • reference
              • TwiddleLists.md
              • SchemaTypes.md
              • Fragments.md
              • Sessions.md
              • Concurrency.md
              • Index.md
              • Identifiers.md
              • Encoders.md
    • macros
      • src
        • main
          • scala
            • util
              • Origin.scala
  • docker-compose.yml

// Copyright (c) 2018-2020 by Rob Norris
// This software is licensed under the MIT License (MIT).
// For more information see LICENSE or https://opensource.org/licenses/MIT

package tests

import cats.effect.{ IO, Resource }
import cats.implicits._
import skunk.Session
import skunk.data._
import skunk.codec.all._
import skunk.implicits._
import skunk.util.Typer
import natchez.Trace.Implicits.noop

abstract class SkunkTest(debug: Boolean = false, strategy: Typer.Strategy = Typer.Strategy.BuiltinsOnly) extends ffstest.FTest {

  val session: Resource[IO, Session[IO]] =
    Session.single(
      host     = "localhost",
      port     = 5432,
      user     = "jimmy",
      database = "world",
      password = Some("banana"),
      strategy = strategy,
      debug    = debug
    )

  def sessionTest[A](name: String)(fa: Session[IO] => IO[A]): Unit =
    test(name)(session.use(fa))

  implicit class SkunkTestSessionOps(s: Session[IO]) {

    def assertTransactionStatus(msg: String, xas: TransactionStatus): IO[Unit] =
      s.transactionStatus.get.flatMap(a => assert(s"$msg (expected $xas, got $a)", a === xas))

    def assertHealthy: IO[Unit] =
      for {
        _ <- assertTransactionStatus("sanity check", TransactionStatus.Idle)
        n <- s.unique(sql"select 'SkunkTest Health Check'::varchar".query(varchar))
        _ <- assert("sanity check", n == "SkunkTest Health Check")
      } yield ()

  }

}