Add a test for encode query

Author: MahdiBM

Tests/PostgresNIOTests/New/Extensions/PSQLFrontendMessageDecoder.swift

@@ -217,6 +217,8 @@ extension PostgresFrontendMessage {
             preconditionFailure("TODO: Unimplemented")
         case .saslResponse:
             preconditionFailure("TODO: Unimplemented")
+        case .query:
+            return .query
         case .sync:
             return .sync
         case .terminate:

Tests/PostgresNIOTests/New/Extensions/PostgresFrontendMessage.swift

@@ -179,6 +179,7 @@ enum PostgresFrontendMessage: Equatable {
     case saslInitialResponse(SASLInitialResponse)
     case saslResponse(SASLResponse)
     case sslRequest
+    case query
     case sync
     case startup(Startup)
     case terminate
@@ -194,6 +195,7 @@ enum PostgresFrontendMessage: Equatable {
         case password
         case saslInitialResponse
         case saslResponse
+        case query
         case sync
         case terminate
 
@@ -217,6 +219,8 @@ enum PostgresFrontendMessage: Equatable {
                 self = .saslInitialResponse
             case UInt8(ascii: "p"):
                 self = .saslResponse
+            case UInt8(ascii: "Q"):
+                self = .query
             case UInt8(ascii: "S"):
                 self = .sync
             case UInt8(ascii: "X"):
@@ -246,6 +250,8 @@ enum PostgresFrontendMessage: Equatable {
                 return UInt8(ascii: "p")
             case .saslResponse:
                 return UInt8(ascii: "p")
+            case .query:
+                return UInt8(ascii: "Q")
             case .sync:
                 return UInt8(ascii: "S")
             case .terminate:
@@ -283,6 +289,8 @@ extension PostgresFrontendMessage {
             preconditionFailure("SSL requests don't have an identifier")
         case .startup:
             preconditionFailure("Startup messages don't have an identifier")
+        case .query:
+            return .query
         case .sync:
             return .sync
         case .terminate:

Tests/PostgresNIOTests/New/PSQLFrontendMessageTests.swift

@@ -31,7 +31,19 @@ class PSQLFrontendMessageTests: XCTestCase {
         XCTAssertEqual(PostgresFrontendMessage.ID.flush.rawValue, byteBuffer.readInteger(as: UInt8.self))
         XCTAssertEqual(4, byteBuffer.readInteger(as: Int32.self)) // payload length
     }
-    
+
+    func testEncodeQuery() {
+        var encoder = PostgresFrontendMessageEncoder(buffer: .init())
+        let query = "SELECT * FROM foo"
+        encoder.query(query)
+        var byteBuffer = encoder.flushBuffer()
+
+        XCTAssertEqual(byteBuffer.readableBytes, 23)
+        XCTAssertEqual(PostgresFrontendMessage.ID.query.rawValue, byteBuffer.readInteger(as: UInt8.self))
+        XCTAssertEqual(22, byteBuffer.readInteger(as: Int32.self)) // payload length
+        XCTAssertEqual([UInt8](query.utf8), byteBuffer.readBytes(length: 17))
+    }
+
     func testEncodeSync() {
         var encoder = PostgresFrontendMessageEncoder(buffer: .init())
         encoder.sync()