SERVER-35393 Flush asio output and limit SecureTransport out buffer

1 files changed, 35 insertions, 16 deletions

diff --git a/jstests/ssl/ssl_fragment.js b/jstests/ssl/ssl_fragment.js
index a3800ea616b..6d08da8e834 100644
--- a/jstests/ssl/ssl_fragment.js
+++ b/jstests/ssl/ssl_fragment.js
@@ -4,25 +4,44 @@
 (function() {
     'use strict';
 
-    const conn = MongoRunner.runMongod({
-        sslMode: "requireSSL",
-        sslPEMKeyFile: "jstests/libs/server.pem",
-        networkMessageCompressors: 'disabled',
-    });
+    function runTest(conn) {
+        // SSL packets have a max size of ~16 kb so to test packet fragmentation support, create a
+        // string larger then 16kb.
+        const chunk = 'E$%G^56w4v5v54Vv$V@#t2#%t56u7B$ub%6 NU@ Y3qv4Yq%yq4C%yx$%zh';  // random data
+        let s = '';
+        while (s.length < (8 * 1024 * 1024)) {
+            s += chunk;
+        }
+
+        const ssl_frag = conn.getCollection('test.ssl_frag');
+        assert.writeOK(ssl_frag.insert({_id: "large_str", foo: s}));
 
-    // SSL packets have a max size of ~16 kb so to test packet fragmentation support, create a
-    // string larger then 16kb.
-    const chunk = 'E$%G^56w4v5v54Vv$V@#t2#%t56u7B$ub%6 NU@ Y3qv4Yq%yq4C%yx$%zh';  // random data
-    let s = '';
-    while (s.length < (8 * 1024 * 1024)) {
-        s += chunk;
+        const read = ssl_frag.find({_id: "large_str"}).toArray()[0].foo;
+        assert.eq(s, read, "Did not receive value written");
     }
 
-    const ssl_frag = conn.getCollection('test.ssl_frag');
-    assert.writeOK(ssl_frag.insert({_id: "large_str", foo: s}));
+    const options = {
+        sslMode: "requireSSL",
+        sslPEMKeyFile: "jstests/libs/server.pem",
+        networkMessageCompressors: 'disabled',
+    };
 
-    const read = ssl_frag.find({_id: "large_str"}).toArray()[0].foo;
-    assert.eq(s, read, "Did not receive value written");
+    const mongod = MongoRunner.runMongod(options);
+    runTest(mongod);
+    MongoRunner.stopMongod(mongod);
 
-    MongoRunner.stopMongod(conn);
+    // TODO: Remove 'shardAsReplicaSet: false' when SERVER-32672 is fixed.
+    const st = new ShardingTest({
+        shards: 3,
+        mongos: 1,
+        config: 1,
+        other: {
+            configOptions: options,
+            mongosOptions: options,
+            shardOptions: options,
+            shardAsReplicaSet: false,
+        }
+    });
+    runTest(st.s0);
+    st.stop();
 })();