public void SendHeartbeat_Always_InvokesSendHeartbeat()
        {
            // Arrange
            var mockDataNodeProtocol = new Mock <IDataNodeProtocol>();
            var dataNodeId           = Guid.NewGuid();

            mockDataNodeProtocol.Setup(x => x.SendHeartbeat(dataNodeId));
            var controller = new DataNodeProtocolController(mockDataNodeProtocol.Object);

            // Act
            controller.SendHeartbeat(dataNodeId);

            // Assert
            mockDataNodeProtocol.VerifyAll();
        }
        public void Register_Always_InvokesRegister()
        {
            // Arrange
            var mockDataNodeProtocol = new Mock <IDataNodeProtocol>();
            var dataNodeId           = new DataNodeId();

            mockDataNodeProtocol.Setup(x => x.RegisterDataNode(dataNodeId));
            var controller = new DataNodeProtocolController(mockDataNodeProtocol.Object);

            // Act
            controller.Register(dataNodeId);

            // Assert
            mockDataNodeProtocol.VerifyAll();
        }