const st = new ShardingTest({shards: 2, mongos: 1, rs: {nodes: 1}});
const mongos = st.s0;
const shard0 = st.shard0.shardName;
const shard1 = st.shard1.shardName;

const kDbName = 'test';
const kCollName = 'coll';
const kFullNs = kDbName + '.' + kCollName;

const kms = {
    key: BinData(
        0,
        "/tu9jUCBqZdwCelwE/EAm/4WqdxrSMi04B8e9uAV+m30rI1J2nhKZZtQjdvsSCwuI4erR6IEcEK+5eGUAODv43NDNIR9QheT2edWFewUfHKsl9cnzTc86meIzOmYl6dr")
};

const csfleOpts = {
    kmsProviders: {
        local: kms,
    },
    keyVaultNamespace: kDbName + ".keystore",
    schemaMap: {},
};

var shell = Mongo(mongos.host, csfleOpts);
var kv = shell.getKeyVault();

const schema = {
    "fields": [
        {
            "path": "foo",
            "keyId": kv.createKey("local", "ignored"),
            "bsonType": "string",
            "queries": {"queryType": "equality"}
        },
    ]
};

// Create FLE collection, shard it and move unique range out of the db primary shard
const db = shell.getDB(kDbName);
assert.commandWorked(db.createCollection(kCollName, {encryptedFields: schema}));
assert.commandWorked(mongos.adminCommand({shardCollection: kFullNs, key: {_id: 1}}));
const toShard = st.getPrimaryShard("test").name == shard0 ? shard1 : shard0;
assert.commandWorked(mongos.adminCommand({moveChunk: kFullNs, find: {_id: 0}, to: toShard}));

// This insert will fail
db.getCollection(kCollName).insertOne({foo: "foovalue"});

st.stop();