unittests: add a MockClient creator with a registry_listener

Change-Id: I1fb26cced4a9032fd5cced0ae8a36580fb6adcae

diff --git a/unittests/mockclient.cpp b/unittests/mockclient.cpp
index 501b405..a225a9a 100644 (file)
--- a/unittests/mockclient.cpp
+++ b/unittests/mockclient.cpp
@@ -36,6 +36,24 @@ MockClient::MockClient()
                std::cout << "fail to connect display\n";
 }
 
+MockClient::MockClient(const struct wl_registry_listener *registry_listener, void *data)
+{
+       int ret;
+       display = wl_display_connect("test-wl-extension-socket");
+       if (!display)
+               std::cout << "fail to connect display\n";
+
+       registry = wl_display_get_registry(display);
+       if (!registry)
+               std::cout << "fail to get registry\n";
+
+       ret = wl_registry_add_listener(registry, registry_listener, data);
+       if (ret)
+               std::cout << "fail to add resistry listener\n";
+
+       wl_display_roundtrip(display);
+}
+
 MockClient::~MockClient()
 {
        wl_display_disconnect(display);

diff --git a/unittests/mockclient.h b/unittests/mockclient.h
index bf59a8d..8c1e9b6 100644 (file)
--- a/unittests/mockclient.h
+++ b/unittests/mockclient.h
@@ -35,11 +35,13 @@ class MockClient
 {
 public:
        MockClient();
+       MockClient(const struct wl_registry_listener *registry_listener, void *data);
        virtual ~MockClient();
        void RoundTrip();
 
 private:
        struct wl_display *display;
+       struct wl_registry *registry;
 };
 
 #endif