Implement tls_server_end_point for postgres-native-tls

Author: sfackler

postgres-native-tls/src/lib.rs

@@ -69,4 +69,8 @@ impl TlsStream for NativeTlsStream {
     fn get_mut(&mut self) -> &mut Stream {
         self.0.get_mut()
     }
+
+    fn tls_server_end_point(&self) -> Option<Vec<u8>> {
+        self.0.tls_server_end_point().ok().and_then(|o| o)
+    }
 }

postgres-native-tls/src/test.rs

@@ -19,3 +19,20 @@ fn connect() {
     ).unwrap();
     conn.execute("SELECT 1::VARCHAR", &[]).unwrap();
 }
+
+#[test]
+fn scram_user() {
+    let cert = include_bytes!("../../test/server.crt");
+    let cert = Certificate::from_pem(cert).unwrap();
+
+    let mut builder = TlsConnector::builder();
+    builder.add_root_certificate(cert);
+    let connector = builder.build().unwrap();
+
+    let handshake = NativeTls::with_connector(connector);
+    let conn = Connection::connect(
+        "postgres://scram_user:password@localhost:5433/postgres",
+        TlsMode::Require(&handshake),
+    ).unwrap();
+    conn.execute("SELECT 1::VARCHAR", &[]).unwrap();
+}