/*
 * Copyright © Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 or 3 as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <boost/throw_exception.hpp>
#include <mir/graphics/display_configuration.h>
#include <mir/graphics/display.h>
#include <mir/graphics//default_display_configuration_policy.h>
#include <mir/time/steady_clock.h>
#include <mir/glib_main_loop.h>
#include "src/platforms/gbm-kms/server/kms/platform.h"
#include "src/server/report/null_report_factory.h"
#include <mir/options/program_option.h>
#include "src/platforms/gbm-kms/server/kms/quirks.h"

#include <mir/test/signal.h>
#include <mir/test/auto_unblock_thread.h>

#include <mir/test/doubles/mock_egl.h>
#include <mir/test/doubles/mock_gl.h>
#include <mir/test/doubles/mock_drm.h>
#include <mir/test/doubles/mock_gbm.h>
#include <mir/test/doubles/null_emergency_cleanup.h>
#include <mir/test/doubles/stub_console_services.h>
#include <mir/test/doubles/stub_gl_config.h>

#include <mir_test_framework/udev_environment.h>

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include <stdexcept>
#include <fcntl.h>

namespace mg = mir::graphics;
namespace mgg = mir::graphics::gbm;
namespace geom = mir::geometry;
namespace mt  = mir::test;
namespace mtd = mir::test::doubles;
namespace mtf = mir_test_framework;

namespace
{

mg::DisplayConfigurationMode conf_mode_from_drm_mode(drmModeModeInfo const& mode)
{
    geom::Size const size{mode.hdisplay, mode.vdisplay};
    double vrefresh_hz{0.0};

    /* Calculate vertical refresh rate from DRM mode information */
    if (mode.htotal != 0.0 && mode.vtotal != 0.0)
    {
        vrefresh_hz = (mode.clock * 100000LL /
                       ((long)mode.htotal * (long)mode.vtotal)) / 100.0;
    }

    return mg::DisplayConfigurationMode{size, vrefresh_hz};
}

struct MainLoop
{
    MainLoop()
    {
        int const owner{0};
        mt::Signal mainloop_started;
        ml.enqueue(&owner, [&] { mainloop_started.raise(); });
        mt::AutoUnblockThread t([this]{ml.stop();}, [this]{ml.run();});
        bool const started = mainloop_started.wait_for(std::chrono::seconds(10));
        if (!started)
            throw std::runtime_error("Failed to start main loop");

        ml_thread = std::move(t);
    }
    mir::GLibMainLoop ml{std::make_shared<mir::time::SteadyClock>()};
    mt::AutoUnblockThread ml_thread;
};

class MesaDisplayConfigurationTest : public ::testing::Test
{
public:
    MesaDisplayConfigurationTest()
        : drm_fd{open(drm_device, 0, 0)}
    {
        using namespace testing;

        /* Needed for display start-up */
        ON_CALL(mock_egl, eglChooseConfig(_, _, _, 1, _))
            .WillByDefault(DoAll(SetArgPointee<2>(mock_egl.fake_configs[0]),
                                 SetArgPointee<4>(1),
                                 Return(EGL_TRUE)));
        ON_CALL(mock_egl, eglGetConfigAttrib(_, mock_egl.fake_configs[0], EGL_NATIVE_VISUAL_ID, _))
            .WillByDefault(
                DoAll(
                    SetArgPointee<3>(GBM_FORMAT_XRGB8888),
                        Return(EGL_TRUE)));

        mock_egl.provide_egl_extensions();
        mock_gl.provide_gles_extensions();

        ON_CALL(mock_gbm, gbm_device_get_fd(_))
            .WillByDefault(Return(drm_fd));

        setup_sample_modes();

        fake_devices.add_standard_device("standard-drm-devices");

        // Remove all outputs from all but one of the DRM devices we access;
        // these tests are not set up to test hybrid.
        mock_drm.reset("/dev/dri/card1");
        mock_drm.reset("/dev/dri/card2");
    }

    std::shared_ptr<mg::DisplayPlatform> create_platform()
    {
        mir::udev::Context ctx;
        // Caution: non-local state!
        // This works because standard-drm-devices contains a udev device with 226:0 and devnode /dev/dri/card0
        auto device = ctx.char_device_from_devnum(makedev(226, 0));

        return std::make_shared<mgg::Platform>(
            *device,
            mir::report::null_display_report(),
            *std::make_shared<mtd::StubConsoleServices>(),
            *std::make_shared<mtd::NullEmergencyCleanup>(),
            mgg::BypassOption::allowed);
    }

    std::shared_ptr<mg::Display> create_display(
        std::shared_ptr<mg::DisplayPlatform> const& platform)
    {
        return platform->create_display(
            std::make_shared<mg::CloneDisplayConfigurationPolicy>(),
            std::make_shared<mtd::StubGLConfig>());
    }

    void setup_sample_modes()
    {
        using fake = mtd::FakeDRMResources;

        /* Add DRM modes */
        modes0.push_back(fake::create_mode(1920, 1080, 138500, 2080, 1111, fake::NormalMode));
        modes0.push_back(fake::create_mode(1920, 1080, 148500, 2200, 1125, fake::PreferredMode));
        modes0.push_back(fake::create_mode(1680, 1050, 119000, 1840, 1080, fake::NormalMode));
        modes0.push_back(fake::create_mode(832, 624, 57284, 1152, 667, fake::NormalMode));

        /* Add the DisplayConfiguration modes corresponding to the DRM modes */
        for (auto const& mode : modes0)
            conf_modes0.push_back(conf_mode_from_drm_mode(mode));

        modes1.push_back(fake::create_mode(123, 456, 138500, 2080, 1111, fake::NormalMode));
        modes1.push_back(fake::create_mode(789, 1011, 148500, 2200, 1125, fake::NormalMode));
        modes1.push_back(fake::create_mode(1213, 1415, 119000, 1840, 1080, fake::PreferredMode));

        for (auto const& mode : modes1)
            conf_modes1.push_back(conf_mode_from_drm_mode(mode));
    }

    ::testing::NiceMock<mtd::MockEGL> mock_egl;
    ::testing::NiceMock<mtd::MockGL> mock_gl;
    ::testing::NiceMock<mtd::MockDRM> mock_drm;
    ::testing::NiceMock<mtd::MockGBM> mock_gbm;

    std::vector<drmModeModeInfo> modes0;
    std::vector<mg::DisplayConfigurationMode> conf_modes0;
    std::vector<drmModeModeInfo> modes1;
    std::vector<mg::DisplayConfigurationMode> conf_modes1;
    std::vector<drmModeModeInfo> modes_empty;

    mtf::UdevEnvironment fake_devices;

    char const* const drm_device = "/dev/dri/card0";
    int const drm_fd;
};

MATCHER_P(Unique, scratch_vector, "")
{
    if (std::any_of(
        scratch_vector.begin(),
        scratch_vector.end(),
        [&arg](auto const& candidate) { return arg == candidate; }))
    {
        return false;
    }
    const_cast<typename std::remove_const<scratch_vector_type>::type&>(scratch_vector).push_back(arg);
    return true;
}

MATCHER_P(DisplayConfigsAreEquivalent, expected_configs, "")
{
    using namespace testing;

    std::vector<mg::DisplayConfigurationOutput> outputs;
    std::vector<mg::DisplayConfigurationOutputId> output_ids;
    arg->for_each_output([&](mg::DisplayConfigurationOutput const& output)
                          {
                              outputs.push_back(output);
                              output_ids.push_back(output.id);
                              outputs.back().id = mg::DisplayConfigurationOutputId{0};
                          });


    Matcher<decltype(outputs)> config_matcher = UnorderedElementsAreArray(expected_configs);
    std::vector<mg::DisplayConfigurationOutputId> scratch_space;
    Matcher<decltype(output_ids)> ids_are_unique = Each(Unique(scratch_space));

    return
        config_matcher.MatchAndExplain(outputs, result_listener) &&
        ids_are_unique.MatchAndExplain(output_ids, result_listener);
}
}

TEST_F(MesaDisplayConfigurationTest, configuration_is_read_correctly)
{
    using namespace ::testing;

    /* Set up DRM resources */
    uint32_t const invalid_id{0};
    uint32_t const crtc0_id{10};
    uint32_t const encoder0_id{20};
    uint32_t const encoder1_id{21};
    uint32_t const connector0_id{30};
    uint32_t const connector1_id{31};
    uint32_t const connector2_id{32};
    geom::Size const connector0_physical_size_mm{480, 270};
    geom::Size const connector1_physical_size_mm{};
    geom::Size const connector2_physical_size_mm{};
    std::vector<uint32_t> possible_encoder_ids_empty;
    uint32_t const possible_crtcs_mask_empty{0};

    mock_drm.reset(drm_device);

    mock_drm.add_crtc(
        drm_device,
        crtc0_id,
        modes0[1]);

    mock_drm.add_encoder(
        drm_device,
        encoder0_id,
        crtc0_id,
        possible_crtcs_mask_empty);
    mock_drm.add_encoder(
        drm_device,
        encoder1_id,
        invalid_id,
        possible_crtcs_mask_empty);

    mock_drm.add_connector(
        drm_device,
        connector0_id,
        DRM_MODE_CONNECTOR_HDMIA,
        DRM_MODE_CONNECTED,
        encoder0_id,
        modes0,
        possible_encoder_ids_empty,
        connector0_physical_size_mm);
    mock_drm.add_connector(
        drm_device,
        connector1_id,
        DRM_MODE_CONNECTOR_Unknown,
        DRM_MODE_DISCONNECTED,
        invalid_id,
        modes_empty,
        possible_encoder_ids_empty,
        connector1_physical_size_mm);
    mock_drm.add_connector(
        drm_device,
        connector2_id,
        DRM_MODE_CONNECTOR_eDP,
        DRM_MODE_DISCONNECTED,
        encoder1_id,
        modes_empty,
        possible_encoder_ids_empty,
        connector2_physical_size_mm);

    mock_drm.prepare(drm_device);

    std::vector<mg::DisplayConfigurationOutput> const expected_outputs =
    {
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::hdmia,
            {},
            conf_modes0,
            1,
            connector0_physical_size_mm,
            true,
            true,
            geom::Point(),
            1,
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::unknown,
            {},
            std::vector<mg::DisplayConfigurationMode>(),
            std::numeric_limits<uint32_t>::max(),
            connector1_physical_size_mm,
            false,
            false,
            geom::Point(),
            std::numeric_limits<uint32_t>::max(),
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::edp,
            {},
            std::vector<mg::DisplayConfigurationMode>(),
            std::numeric_limits<uint32_t>::max(),
            connector2_physical_size_mm,
            false,
            false,
            geom::Point(),
            std::numeric_limits<uint32_t>::max(),
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        }
    };

    /* Test body */
    auto display = create_display(create_platform());

    auto conf = display->configuration();

    EXPECT_THAT(conf, DisplayConfigsAreEquivalent(expected_outputs));
}

TEST_F(MesaDisplayConfigurationTest, reads_subpixel_information_correctly)
{
    using namespace ::testing;

    /* Set up DRM resources */
    uint32_t const crtc0_id{10};
    uint32_t const encoder0_id{20};
    uint32_t const connector0_id{30};
    geom::Size const connector0_physical_size_mm{480, 270};
    std::vector<uint32_t> possible_encoder_ids_empty;
    uint32_t const possible_crtcs_mask_empty{0};

    struct TestData
    {
        drmModeSubPixel drm_subpixel;
        MirSubpixelArrangement mir_subpixel;
    };
    std::vector<TestData> const test_data = {
        { DRM_MODE_SUBPIXEL_UNKNOWN, mir_subpixel_arrangement_unknown },
        { DRM_MODE_SUBPIXEL_HORIZONTAL_RGB, mir_subpixel_arrangement_horizontal_rgb },
        { DRM_MODE_SUBPIXEL_HORIZONTAL_BGR, mir_subpixel_arrangement_horizontal_bgr },
        { DRM_MODE_SUBPIXEL_VERTICAL_RGB, mir_subpixel_arrangement_vertical_rgb },
        { DRM_MODE_SUBPIXEL_VERTICAL_BGR, mir_subpixel_arrangement_vertical_bgr },
        { DRM_MODE_SUBPIXEL_NONE, mir_subpixel_arrangement_none }
    };

    for (auto& data : test_data)
    {

        mock_drm.reset(drm_device);

        mock_drm.add_crtc(
            drm_device,
            crtc0_id,
            modes0[1]);

        mock_drm.add_encoder(
            drm_device,
            encoder0_id,
            crtc0_id,
            possible_crtcs_mask_empty);

        mock_drm.add_connector(
            drm_device,
            connector0_id,
            DRM_MODE_CONNECTOR_HDMIA,
            DRM_MODE_CONNECTED,
            encoder0_id,
            modes0,
            possible_encoder_ids_empty,
            connector0_physical_size_mm,
            data.drm_subpixel);

        mock_drm.prepare(drm_device);

        /* Test body */
        auto display = create_display(create_platform());

        auto conf = display->configuration();


        int output_count{0};

        conf->for_each_output([&](mg::DisplayConfigurationOutput const& output)
                              {
                                  EXPECT_THAT(output.subpixel_arrangement, Eq(data.mir_subpixel));
                                  ++output_count;
                              });

        EXPECT_THAT(output_count, Ge(1));
    }
}

TEST_F(MesaDisplayConfigurationTest, reads_updated_subpixel_information)
{
    using namespace ::testing;
    using namespace std::chrono_literals;

    /* Set up DRM resources */
    uint32_t const crtc0_id{10};
    uint32_t const encoder0_id{20};
    uint32_t const connector0_id{30};
    geom::Size const connector0_physical_size_mm{480, 270};
    std::vector<uint32_t> possible_encoder_ids_empty;
    uint32_t const possible_crtcs_mask_empty{0};

    mock_drm.reset(drm_device);

    mock_drm.add_crtc(
        drm_device,
        crtc0_id,
        modes0[1]);

    mock_drm.add_encoder(
        drm_device,
        encoder0_id,
        crtc0_id,
        possible_crtcs_mask_empty);

    mock_drm.add_connector(
        drm_device,
        connector0_id,
        DRM_MODE_CONNECTOR_HDMIA,
        DRM_MODE_CONNECTED,
        encoder0_id,
        modes0,
        possible_encoder_ids_empty,
        connector0_physical_size_mm,
        DRM_MODE_SUBPIXEL_NONE);

    mock_drm.prepare(drm_device);

    struct TestData
    {
        drmModeSubPixel drm_subpixel;
        MirSubpixelArrangement mir_subpixel;
    };
    std::vector<TestData> const test_data = {
        { DRM_MODE_SUBPIXEL_UNKNOWN, mir_subpixel_arrangement_unknown },
        { DRM_MODE_SUBPIXEL_HORIZONTAL_RGB, mir_subpixel_arrangement_horizontal_rgb },
        { DRM_MODE_SUBPIXEL_HORIZONTAL_BGR, mir_subpixel_arrangement_horizontal_bgr },
        { DRM_MODE_SUBPIXEL_VERTICAL_RGB, mir_subpixel_arrangement_vertical_rgb },
        { DRM_MODE_SUBPIXEL_VERTICAL_BGR, mir_subpixel_arrangement_vertical_bgr },
        { DRM_MODE_SUBPIXEL_NONE, mir_subpixel_arrangement_none }
    };

    auto display = create_display(create_platform());
    auto const syspath = fake_devices.add_device("drm", "card2", nullptr, {}, {"DEVTYPE", "drm_minor"});

    for (auto& data : test_data)
    {

        mock_drm.reset(drm_device);

        mock_drm.add_crtc(
            drm_device,
            crtc0_id,
            modes0[1]);

        mock_drm.add_encoder(
            drm_device,
            encoder0_id,
            crtc0_id,
            possible_crtcs_mask_empty);

        mock_drm.add_connector(
            drm_device,
            connector0_id,
            DRM_MODE_CONNECTOR_HDMIA,
            DRM_MODE_CONNECTED,
            encoder0_id,
            modes0,
            possible_encoder_ids_empty,
            connector0_physical_size_mm,
            data.drm_subpixel);

        mock_drm.prepare(drm_device);

        mt::Signal handler_signal;
        MainLoop ml;
        display->register_configuration_change_handler(ml.ml, [&handler_signal]{handler_signal.raise();});
        fake_devices.emit_device_changed(syspath);
        ASSERT_TRUE(handler_signal.wait_for(10s));


        /* Test body */
        auto conf = display->configuration();


        int output_count{0};

        conf->for_each_output([&](mg::DisplayConfigurationOutput const& output)
                              {
                                  EXPECT_THAT(output.subpixel_arrangement, Eq(data.mir_subpixel));
                                  ++output_count;
                              });

        EXPECT_THAT(output_count, Ge(1));
    }
}

TEST_F(MesaDisplayConfigurationTest, returns_updated_configuration)
{
    using namespace ::testing;
    using namespace std::chrono_literals;

    uint32_t const invalid_id{0};
    std::vector<uint32_t> const crtc_ids{10, 11};
    std::vector<uint32_t> const encoder_ids{20, 21};
    std::vector<uint32_t> const connector_ids{30, 31};
    std::vector<geom::Size> const connector_physical_sizes_mm_before{
        {480, 270}, {}
    };
    std::vector<geom::Size> const connector_physical_sizes_mm_after{
        {}, {512, 642}
    };
    std::vector<uint32_t> possible_encoder_ids_empty;
    uint32_t const possible_crtcs_mask_empty{0};

    std::vector<mg::DisplayConfigurationOutput> const expected_outputs_before =
    {
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::composite,
            {},
            conf_modes0,
            1,
            connector_physical_sizes_mm_before[0],
            true,
            true,
            geom::Point(),
            1,
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::vga,
            {},
            std::vector<mg::DisplayConfigurationMode>(),
            std::numeric_limits<uint32_t>::max(),
            connector_physical_sizes_mm_before[1],
            false,
            false,
            geom::Point(),
            std::numeric_limits<uint32_t>::max(),
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
    };

    std::vector<mg::DisplayConfigurationOutput> const expected_outputs_after =
    {
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::composite,
            {},
            std::vector<mg::DisplayConfigurationMode>(),
            std::numeric_limits<uint32_t>::max(),
            connector_physical_sizes_mm_after[0],
            false,
            true,
            geom::Point(),
            std::numeric_limits<uint32_t>::max(),
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::vga,
            {},
            conf_modes0,
            1,
            connector_physical_sizes_mm_after[1],
            true,
            false,
            geom::Point(),
            1,
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
    };

    /* Set up DRM resources and check */
    auto const syspath = fake_devices.add_device(
        "drm",
        "card2",
        nullptr,
        {},
        {
            "DEVTYPE", "drm_minor",
            "DEVNAME", "/dev/dri/card2",
            "MAJOR", "226",
            "MINOR", "2"
        });

    mock_drm.reset(drm_device);

    mock_drm.add_crtc(
        drm_device,
        crtc_ids[0],
        modes0[1]);

    mock_drm.add_encoder(
        drm_device,
        encoder_ids[0],
        crtc_ids[0],
        possible_crtcs_mask_empty);
    mock_drm.add_encoder(
        drm_device,
        encoder_ids[1],
        invalid_id,
        possible_crtcs_mask_empty);

    mock_drm.add_connector(
        drm_device,
        connector_ids[0],
        DRM_MODE_CONNECTOR_Composite,
        DRM_MODE_CONNECTED,
        encoder_ids[0],
        modes0,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_before[0]);
    mock_drm.add_connector(
        drm_device,
        connector_ids[1],
        DRM_MODE_CONNECTOR_VGA,
        DRM_MODE_DISCONNECTED,
        invalid_id,
        modes_empty,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_before[1]);

    mock_drm.prepare(drm_device);

    auto display = create_display(create_platform());

    auto conf = display->configuration();

    EXPECT_THAT(conf, DisplayConfigsAreEquivalent(expected_outputs_before));

    /* Reset DRM resources and check again */
    mock_drm.reset(drm_device);

    mock_drm.add_crtc(
        drm_device,
        crtc_ids[1],
        modes0[1]);

    mock_drm.add_encoder(
        drm_device,
        encoder_ids[0],
        invalid_id,
        possible_crtcs_mask_empty);
    mock_drm.add_encoder(
        drm_device,
        encoder_ids[1],
        crtc_ids[1],
        possible_crtcs_mask_empty);

    mock_drm.add_connector(
        drm_device,
        connector_ids[0],
        DRM_MODE_CONNECTOR_Composite,
        DRM_MODE_DISCONNECTED,
        invalid_id,
        modes_empty,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_after[0]);
    mock_drm.add_connector(
        drm_device,
        connector_ids[1],
        DRM_MODE_CONNECTOR_VGA,
        DRM_MODE_CONNECTED,
        encoder_ids[1],
        modes0,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_after[1]);

    mock_drm.prepare(drm_device);

    /* Fake a device change so display can fetch updated configuration*/
    MainLoop ml;
    mt::Signal handler_signal;
    display->register_configuration_change_handler(ml.ml, [&handler_signal]{handler_signal.raise();});
    fake_devices.emit_device_changed(syspath);
    ASSERT_TRUE(handler_signal.wait_for(10s));

    conf = display->configuration();

    EXPECT_THAT(conf, DisplayConfigsAreEquivalent(expected_outputs_after));
}

TEST_F(MesaDisplayConfigurationTest, new_monitor_matches_hardware_state)
{
    using namespace ::testing;
    using namespace std::chrono_literals;

    uint32_t const crtc_ids[1] = {10};
    uint32_t const encoder_ids[2] = {20, 21};
    uint32_t const connector_ids[1] = {30};
    geom::Size const connector_physical_sizes_mm_before{480, 270};
    geom::Size const connector_physical_sizes_mm_after{512, 642};
    std::vector<uint32_t> possible_encoder_ids_empty;
    uint32_t const possible_crtcs_mask_empty{0};

    std::vector<mg::DisplayConfigurationOutput> const expected_outputs_before =
    {
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::composite,
            {},
            conf_modes0,
            1,
            connector_physical_sizes_mm_before,
            true,
            true,
            geom::Point(),
            1,
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
    };

    std::vector<mg::DisplayConfigurationOutput> const expected_outputs_after =
    {
        {
            mg::DisplayConfigurationOutputId{0},
            mg::DisplayConfigurationCardId{0},
            mg::DisplayConfigurationLogicalGroupId{0},
            mg::DisplayConfigurationOutputType::composite,
            {},
            conf_modes1,
            2,
            connector_physical_sizes_mm_after,
            true,
            true,
            geom::Point(),
            std::numeric_limits<uint32_t>::max(),   // The new state doesn't have a CRTC
                                                    // associated, so doesn't have a current mode.
                                                    // We should mirror the hardware state.
            mir_pixel_format_invalid,
            mir_power_mode_on,
            mir_orientation_normal,
            1.0f,
            mir_form_factor_monitor,
            mir_subpixel_arrangement_unknown,
            {},
            mir_output_gamma_unsupported,
            {},
            {}
        },
    };

    auto const syspath = fake_devices.add_device(
        "drm",
        "card2",
        nullptr,
        {},
        {
            "DEVTYPE", "drm_minor",
            "DEVNAME", "/dev/dri/card2",
            "MAJOR", "226",
            "MINOR", "2"
        });

    mock_drm.reset(drm_device);
    mock_drm.add_crtc(
        drm_device,
        crtc_ids[0],
        modes0[1]);
    mock_drm.add_encoder(
        drm_device,
        encoder_ids[0],
        crtc_ids[0],
        possible_crtcs_mask_empty);
    mock_drm.add_connector(
        drm_device,
        connector_ids[0],
        DRM_MODE_CONNECTOR_Composite,
        DRM_MODE_CONNECTED,
        encoder_ids[0],
        modes0,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_before);
    mock_drm.prepare(drm_device);

    auto display = create_display(create_platform());
    auto conf = display->configuration();
    EXPECT_THAT(conf, DisplayConfigsAreEquivalent(expected_outputs_before));

    // Now simulate a change of monitor, with no CRTC attached (and hence no current mode).
    // The configuration should mirror this state.
    mock_drm.reset(drm_device);
    mock_drm.add_crtc(
        drm_device,
        crtc_ids[0],
        modes1[1]);
    mock_drm.add_encoder(
        drm_device,
        encoder_ids[0],
        crtc_ids[0],
        possible_crtcs_mask_empty);
    mock_drm.add_encoder(
        drm_device,
        encoder_ids[1],
        0,
        possible_crtcs_mask_empty);
    mock_drm.add_connector(
        drm_device,
        connector_ids[0],
        DRM_MODE_CONNECTOR_Composite,
        DRM_MODE_CONNECTED,
        encoder_ids[1],
        modes1,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm_after);
    mock_drm.prepare(drm_device);

    MainLoop ml;
    mt::Signal handler_signal;
    display->register_configuration_change_handler(ml.ml, [&handler_signal]{handler_signal.raise();});
    fake_devices.emit_device_changed(syspath);
    ASSERT_TRUE(handler_signal.wait_for(10s));

    conf = display->configuration();
    EXPECT_THAT(conf, DisplayConfigsAreEquivalent(expected_outputs_after));
}

TEST_F(MesaDisplayConfigurationTest, does_not_query_drm_unnecesarily)
{
    using namespace ::testing;
    using namespace std::chrono_literals;

    uint32_t const crtc_id{10};
    uint32_t const encoder_id{20};
    uint32_t const connector_id{30};
    geom::Size const connector_physical_sizes_mm{480, 270};
    std::vector<uint32_t> possible_encoder_ids_empty;

    uint32_t const possible_crtcs_mask_empty{0};
    mock_drm.reset(drm_device);
    mock_drm.add_crtc(
        drm_device,
        crtc_id,
        modes0[1]);
    mock_drm.add_encoder(
        drm_device,
        encoder_id,
        crtc_id,
        possible_crtcs_mask_empty);
    mock_drm.add_connector(
        drm_device,
        connector_id,
        DRM_MODE_CONNECTOR_Composite,
        DRM_MODE_CONNECTED,
        encoder_id,
        modes0,
        possible_encoder_ids_empty,
        connector_physical_sizes_mm);
    mock_drm.prepare(drm_device);

    auto const syspath = fake_devices.add_device(
        "drm",
        "card2",
        nullptr,
        {},
        {
            "DEVTYPE", "drm_minor",
            "DEVNAME", "/dev/dri/card2",
            "MAJOR", "226",
            "MINOR", "2"
        });

    auto display = create_display(create_platform());

    // No display change reported yet so display should not query drm
    EXPECT_CALL(mock_drm, drmModeGetConnector(_,_)).Times(0);
    display->configuration();

    // Now signal a device change
    MainLoop ml;
    mt::Signal handler_signal;
    display->register_configuration_change_handler(ml.ml, [&handler_signal]{handler_signal.raise();});
    fake_devices.emit_device_changed(syspath);
    ASSERT_TRUE(handler_signal.wait_for(10s));

    // It needs to query DRM at least once; we don't really care how many times it does, though.
    EXPECT_CALL(mock_drm, drmModeGetConnector(_,_)).Times(AtLeast(1));
    display->configuration();
}
