/*
 * test-gconfig.c: unit tests for libvirt-gconfig
 *
 * Copyright (C) 2014 Red Hat, Inc.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library. If not, see
 * <http://www.gnu.org/licenses/>.
 *
 * Author: Christophe Fergeau <cfergeau@redhat.com>
 */

#include <config.h>

#include <libvirt-gconfig/libvirt-gconfig.h>

#define g_assert_cmpstr_free1(str1, op, str2) G_STMT_START { \
    char *alloced_str = (str1); \
    g_assert_cmpstr(alloced_str, op, (str2)); \
    g_free(alloced_str); \
} G_STMT_END

static char * load_xml(const char *reference_file)
{
    const char *reference_path;
    GError *error = NULL;
    char *reference_xml;

    reference_path = g_test_get_filename(G_TEST_DIST, "xml",
                                         reference_file, NULL);
    g_file_get_contents(reference_path, &reference_xml, NULL, &error);
    g_assert_no_error(error);
    /* I could not generate text files without a trailing \n with vim or
     * gedit, workaround this issue by removing trailing whitespace from
     * the reference file */
    g_strchomp(reference_xml);

    return reference_xml;
}


static void check_xml(GVirConfigDomain *domain, const char *reference_file)
{
    char *reference_xml;
    char *xml;

    reference_xml = load_xml(reference_file);

    xml = gvir_config_object_to_xml(GVIR_CONFIG_OBJECT(domain));
    g_assert_cmpstr(xml, ==, reference_xml);
    g_free(xml);
    g_free(reference_xml);
}


static GList *get_devices_by_type(GVirConfigDomain *domain, GType type)
{
    GList *devices;
    GList *it;
    GList *matches = NULL;

    devices = gvir_config_domain_get_devices(domain);
    for (it = devices; it != NULL; it = it->next) {
        if (g_type_is_a(G_OBJECT_TYPE(it->data), type)) {
            matches = g_list_prepend(matches, g_object_ref(G_OBJECT(it->data)));
        }
    }
    g_list_free_full(devices, g_object_unref);

    return g_list_reverse(matches);
}


static void test_domain(void)
{
    GVirConfigDomain *domain;
    const char *features[] = { "foo", "bar", "baz", NULL };
    GStrv feat;
    unsigned int i;

    domain = gvir_config_domain_new();
    g_assert(domain != NULL);

    gvir_config_domain_set_virt_type(domain, GVIR_CONFIG_DOMAIN_VIRT_KVM);
    g_assert_cmpint(gvir_config_domain_get_virt_type(domain), ==, GVIR_CONFIG_DOMAIN_VIRT_KVM);

    gvir_config_domain_set_name(domain, "foo");
    g_assert_cmpstr(gvir_config_domain_get_name(domain), ==, "foo");

    gvir_config_domain_set_memory(domain, 1234);
    g_assert_cmpint(gvir_config_domain_get_memory(domain), ==, 1234);
    gvir_config_domain_set_vcpus(domain, 3);
    g_assert_cmpint(gvir_config_domain_get_vcpus(domain), ==, 3);

    gvir_config_domain_set_features(domain, (const GStrv)features);
    feat = gvir_config_domain_get_features(domain);
    for (i = 0; features[i] != NULL; i++) {
        g_assert(feat[i] != NULL);
        g_assert_cmpstr(feat[i], ==, features[i]);
    }
    g_strfreev(feat);

    gvir_config_domain_set_lifecycle(domain,
                                     GVIR_CONFIG_DOMAIN_LIFECYCLE_ON_POWEROFF,
                                     GVIR_CONFIG_DOMAIN_LIFECYCLE_RESTART);

    gvir_config_domain_set_custom_xml_ns_children(domain, "<libosinfo><os id=\"http://centos.org/centos/7\"/></libosinfo>", "libosinfo", "http://libosinfo.org/xmlns/libvirt/domain/1.0", NULL);
    gvir_config_domain_set_custom_xml_ns_children(domain, "<instance><flavor name=\"m1.small\"><memory>512</memory><disk>10</disk></flavor></instance>", "nova", "http://openstack.org/nova/instance/1", NULL);
    gvir_config_domain_set_custom_xml(domain, "<gnome-boxes><os-state>installed</os-state></gnome-boxes>", "boxes", "https://wiki.gnome.org/Apps/Boxes", NULL);
    gvir_config_domain_set_custom_xml(domain, "<foo/>", "ns", "http://foo", NULL);
    gvir_config_domain_set_custom_xml(domain, "<foo/>", "nsbar", "http://bar", NULL);
    gvir_config_domain_set_custom_xml(domain, "<foo/>", "ns", "http://bar", NULL);
    gvir_config_domain_set_custom_xml(domain, "<bar/>", "ns", "http://foo", NULL);

    g_assert_cmpstr_free1(gvir_config_domain_get_custom_xml(domain, "http://foo"), ==, "<ns:bar xmlns:ns=\"http://foo\"/>");
    g_assert_cmpstr_free1(gvir_config_domain_get_custom_xml(domain, "http://bar"), ==, "<ns:foo xmlns:ns=\"http://bar\"/>");

    check_xml(domain, "gconfig-domain.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_clock(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainClock *klock;
    GVirConfigDomainTimerPit *pit;
    GVirConfigDomainTimerRtc *rtc;
    GVirConfigDomainTimerHpet *hpet;

    domain = gvir_config_domain_new();

    klock = gvir_config_domain_clock_new();
    gvir_config_domain_clock_set_offset(klock, GVIR_CONFIG_DOMAIN_CLOCK_TIMEZONE);
    gvir_config_domain_clock_set_timezone(klock, "CEST");
    g_assert_cmpint(gvir_config_domain_clock_get_offset(klock), ==, GVIR_CONFIG_DOMAIN_CLOCK_TIMEZONE);
    g_assert_cmpstr(gvir_config_domain_clock_get_timezone(klock), ==, "CEST");

    pit = gvir_config_domain_timer_pit_new();
    gvir_config_domain_timer_set_tick_policy(GVIR_CONFIG_DOMAIN_TIMER(pit),
                                             GVIR_CONFIG_DOMAIN_TIMER_TICK_POLICY_DELAY);
    gvir_config_domain_clock_add_timer(klock, GVIR_CONFIG_DOMAIN_TIMER(pit));
    g_assert_cmpint(gvir_config_domain_timer_get_tick_policy(GVIR_CONFIG_DOMAIN_TIMER(pit)), ==, GVIR_CONFIG_DOMAIN_TIMER_TICK_POLICY_DELAY);
    g_assert(gvir_config_domain_timer_get_present(GVIR_CONFIG_DOMAIN_TIMER(pit)) != FALSE);
    g_object_unref(G_OBJECT(pit));

    rtc = gvir_config_domain_timer_rtc_new();
    gvir_config_domain_timer_set_tick_policy(GVIR_CONFIG_DOMAIN_TIMER(rtc),
                                             GVIR_CONFIG_DOMAIN_TIMER_TICK_POLICY_CATCHUP);
    gvir_config_domain_clock_add_timer(klock, GVIR_CONFIG_DOMAIN_TIMER(rtc));
    g_assert_cmpint(gvir_config_domain_timer_get_tick_policy(GVIR_CONFIG_DOMAIN_TIMER(rtc)), ==, GVIR_CONFIG_DOMAIN_TIMER_TICK_POLICY_CATCHUP);
    g_assert(gvir_config_domain_timer_get_present(GVIR_CONFIG_DOMAIN_TIMER(rtc)) != FALSE);
    g_object_unref(G_OBJECT(rtc));

    hpet = gvir_config_domain_timer_hpet_new();
    gvir_config_domain_timer_set_present(GVIR_CONFIG_DOMAIN_TIMER(hpet), FALSE);
    gvir_config_domain_clock_add_timer(klock, GVIR_CONFIG_DOMAIN_TIMER(hpet));
    g_assert(gvir_config_domain_timer_get_present(GVIR_CONFIG_DOMAIN_TIMER(hpet)) == FALSE);
    g_object_unref(G_OBJECT(hpet));

    gvir_config_domain_set_clock(domain, klock);
    g_object_unref(G_OBJECT(klock));

    klock = gvir_config_domain_get_clock(domain);
    g_assert(klock != NULL);
    g_assert_cmpint(gvir_config_domain_clock_get_offset(klock), ==, GVIR_CONFIG_DOMAIN_CLOCK_TIMEZONE);
    g_assert_cmpstr(gvir_config_domain_clock_get_timezone(klock), ==, "CEST");
    g_object_unref(G_OBJECT(klock));
    /* FIXME: gvir_config_domain_clock_get_timers() */

    check_xml(domain, "gconfig-domain-clock.xml");

    gvir_config_domain_set_clock(domain, NULL);
    klock = gvir_config_domain_get_clock(domain);
    g_assert(klock == NULL);

    g_object_unref(G_OBJECT(domain));
}


static void domain_os_check_boot_devices(GVirConfigDomainOs *os)
{
    GList *devices = NULL;
    GList *it;

    devices = gvir_config_domain_os_get_boot_devices(os);
    it = devices;
    g_assert_cmpint(GPOINTER_TO_INT(it->data), ==, GVIR_CONFIG_DOMAIN_OS_BOOT_DEVICE_CDROM);
    it = it->next;
    g_assert_cmpint(GPOINTER_TO_INT(it->data), ==, GVIR_CONFIG_DOMAIN_OS_BOOT_DEVICE_NETWORK);
    g_assert(it->next == NULL);
    g_list_free(devices);
}


static void test_domain_os(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainOs *os;
    GList *devices = NULL;

    domain = gvir_config_domain_new();
    os = gvir_config_domain_os_new();
    gvir_config_domain_os_set_os_type(os, GVIR_CONFIG_DOMAIN_OS_TYPE_HVM);
    gvir_config_domain_os_set_arch(os, "x86_64");
    gvir_config_domain_os_set_machine(os, "q35");
    gvir_config_domain_os_set_firmware(os, GVIR_CONFIG_DOMAIN_OS_FIRMWARE_EFI);
    g_assert_cmpint(gvir_config_domain_os_get_os_type(os), ==, GVIR_CONFIG_DOMAIN_OS_TYPE_HVM);
    g_assert_cmpstr(gvir_config_domain_os_get_arch(os), ==, "x86_64");
    g_assert_cmpstr(gvir_config_domain_os_get_machine(os), ==, "q35");
    g_assert_cmpint(gvir_config_domain_os_get_firmware(os), ==, GVIR_CONFIG_DOMAIN_OS_FIRMWARE_EFI);
    devices = g_list_append(devices,
                             GINT_TO_POINTER(GVIR_CONFIG_DOMAIN_OS_BOOT_DEVICE_CDROM));
    devices = g_list_append(devices,
                            GINT_TO_POINTER(GVIR_CONFIG_DOMAIN_OS_BOOT_DEVICE_NETWORK));
    gvir_config_domain_os_set_boot_devices(os, devices);
    g_list_free(devices);
    devices = NULL;
    domain_os_check_boot_devices(os);
    gvir_config_domain_set_os(domain, os);
    g_object_unref(G_OBJECT(os));

    os = gvir_config_domain_get_os(domain);
    g_assert(os != NULL);
    g_assert_cmpstr(gvir_config_domain_os_get_arch(os), ==, "x86_64");
    g_assert_cmpint(gvir_config_domain_os_get_os_type(os), ==, GVIR_CONFIG_DOMAIN_OS_TYPE_HVM);
    g_assert_cmpstr(gvir_config_domain_os_get_machine(os), ==, "q35");
    g_assert_cmpint(gvir_config_domain_os_get_firmware(os), ==, GVIR_CONFIG_DOMAIN_OS_FIRMWARE_EFI);
    domain_os_check_boot_devices(os);
    g_object_unref(G_OBJECT(os));

    check_xml(domain, "gconfig-domain-os.xml");

    gvir_config_domain_set_os(domain, NULL);
    os = gvir_config_domain_get_os(domain);
    g_assert(os == NULL);

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_cpu(void)
{
    GVirConfigDomain *domain;
    /* cpu node */
    GVirConfigDomainCpu *cpu;

    domain = gvir_config_domain_new();
    cpu = gvir_config_domain_cpu_new();
    gvir_config_domain_set_vcpus(domain, 3);

    /* cpu/feature nodes */
    GVirConfigDomainCpuFeature *feature;
    GList *cpu_features = NULL;

    gvir_config_domain_cpu_set_match_policy(cpu, GVIR_CONFIG_DOMAIN_CPU_MATCH_POLICY_STRICT);
    gvir_config_domain_cpu_set_mode(cpu, GVIR_CONFIG_DOMAIN_CPU_MODE_HOST_PASSTHROUGH);
    feature = gvir_config_domain_cpu_feature_new();
    gvir_config_capabilities_cpu_feature_set_name(GVIR_CONFIG_CAPABILITIES_CPU_FEATURE(feature),
                                                  "foo");
    gvir_config_domain_cpu_feature_set_policy(feature, GVIR_CONFIG_DOMAIN_CPU_FEATURE_POLICY_REQUIRE);
    gvir_config_capabilities_cpu_add_feature(GVIR_CONFIG_CAPABILITIES_CPU(cpu),
                                             GVIR_CONFIG_CAPABILITIES_CPU_FEATURE(feature));
    g_object_unref(feature);

    /* cpu/topology nodes */
    GVirConfigCapabilitiesCpuTopology *topology;

    topology = gvir_config_capabilities_cpu_topology_new();
    gvir_config_capabilities_cpu_topology_set_cores(topology, 1);
    gvir_config_capabilities_cpu_topology_set_sockets(topology, 2);
    gvir_config_capabilities_cpu_topology_set_threads(topology, 3);
    gvir_config_capabilities_cpu_set_topology(GVIR_CONFIG_CAPABILITIES_CPU(cpu),
                                              topology);
    g_object_unref(topology);
    gvir_config_domain_set_cpu(domain, cpu);
    g_object_unref(cpu);


    cpu = gvir_config_domain_get_cpu(domain);
    g_assert(cpu != NULL);
    g_assert(GVIR_CONFIG_IS_CAPABILITIES_CPU(cpu));
    g_assert_cmpint(gvir_config_domain_cpu_get_match_policy(cpu), ==, GVIR_CONFIG_DOMAIN_CPU_MATCH_POLICY_STRICT);
    g_assert_cmpint(gvir_config_domain_cpu_get_mode(cpu), ==, GVIR_CONFIG_DOMAIN_CPU_MODE_HOST_PASSTHROUGH);

    cpu_features = gvir_config_capabilities_cpu_get_features(GVIR_CONFIG_CAPABILITIES_CPU(cpu));
    g_assert_cmpint(g_list_length(cpu_features), ==, 1);
    g_assert(GVIR_CONFIG_IS_DOMAIN_CPU_FEATURE(cpu_features->data));
    feature = GVIR_CONFIG_DOMAIN_CPU_FEATURE(cpu_features->data);
    g_assert_cmpstr(gvir_config_capabilities_cpu_feature_get_name(GVIR_CONFIG_CAPABILITIES_CPU_FEATURE(feature)), ==, "foo");
    g_assert_cmpint(gvir_config_domain_cpu_feature_get_policy(feature), ==, GVIR_CONFIG_DOMAIN_CPU_FEATURE_POLICY_REQUIRE);
    g_list_free_full(cpu_features, g_object_unref);

    topology = gvir_config_capabilities_cpu_get_topology(GVIR_CONFIG_CAPABILITIES_CPU(cpu));
    g_assert(topology != NULL);
    g_assert_cmpint(gvir_config_capabilities_cpu_topology_get_cores(topology), ==, 1);
    g_assert_cmpint(gvir_config_capabilities_cpu_topology_get_sockets(topology), ==, 2);
    g_assert_cmpint(gvir_config_capabilities_cpu_topology_get_threads(topology), ==, 3);
    g_object_unref(topology);
    g_object_unref(cpu);

    check_xml(domain, "gconfig-domain-cpu.xml");

    cpu = gvir_config_domain_get_cpu(domain);
    gvir_config_capabilities_cpu_set_topology(GVIR_CONFIG_CAPABILITIES_CPU(cpu),
                                              NULL);
    topology = gvir_config_capabilities_cpu_get_topology(GVIR_CONFIG_CAPABILITIES_CPU(cpu));
    g_assert(topology == NULL);
    g_object_unref(G_OBJECT(cpu));

    gvir_config_domain_set_cpu(domain, NULL);
    cpu = gvir_config_domain_get_cpu(domain);
    g_assert(cpu == NULL);

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_disk(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainDisk *disk;
    GVirConfigDomainDiskDriver *driver;
    GList *disks;

    domain = gvir_config_domain_new();

    driver = gvir_config_domain_disk_driver_new();
    gvir_config_domain_disk_driver_set_name(driver, "foo");
    gvir_config_domain_disk_driver_set_format(driver, GVIR_CONFIG_DOMAIN_DISK_FORMAT_BOCHS);
    gvir_config_domain_disk_driver_set_name(driver, "qemu");
    gvir_config_domain_disk_driver_set_cache(driver, GVIR_CONFIG_DOMAIN_DISK_CACHE_NONE);
    gvir_config_domain_disk_driver_set_format(driver, GVIR_CONFIG_DOMAIN_DISK_FORMAT_QCOW2);
    gvir_config_domain_disk_driver_set_copy_on_read(driver, TRUE);

    disk = gvir_config_domain_disk_new();
    gvir_config_domain_disk_set_type(disk, GVIR_CONFIG_DOMAIN_DISK_FILE);
    gvir_config_domain_disk_set_guest_device_type(disk, GVIR_CONFIG_DOMAIN_DISK_GUEST_DEVICE_DISK);
    gvir_config_domain_disk_set_source(disk, "/tmp/foo/bar");
    gvir_config_domain_disk_set_startup_policy (disk, GVIR_CONFIG_DOMAIN_DISK_STARTUP_POLICY_REQUISITE);
    gvir_config_domain_disk_set_target_bus(disk, GVIR_CONFIG_DOMAIN_DISK_BUS_IDE);
    gvir_config_domain_disk_set_target_dev(disk, "hda");
    gvir_config_domain_disk_set_driver(disk, driver);
    g_object_unref(G_OBJECT(driver));
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(disk));
    g_object_unref(disk);

    check_xml(domain, "gconfig-domain-device-disk.xml");

    disks = get_devices_by_type(domain, GVIR_CONFIG_TYPE_DOMAIN_DISK);
    g_assert_cmpint(g_list_length(disks), ==, 1);
    g_assert(GVIR_CONFIG_IS_DOMAIN_DISK(disks->data));
    disk = GVIR_CONFIG_DOMAIN_DISK(disks->data);
    g_assert_cmpint(gvir_config_domain_disk_get_disk_type(disk), ==, GVIR_CONFIG_DOMAIN_DISK_FILE);
    g_assert_cmpint(gvir_config_domain_disk_get_guest_device_type(disk), ==, GVIR_CONFIG_DOMAIN_DISK_GUEST_DEVICE_DISK);
    g_assert_cmpint(gvir_config_domain_disk_get_startup_policy (disk), ==, GVIR_CONFIG_DOMAIN_DISK_STARTUP_POLICY_REQUISITE);
    g_assert_cmpstr(gvir_config_domain_disk_get_source(disk), ==, "/tmp/foo/bar");
    driver = gvir_config_domain_disk_get_driver(disk);
    g_assert(driver != NULL);
    g_assert_cmpint(gvir_config_domain_disk_driver_get_cache(driver), ==, GVIR_CONFIG_DOMAIN_DISK_CACHE_NONE);
    g_assert_cmpstr(gvir_config_domain_disk_driver_get_name(driver), ==, "qemu");
    g_assert_cmpint(gvir_config_domain_disk_driver_get_format(driver), ==, GVIR_CONFIG_DOMAIN_DISK_FORMAT_QCOW2);
    g_assert(gvir_config_domain_disk_driver_get_copy_on_read(driver));
    g_assert_cmpint(gvir_config_domain_disk_get_target_bus(disk), ==, GVIR_CONFIG_DOMAIN_DISK_BUS_IDE);
    g_assert_cmpstr(gvir_config_domain_disk_get_target_dev(disk), ==, "hda");
    g_object_unref(G_OBJECT(driver));

    gvir_config_domain_disk_set_driver(disk, NULL);
    driver = gvir_config_domain_disk_get_driver(disk);
    g_assert(driver == NULL);

    g_list_free_full(disks, g_object_unref);

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_filesystem(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainFilesys *fs;

    domain = gvir_config_domain_new();

    fs = gvir_config_domain_filesys_new();
    gvir_config_domain_filesys_set_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_FILE);
    gvir_config_domain_filesys_set_access_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_ACCESS_MAPPED);
    gvir_config_domain_filesys_set_driver_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_DRIVER_NBD);
    gvir_config_domain_filesys_set_driver_format(fs, GVIR_CONFIG_DOMAIN_DISK_FORMAT_QCOW2);
    gvir_config_domain_filesys_set_source(fs, "/path/to/source");
    gvir_config_domain_filesys_set_target(fs, "/path/to/target1");
    gvir_config_domain_filesys_set_readonly(fs, TRUE);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(fs));
    g_object_unref(fs);

    /* Add a RAM fs */
    fs = gvir_config_domain_filesys_new();
    gvir_config_domain_filesys_set_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_RAM);
    gvir_config_domain_filesys_set_access_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_ACCESS_PASSTHROUGH);
    gvir_config_domain_filesys_set_driver_type(fs, GVIR_CONFIG_DOMAIN_FILESYS_DRIVER_PATH);
    gvir_config_domain_filesys_set_ram_usage(fs, 1234);
    gvir_config_domain_filesys_set_target(fs, "/path/to/target2");
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(fs));
    g_object_unref(fs);

    check_xml(domain, "gconfig-domain-device-filesys.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_network(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainInterface *interface;

    domain = gvir_config_domain_new();

    interface = GVIR_CONFIG_DOMAIN_INTERFACE(gvir_config_domain_interface_network_new());
    gvir_config_domain_interface_network_set_source(GVIR_CONFIG_DOMAIN_INTERFACE_NETWORK(interface),
                                                    "default");
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(interface));
    g_object_unref(G_OBJECT(interface));

    interface = GVIR_CONFIG_DOMAIN_INTERFACE(gvir_config_domain_interface_user_new());
    gvir_config_domain_interface_set_ifname(interface, "eth0");
    gvir_config_domain_interface_set_link_state(interface,
                                                GVIR_CONFIG_DOMAIN_INTERFACE_LINK_STATE_UP);
    gvir_config_domain_interface_set_mac(interface, "00:11:22:33:44:55");
    gvir_config_domain_interface_set_model(interface, "foo");
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(interface));
    g_object_unref(G_OBJECT(interface));

    check_xml(domain, "gconfig-domain-device-network.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_input(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainInput *input;
    GList *inputs;

    domain = gvir_config_domain_new();

    input = gvir_config_domain_input_new();
    gvir_config_domain_input_set_device_type(input,
                                             GVIR_CONFIG_DOMAIN_INPUT_DEVICE_TABLET);
    gvir_config_domain_input_set_bus(input, GVIR_CONFIG_DOMAIN_INPUT_BUS_USB);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(input));
    g_object_unref(G_OBJECT(input));

    check_xml(domain, "gconfig-domain-device-input.xml");

    inputs = get_devices_by_type(domain, GVIR_CONFIG_TYPE_DOMAIN_INPUT);
    g_assert_cmpint(g_list_length(inputs), ==, 1);
    g_assert(GVIR_CONFIG_IS_DOMAIN_INPUT(inputs->data));
    input = GVIR_CONFIG_DOMAIN_INPUT(inputs->data);
    g_assert_cmpint(gvir_config_domain_input_get_device_type(input), ==, GVIR_CONFIG_DOMAIN_INPUT_DEVICE_TABLET);
    g_assert_cmpint(gvir_config_domain_input_get_bus(input), ==, GVIR_CONFIG_DOMAIN_INPUT_BUS_USB);
    g_list_free_full(inputs, g_object_unref);

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_graphics(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainGraphicsSpice *graphics;

    domain = gvir_config_domain_new();
    graphics = gvir_config_domain_graphics_spice_new();
    gvir_config_domain_graphics_spice_set_port(graphics, 1234);
    g_assert_cmpint(gvir_config_domain_graphics_spice_get_port(graphics), ==, 1234);

    /* SPICE image compression configuration */
    gvir_config_domain_graphics_spice_set_image_compression
        (graphics, GVIR_CONFIG_DOMAIN_GRAPHICS_SPICE_IMAGE_COMPRESSION_AUTO_LZ);
    g_assert_cmpint(gvir_config_domain_graphics_spice_get_image_compression(graphics), ==,
             GVIR_CONFIG_DOMAIN_GRAPHICS_SPICE_IMAGE_COMPRESSION_AUTO_LZ);

    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(graphics));
    g_object_unref(G_OBJECT(graphics));

    check_xml(domain, "gconfig-domain-device-graphics.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_video(void)
{
    GVirConfigDomain *domain;

    domain = gvir_config_domain_new();

    /* video node */
    GVirConfigDomainVideo *video;

    video = gvir_config_domain_video_new();
    gvir_config_domain_video_set_model(video,
                                       GVIR_CONFIG_DOMAIN_VIDEO_MODEL_QXL);
    gvir_config_domain_video_set_heads(video, 4);
    gvir_config_domain_video_set_vram(video, 256*1024);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(video));
    g_object_unref(G_OBJECT(video));

    check_xml(domain, "gconfig-domain-device-video.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_sound(void)
{
    GVirConfigDomain *domain;

    domain = gvir_config_domain_new();
    /* sound node */
    GVirConfigDomainSound *sound;

    sound = gvir_config_domain_sound_new();
    gvir_config_domain_sound_set_model(sound,
                                       GVIR_CONFIG_DOMAIN_SOUND_MODEL_ES1370);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(sound));
    g_object_unref(G_OBJECT(sound));

    check_xml(domain, "gconfig-domain-device-sound.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_console(void)
{
    GVirConfigDomain *domain;

    domain = gvir_config_domain_new();
    /* console node */
    GVirConfigDomainConsole *console;
    GVirConfigDomainChardevSourcePty *pty;

    console = gvir_config_domain_console_new();
    pty = gvir_config_domain_chardev_source_pty_new();
    gvir_config_domain_chardev_set_source(GVIR_CONFIG_DOMAIN_CHARDEV(console),
                                          GVIR_CONFIG_DOMAIN_CHARDEV_SOURCE(pty));
    g_object_unref(G_OBJECT(pty));
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(console));
    g_object_unref(G_OBJECT(console));

    check_xml(domain, "gconfig-domain-device-console.xml");

    g_object_unref(G_OBJECT(domain));
}


static void test_domain_device_channel(void)
{
    GVirConfigDomain *domain;

    domain = gvir_config_domain_new();
    /* spice agent channel */
    GVirConfigDomainChannel *channel;
    GVirConfigDomainChardevSourceSpiceVmc *spicevmc;
    GVirConfigDomainChardevSourceSpicePort *spiceport;
    GVirConfigDomainChardevSourceUnix *unix_source;

    channel = gvir_config_domain_channel_new();
    gvir_config_domain_channel_set_target_type(channel,
                                               GVIR_CONFIG_DOMAIN_CHANNEL_TARGET_VIRTIO);
    spicevmc = gvir_config_domain_chardev_source_spicevmc_new();
    gvir_config_domain_chardev_set_source(GVIR_CONFIG_DOMAIN_CHARDEV(channel),
                                          GVIR_CONFIG_DOMAIN_CHARDEV_SOURCE(spicevmc));
    g_object_unref(G_OBJECT(spicevmc));
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(channel));
    g_object_unref(G_OBJECT(channel));

    channel = gvir_config_domain_channel_new();
    gvir_config_domain_channel_set_target_type(channel,
                                               GVIR_CONFIG_DOMAIN_CHANNEL_TARGET_VIRTIO);
    gvir_config_domain_channel_set_target_name(channel,
                                               "org.spice-space.webdav.0");
    spiceport = gvir_config_domain_chardev_source_spiceport_new();
    gvir_config_domain_chardev_source_spiceport_set_channel(spiceport, "org.spice-space.webdav.0");
    gvir_config_domain_chardev_set_source(GVIR_CONFIG_DOMAIN_CHARDEV(channel),
                                          GVIR_CONFIG_DOMAIN_CHARDEV_SOURCE(spiceport));
    g_object_unref(G_OBJECT(spiceport));
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(channel));
    g_object_unref(G_OBJECT(channel));

    channel = gvir_config_domain_channel_new();
    gvir_config_domain_channel_set_target_type(channel,
                                               GVIR_CONFIG_DOMAIN_CHANNEL_TARGET_VIRTIO);
    gvir_config_domain_channel_set_target_name(channel, "org.qemu.guest_agent.0");
    unix_source = gvir_config_domain_chardev_source_unix_new();
    gvir_config_domain_chardev_set_source(GVIR_CONFIG_DOMAIN_CHARDEV(channel),
                                          GVIR_CONFIG_DOMAIN_CHARDEV_SOURCE(unix_source));
    g_object_unref(G_OBJECT(unix_source));
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(channel));
    g_object_unref(G_OBJECT(channel));
    check_xml(domain, "gconfig-domain-device-channel.xml");

    g_object_unref(G_OBJECT(domain));
}


static GVirConfigDomainControllerUsb *
domain_device_usb_redir_create_usb_controller(GVirConfigDomainControllerUsbModel model,
                                              guint indx,
                                              GVirConfigDomainControllerUsb *master,
                                              guint start_port,
                                              guint domain, guint bus,
                                              guint slot, guint function,
                                              gboolean multifunction)
{
    GVirConfigDomainControllerUsb *controller;
    GVirConfigDomainAddressPci *address;

    controller = gvir_config_domain_controller_usb_new();
    gvir_config_domain_controller_usb_set_model(controller, model);
    gvir_config_domain_controller_set_index(GVIR_CONFIG_DOMAIN_CONTROLLER(controller), indx);
    if (master)
        gvir_config_domain_controller_usb_set_master(controller, master, start_port);
    address = gvir_config_domain_address_pci_new();
    gvir_config_domain_address_pci_set_domain(address, domain);
    gvir_config_domain_address_pci_set_bus(address, bus);
    gvir_config_domain_address_pci_set_slot(address, slot);
    gvir_config_domain_address_pci_set_function(address, function);
    if (multifunction)
        gvir_config_domain_address_pci_set_multifunction(address, multifunction);
    gvir_config_domain_controller_set_address(GVIR_CONFIG_DOMAIN_CONTROLLER(controller),
                                              GVIR_CONFIG_DOMAIN_ADDRESS(address));
    g_object_unref(G_OBJECT(address));

    return controller;
}


static GVirConfigDomainRedirdev *
domain_device_usb_redir_create_redirdev(guint bus, guint port)
{
    GVirConfigDomainRedirdev *redirdev;
    GVirConfigDomainAddressUsb *address;
    GVirConfigDomainChardevSourceSpiceVmc *spicevmc;
    gchar *port_str;

    redirdev = gvir_config_domain_redirdev_new();
    gvir_config_domain_redirdev_set_bus(redirdev,
                                        GVIR_CONFIG_DOMAIN_REDIRDEV_BUS_USB);
    spicevmc = gvir_config_domain_chardev_source_spicevmc_new();
    gvir_config_domain_chardev_set_source(GVIR_CONFIG_DOMAIN_CHARDEV(redirdev),
                                          GVIR_CONFIG_DOMAIN_CHARDEV_SOURCE(spicevmc));
    g_object_unref(G_OBJECT(spicevmc));

    address = gvir_config_domain_address_usb_new();
    gvir_config_domain_address_usb_set_bus(address, bus);
    port_str = g_strdup_printf("%d", port);
    gvir_config_domain_address_usb_set_port(address, port_str);
    g_free(port_str);
    gvir_config_domain_redirdev_set_address(redirdev,
                                            GVIR_CONFIG_DOMAIN_ADDRESS(address));
    g_object_unref(G_OBJECT(address));

    return redirdev;
}


static void test_domain_device_usb_redir(void)
{
    GVirConfigDomain *domain;

    domain = gvir_config_domain_new();
    /* spice usb redirection */
    GVirConfigDomainControllerUsb *ehci;
    GVirConfigDomainControllerUsb *uhci1;
    GVirConfigDomainControllerUsb *uhci2;
    GVirConfigDomainControllerUsb *uhci3;
    GVirConfigDomainRedirdev *redirdev;

    ehci = domain_device_usb_redir_create_usb_controller(GVIR_CONFIG_DOMAIN_CONTROLLER_USB_MODEL_ICH9_EHCI1,
                                                         1, NULL, 0, 0, 0, 8, 7, FALSE);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(ehci));
    uhci1 = domain_device_usb_redir_create_usb_controller(GVIR_CONFIG_DOMAIN_CONTROLLER_USB_MODEL_ICH9_UHCI1,
                                                          7, ehci, 0, 0, 0, 8, 0, TRUE);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(uhci1));
    uhci2 = domain_device_usb_redir_create_usb_controller(GVIR_CONFIG_DOMAIN_CONTROLLER_USB_MODEL_ICH9_UHCI2,
                                                          7, ehci, 2, 0, 0, 8, 1, FALSE);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(uhci2));
    uhci3 = domain_device_usb_redir_create_usb_controller(GVIR_CONFIG_DOMAIN_CONTROLLER_USB_MODEL_ICH9_UHCI3,
                                                          7, ehci, 4, 0, 0, 8, 2, FALSE);
    g_assert_cmpint(gvir_config_domain_controller_get_index(GVIR_CONFIG_DOMAIN_CONTROLLER(uhci1)), ==, 1);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(uhci3));
    g_object_unref(G_OBJECT(uhci1));
    g_object_unref(G_OBJECT(uhci2));
    g_object_unref(G_OBJECT(uhci3));
    g_object_unref(G_OBJECT(ehci));


    /* three redirdev channels allows to redirect a maximum of 3 USB
     * devices at a time. The address which create_redirdev assigns to the
     * redirdev object is optional
     */
    redirdev = domain_device_usb_redir_create_redirdev(0, 3);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(redirdev));
    g_object_unref(G_OBJECT(redirdev));
    redirdev = domain_device_usb_redir_create_redirdev(0, 4);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(redirdev));
    g_object_unref(G_OBJECT(redirdev));
    redirdev = domain_device_usb_redir_create_redirdev(0, 5);
    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE(redirdev));
    g_object_unref(G_OBJECT(redirdev));

    check_xml(domain, "gconfig-domain-device-usbredir.xml");

    g_object_unref(G_OBJECT(domain));
}

static void test_domain_device_pci_hostdev(void)
{
    GVirConfigDomain *domain;
    GVirConfigDomainAddressPci *address;
    GVirConfigDomainHostdevPci *hostdev;

    domain = gvir_config_domain_new();

    hostdev = gvir_config_domain_hostdev_pci_new();
    gvir_config_domain_hostdev_set_boot_order(GVIR_CONFIG_DOMAIN_HOSTDEV(hostdev), 1);
    g_assert_cmpint(gvir_config_domain_hostdev_get_boot_order(GVIR_CONFIG_DOMAIN_HOSTDEV(hostdev)), ==, 1);
    gvir_config_domain_hostdev_pci_set_managed(hostdev, TRUE);
    g_assert(gvir_config_domain_hostdev_pci_get_managed(hostdev) == TRUE);
    gvir_config_domain_hostdev_pci_set_rom_bar(hostdev, TRUE);
    gvir_config_domain_hostdev_pci_set_rom_file(hostdev, "/etc/fake/boot.bin");
    g_assert_cmpstr(gvir_config_domain_hostdev_pci_get_rom_file(hostdev), ==, "/etc/fake/boot.bin");
    g_assert(gvir_config_domain_hostdev_pci_get_rom_bar(hostdev));

    address = gvir_config_domain_address_pci_new();
    gvir_config_domain_address_pci_set_domain(address, 1);
    gvir_config_domain_address_pci_set_bus(address, 2);
    gvir_config_domain_address_pci_set_slot(address, 3);
    gvir_config_domain_address_pci_set_function(address, 4);
    gvir_config_domain_hostdev_pci_set_address(hostdev, address);
    g_object_unref(G_OBJECT(address));

    address = gvir_config_domain_hostdev_pci_get_address(hostdev);
    g_assert(address != NULL);
    g_assert_cmpint(gvir_config_domain_address_pci_get_domain(address), ==, 1);
    g_assert_cmpint(gvir_config_domain_address_pci_get_bus(address), ==, 2);
    g_assert_cmpint(gvir_config_domain_address_pci_get_slot(address), ==, 3);
    g_assert_cmpint(gvir_config_domain_address_pci_get_function(address), ==, 4);
    g_object_unref(G_OBJECT(address));

    gvir_config_domain_add_device(domain, GVIR_CONFIG_DOMAIN_DEVICE (hostdev));
    g_object_unref(G_OBJECT(hostdev));

    check_xml(domain, "gconfig-domain-device-pci-hostdev.xml");

    g_object_unref(G_OBJECT(domain));
}

static void test_domain_device_unknown(void)
{
    GVirConfigDomain *domain;
    GList *devices;
    GError *error = NULL;
    char *xml;

    xml = load_xml("gconfig-domain-device-unknown.xml");

    domain = gvir_config_domain_new_from_xml(xml, &error);
    g_assert_no_error(error);

    devices = gvir_config_domain_get_devices(domain);
    g_assert_nonnull(devices);
    gvir_config_domain_set_devices(domain, devices);

    check_xml(domain, "gconfig-domain-device-unknown.xml");

    g_list_free_full(devices, g_object_unref);
    g_object_unref(G_OBJECT(domain));
    g_free(xml);
}

static void test_domain_capabilities_os(void)
{
    GVirConfigDomainCapabilities *domain_caps;
    GVirConfigDomainCapabilitiesOs *os;
    GList *firmwares, *l;
    gsize i;
    GVirConfigDomainOsFirmware expected_firmwares[] = {GVIR_CONFIG_DOMAIN_OS_FIRMWARE_BIOS,
                                                       GVIR_CONFIG_DOMAIN_OS_FIRMWARE_EFI};
    GError *error = NULL;
    gchar *xml;

    xml = load_xml("gconfig-domain-capabilities-os.xml");

    domain_caps =  gvir_config_domain_capabilities_new_from_xml(xml, &error);
    g_assert_no_error(error);

    os = gvir_config_domain_capabilities_get_os(domain_caps);
    g_assert_nonnull(os);

    firmwares = gvir_config_domain_capabilities_os_get_firmwares(os);
    g_assert_nonnull(firmwares);

    for (l = firmwares, i = 0; l != NULL; l = l->next, i++)
        g_assert_cmpint(GPOINTER_TO_INT(l->data), ==, expected_firmwares[i]);


    g_list_free(firmwares);
    g_object_unref(os);
    g_object_unref(domain_caps);
    g_free(xml);
}

int main(int argc, char **argv)
{
    gvir_config_init(&argc, &argv);
    g_test_init(&argc, &argv, NULL);

    g_test_add_func("/libvirt-gconfig/domain", test_domain);
    g_test_add_func("/libvirt-gconfig/domain-clock", test_domain_clock);
    g_test_add_func("/libvirt-gconfig/domain-os", test_domain_os);
    g_test_add_func("/libvirt-gconfig/domain-cpu", test_domain_cpu);
    g_test_add_func("/libvirt-gconfig/domain-device-disk",
                    test_domain_device_disk);
    g_test_add_func("/libvirt-gconfig/domain-device-filesystem",
                    test_domain_device_filesystem);
    g_test_add_func("/libvirt-gconfig/domain-device-network",
                    test_domain_device_network);
    g_test_add_func("/libvirt-gconfig/domain-device-input",
                    test_domain_device_input);
    g_test_add_func("/libvirt-gconfig/domain-device-graphics",
                    test_domain_device_graphics);
    g_test_add_func("/libvirt-gconfig/domain-device-video",
                    test_domain_device_video);
    g_test_add_func("/libvirt-gconfig/domain-device-sound",
                    test_domain_device_sound);
    g_test_add_func("/libvirt-gconfig/domain-device-console",
                    test_domain_device_console);
    g_test_add_func("/libvirt-gconfig/domain-device-channel",
                    test_domain_device_channel);
    g_test_add_func("/libvirt-gconfig/domain-device-usb-redir",
                    test_domain_device_usb_redir);
    g_test_add_func("/libvirt-gconfig/domain-device-pci-hostdev",
                    test_domain_device_pci_hostdev);
    g_test_add_func("/libvirt-gconfig/domain-device-unknown",
                    test_domain_device_unknown);
    g_test_add_func("/libvirt-gconfig/domain-capabilities-os",
                    test_domain_capabilities_os);

    return g_test_run();
}