From 44948805bc0cf198f431cdbbace10c898ab2d1a2 Mon Sep 17 00:00:00 2001
From: Mircea Ulinic <mirceaulinic@users.noreply.github.com>
Date: Thu, 2 Feb 2017 12:06:48 +0000
Subject: [PATCH] Add vrf arg to ping

Looks like vyos doesn't know about VRF
so these changes should be reflected only
in the method header
---
 napalm_vyos/vyos.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/napalm_vyos/vyos.py b/napalm_vyos/vyos.py
index fdf1c02..dd98e86 100644
--- a/napalm_vyos/vyos.py
+++ b/napalm_vyos/vyos.py
@@ -30,6 +30,7 @@
 from netmiko import SCPConn
 
 # NAPALM base
+import napalm_base.constants as C
 from napalm_base.utils import py23_compat
 from napalm_base.base import NetworkDriver
 from napalm_base.exceptions import ConnectionException, \
@@ -782,7 +783,14 @@ def get_users(self):
 
         return user_auth
 
-    def ping(self, destination, source='', ttl=255, timeout=2, size=100, count=5):
+    def ping(self,
+             destination,
+             source=C.PING_SOURCE,
+             ttl=C.PING_TTL,
+             timeout=C.PING_TIMEOUT,
+             size=C.PING_SIZE,
+             count=C.PING_COUNT,
+             vrf=C.PING_VRF):
         # does not support multiple destination yet
 
         deadline = timeout * count