driver/provider: add export support

Signed-off-by: Jan Luebbe <[email protected]>

Author: jluebbe

labgrid/driver/provider.py

@@ -14,6 +14,14 @@ class BaseProviderDriver(Driver):
     def __attrs_post_init__(self):
         super().__attrs_post_init__()
 
+    @Driver.check_bound
+    def get_export_vars(self):
+        return {
+            "host": self.provider.host,
+            "internal": self.provider.internal,
+            "external": self.provider.external,
+        }
+
     @Driver.check_active
     @step(args=['filename'], result=True)
     def stage(self, filename):

tests/test_export.py

@@ -1,8 +1,8 @@
 import pytest
 
 from labgrid.resource import Resource, NetworkSerialPort
-from labgrid.resource.remote import RemoteNetworkInterface
-from labgrid.driver import Driver, SerialDriver, NetworkInterfaceDriver
+from labgrid.resource.remote import RemoteNetworkInterface, RemoteTFTPProvider
+from labgrid.driver import Driver, SerialDriver, NetworkInterfaceDriver, TFTPProviderDriver
 from labgrid.strategy import Strategy
 from labgrid.binding import StateError
 
@@ -86,3 +86,15 @@ def test_export_remote_network_interface(target):
         'LG__NETIF_HOST': 'testhost',
         'LG__NETIF_IFNAME': 'wlan0'
     }
+
+
+def test_export_remote_tftp_provider(target):
+    RemoteTFTPProvider(target, None, host='testhost', internal='/srv/tftp/testboard/', external='testboard/')
+    TFTPProviderDriver(target, "tftp")
+
+    exported = target.export()
+    assert exported == {
+        'LG__TFTP_HOST': 'testhost',
+        'LG__TFTP_INTERNAL': '/srv/tftp/testboard/',
+        'LG__TFTP_EXTERNAL': 'testboard/',
+    }