unit tests, fix returned data

test/unit/TestVyOSDriver.py

@@ -0,0 +1,37 @@
+# Copyright 2015 Spotify AB. All rights reserved.
+#
+# The contents of this file are licensed under the Apache License, Version 2.0
+# (the "License"); you may not use this file except in compliance with the
+# License. You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+
+import unittest
+
+from napalm_vyos import vyos
+from napalm_base.test.base import TestConfigNetworkDriver
+
+
+class TestConfigVyOSDriver(unittest.TestCase, TestConfigNetworkDriver):
+
+    @classmethod
+    def setUpClass(cls):
+        hostname = '127.0.0.1'
+        username = 'vagrant'
+        password = 'vagrant'
+        cls.vendor = 'vyos'
+        cls.port = '2200'
+
+        optional_args = {'port': '2200' }
+        cls.device = vyos.VyOSDriver(hostname, username, password,
+                                   timeout=60, optional_args=optional_args)
+        cls.device.open()
+
+        cls.device.load_replace_candidate(filename='%s/initial.conf' % cls.vendor)
+        cls.device.commit_config()