test: add SetDisplayString()

Don't rely on the environment alone, take the display string if it's been
set.

Signed-off-by: Peter Hutterer <peter.hutterer@who-t.net>
Reviewed-by: Chase Douglas <chase.douglas@canonical.com>

1 files changed, 10 insertions, 0 deletions

diff --git a/include/xorg/gtest/xorg-gtest-test.h b/include/xorg/gtest/xorg-gtest-test.h
index 3b78a17..a776693 100644
--- a/include/xorg/gtest/xorg-gtest-test.h
+++ b/include/xorg/gtest/xorg-gtest-test.h
@@ -88,6 +88,16 @@ class Test : public ::testing::Test {
    */
   ::Display* Display() const;
 
+  /**
+   * Set the display string used for XOpenDisplay() and thus affects
+   * Test::Display(). This function must be called before
+   * xorg::testing::Test::SetUp() to have any effect.
+   *
+   * @param display The string representing the display connection, or an
+   * empty string for NULL
+   */
+  void SetDisplayString(const std::string &display);
+
   /** @cond Implementation */
   struct Private;
   std::auto_ptr<Private> d_;