From 311554324aed1f2db0eb4beee716a32c13e8ca17 Mon Sep 17 00:00:00 2001
From: bwilson <bwilson@hbk.com>
Date: Tue, 13 Nov 2018 10:37:37 -0600
Subject: [PATCH] Adding RemoteTarget class to wrap module/function

---
 xdist/newhooks.py     |  7 +++++--
 xdist/workermanage.py | 26 ++++++++++++++++++++++----
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/xdist/newhooks.py b/xdist/newhooks.py
index f389192..b10049c 100644
--- a/xdist/newhooks.py
+++ b/xdist/newhooks.py
@@ -31,8 +31,11 @@ def pytest_xdist_rsyncfinish(source, gateways):
 
 
 @pytest.mark.firstresult
-def pytest_xdist_getremotemodule():
-    """ called when creating remote node"""
+def pytest_xdist_getremotetarget():
+    """
+    called to get the remote target.
+    :return: a workermanager.RemoteTarget
+    """
 
 
 def pytest_configure_node(node):
diff --git a/xdist/workermanage.py b/xdist/workermanage.py
index a25243a..a3fc382 100644
--- a/xdist/workermanage.py
+++ b/xdist/workermanage.py
@@ -3,6 +3,7 @@ import fnmatch
 import os
 import re
 import threading
+import types
 
 import py
 import pytest
@@ -205,8 +206,8 @@ class WorkerController(object):
 
     class RemoteHook:
         @pytest.mark.trylast
-        def pytest_xdist_getremotemodule(self):
-            return xdist.remote
+        def pytest_xdist_getremotetarget(self):
+            return RemoteTarget(xdist.remote)
 
     def __init__(self, nodemanager, gateway, config, putevent):
         config.pluginmanager.register(self.RemoteHook())
@@ -248,8 +249,8 @@ class WorkerController(object):
                 basetemp = self.config._tmpdirhandler.getbasetemp()
                 option_dict["basetemp"] = str(basetemp.join(name))
         self.config.hook.pytest_configure_node(node=self)
-        remote_module = self.config.hook.pytest_xdist_getremotemodule()
-        self.channel = self.gateway.remote_exec(remote_module)
+        target, _, target_kwargs = self.config.hook.pytest_xdist_getremotetarget().pack()
+        self.channel = self.gateway.remote_exec(target, **target_kwargs)
         self.channel.send((self.workerinput, args, option_dict))
         if self.putevent:
             self.channel.setcallback(self.process_from_remote, endmarker=self.ENDMARK)
@@ -359,6 +360,23 @@ class WorkerController(object):
             self.notify_inproc("errordown", node=self, error=excinfo)
 
 
+class RemoteTarget:
+    def __init__(self, target, *args, **kwargs):
+        self._target = target
+        self._args = args
+        self._kwargs = kwargs
+        self.validate()
+
+    def validate(self):
+        # this is based on execnet criteria
+        assert not self._args, "Positional arguments are not yet supported"
+        if isinstance(self._target, types.ModuleType):
+            assert not (self._args or self._kwargs), "Arguments are not used with module targets"
+
+    def pack(self):
+        return self._target, self._args, self._kwargs
+
+
 def unserialize_report(name, reportdict):
     def assembled_report(reportdict):
         from _pytest._code.code import (