Hang when network fails and recovers

[mjanulaitis]

I have written a simple test program which exposes a hang in the library. When I open a connection, add a vertex then disable my network the next call to the library hangs indefinitely. If I re-enable the network the library continues to hang. The library should error and/or timeout when there is a network failure.

using System;
using System.Diagnostics;
using System.Linq;
using System.Threading.Tasks;
using Gremlin.Net.CosmosDb;
using Gremlin.Net.CosmosDb.Structure;

namespace GremlinTest
{
class Program
{
string Endpoint = "[ENDPOINT]";
string DatabaseName = "[DATABASE]";
string CollectionName = "[COLLECTION]";
string AuthKey = "[AUTHKEY]";
GraphClient graphClient;

    static void Main(string[] args)
    {
        new Program(args);
    }

    public Program(string[] args)
    {
        try
        {
            Task.Run(async () =>
            {
                graphClient = GetGraphClient();
                var vertex = await CreateVertex();

                // Drop the network here
                var savedVertex = await GetVertex(vertex.partitionKey, vertex.id);
                Debug.Assert(savedVertex != null);

            })
            .GetAwaiter().GetResult();
        }
        catch (Exception ex)
        {
            Console.WriteLine(ex);
        }

        Console.WriteLine("Press any key to exit.");
        Console.ReadKey();
    }

    async Task<MyObject> CreateVertex()
    {
        var myObject = new MyObject();
        var g = graphClient.CreateTraversalSource();
        var createCommand = g.AddV(myObject);
        await graphClient.ExecuteAsync(createCommand);

        return myObject;
    }

    async Task<object> GetVertex(string partitionKey, string id)
    {
        var g = graphClient.CreateTraversalSource();
        var queryCommand = g.V(partitionKey, id);

        var result = await graphClient.QueryAsync<object>(queryCommand);
        return result.FirstOrDefault();
    }

    GraphClient GetGraphClient()
    {
        return new GraphClient(Endpoint, DatabaseName, CollectionName, AuthKey);
    }
}

public class MyObject : IVertex
{
    public string id { get; set; } = Guid.NewGuid().ToString("N").ToLower();
    public string partitionKey { get; set; } = "DeadlockTest";
    public string MyValue { get; set; } = "My Test Value";
}

}