openkylin
/
libgovirt
mirror of https://gitee.com/openkylin/libgovirt.git
9
0
Fork 0
Code Issues Projects Releases Wiki Activity
libgovirt/tests/test-govirt.c

256 lines
9.1 KiB
C
Raw Permalink Blame History

/* Copyright 2016 Red Hat, Inc. and/or its affiliates.
*
* Permission is hereby granted, free of charge, to any person
* obtaining a copy of this software and associated documentation files
* (the "Software"), to deal in the Software without restriction,
* including without limitation the rights to use, copy, modify, merge,
* publish, distribute, sublicense, and/or sell copies of the Software,
* and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include <config.h>
#include <govirt/govirt.h>
#include <stdlib.h>
#include "mock-httpd.h"
#define GOVIRT_HTTPS_PORT 8088
static void ovirt_proxy_set_mock_ca(OvirtProxy *proxy)
{
gchar *data;
gsize size;
GByteArray *cacert;
g_file_get_contents(abs_srcdir "/https-cert/ca-cert.pem", &data, &size, NULL);
cacert = g_byte_array_new_take((guint8 *)data, size);
g_object_set(proxy, "ca-cert", cacert, NULL);
g_byte_array_unref(cacert);
}
static void test_govirt_https_ca(void)
{
OvirtProxy *proxy;
OvirtApi *api;
GError *error = NULL;
GovirtMockHttpd *httpd;
httpd = govirt_mock_httpd_new(GOVIRT_HTTPS_PORT);
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api", "<api></api>");
govirt_mock_httpd_start(httpd);
g_setenv("GOVIRT_NO_SSL_STRICT", "1", TRUE);
g_test_expect_message("libgovirt", G_LOG_LEVEL_WARNING,
"Disabling strict checking of SSL certificates");
g_test_expect_message("libgovirt", G_LOG_LEVEL_CRITICAL,
"ovirt_proxy_set_api_from_xml: assertion 'vms != NULL' failed");
proxy = ovirt_proxy_new("localhost:" G_STRINGIFY(GOVIRT_HTTPS_PORT));
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_nonnull(api);
g_assert_no_error(error);
g_object_unref(proxy);
g_unsetenv("GOVIRT_NO_SSL_STRICT");
g_test_expect_message("libgovirt", G_LOG_LEVEL_WARNING,
"Error while getting collection: Unacceptable TLS certificate");
proxy = ovirt_proxy_new("localhost:" G_STRINGIFY(GOVIRT_HTTPS_PORT));
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_null(api);
g_assert_error(error, REST_PROXY_ERROR, REST_PROXY_ERROR_SSL);
g_assert_cmpstr(error->message, ==, "Unacceptable TLS certificate");
g_clear_error(&error);
g_test_expect_message("libgovirt", G_LOG_LEVEL_CRITICAL,
"ovirt_proxy_set_api_from_xml: assertion 'vms != NULL' failed");
ovirt_proxy_set_mock_ca(proxy);
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_nonnull(api);
g_assert_no_error(error);
g_clear_object(&proxy);
govirt_mock_httpd_stop(httpd);
}
static void test_govirt_http(void)
{
OvirtProxy *proxy;
OvirtApi *api;
GError *error = NULL;
GovirtMockHttpd *httpd;
httpd = govirt_mock_httpd_new(GOVIRT_HTTPS_PORT);
govirt_mock_httpd_disable_tls(httpd, TRUE);
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api", "<api></api>");
govirt_mock_httpd_start(httpd);
g_setenv("GOVIRT_DISABLE_HTTPS", "1", TRUE);
g_test_expect_message("libgovirt", G_LOG_LEVEL_WARNING,
"Using plain text HTTP connection");
g_test_expect_message("libgovirt", G_LOG_LEVEL_CRITICAL,
"ovirt_proxy_set_api_from_xml: assertion 'vms != NULL' failed");
proxy = ovirt_proxy_new("localhost:" G_STRINGIFY(GOVIRT_HTTPS_PORT));
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_nonnull(api);
g_assert_no_error(error);
g_object_unref(proxy);
g_unsetenv("GOVIRT_DISABLE_HTTPS");
govirt_mock_httpd_stop(httpd);
}
static void test_govirt_list_vms(void)
{
OvirtProxy *proxy;
OvirtApi *api;
OvirtCollection *vms;
OvirtResource *vm;
GError *error = NULL;
GovirtMockHttpd *httpd;
const char *vms_body = "<vms> \
<vm href=\"api/vms/uuid0\" id=\"uuid0\"> \
<name>vm0</name> \
</vm> \
<vm href=\"api/vms/uuid1\" id=\"uuid1\"> \
<name>vm1</name> \
<type>desktop</type> \
<status><state>up</state></status> \
<display> \
<type>spice</type> \
<address>10.0.0.123</address> \
<secure_port>5900</secure_port> \
<monitors>1</monitors> \
<single_qxl_pci>true</single_qxl_pci> \
<allow_override>false</allow_override> \
<smartcard_enabled>false</smartcard_enabled> \
<proxy>10.0.0.10</proxy> \
<file_transfer_enabled>true</file_transfer_enabled> \
<copy_paste_enabled>true</copy_paste_enabled> \
</display> \
</vm> \
<vm href=\"api/vms/uuid2\" id=\"uuid2\"> \
<name>vm2</name> \
</vm> \
</vms>";
httpd = govirt_mock_httpd_new(GOVIRT_HTTPS_PORT);
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api",
"<api><link href=\"/ovirt-engine/api/vms\" rel=\"vms\"/></api>");
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api/vms", vms_body);
govirt_mock_httpd_start(httpd);
proxy = ovirt_proxy_new("localhost:" G_STRINGIFY(GOVIRT_HTTPS_PORT));
ovirt_proxy_set_mock_ca(proxy);
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_nonnull(api);
g_assert_no_error(error);
vms = ovirt_api_get_vms(api);
ovirt_collection_fetch(vms, proxy, &error);
g_assert_no_error(error);
vm = ovirt_collection_lookup_resource(vms, "vm1");
g_assert_nonnull(vm);
g_object_unref(vm);
g_object_unref(proxy);
govirt_mock_httpd_stop(httpd);
}
static void test_govirt_list_duplicate_vms(void)
{
OvirtProxy *proxy;
OvirtApi *api;
OvirtCollection *vms;
OvirtResource *vm;
GError *error = NULL;
GovirtMockHttpd *httpd;
const char *vms_body = "<vms> \
<vm href=\"api/vms/uuid0\" id=\"uuid0\"> \
<name>vm0</name> \
</vm> \
<vm href=\"api/vms/uuid1\" id=\"uuid1\"> \
<name>vm0</name> \
</vm> \
</vms>";
httpd = govirt_mock_httpd_new(GOVIRT_HTTPS_PORT);
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api",
"<api><link href=\"/ovirt-engine/api/vms\" rel=\"vms\"/></api>");
govirt_mock_httpd_add_request(httpd, "GET", "/ovirt-engine/api/vms", vms_body);
govirt_mock_httpd_start(httpd);
proxy = ovirt_proxy_new("localhost:" G_STRINGIFY(GOVIRT_HTTPS_PORT));
ovirt_proxy_set_mock_ca(proxy);
api = ovirt_proxy_fetch_api(proxy, &error);
g_test_assert_expected_messages();
g_assert_nonnull(api);
g_assert_no_error(error);
g_test_expect_message("libgovirt", G_LOG_LEVEL_MESSAGE,
"'vm' resource with the same name ('vm0') already exists");
vms = ovirt_api_get_vms(api);
ovirt_collection_fetch(vms, proxy, &error);
g_test_assert_expected_messages();
g_assert_no_error(error);
vm = ovirt_collection_lookup_resource(vms, "vm0");
g_assert_nonnull(vm);
g_object_unref(vm);
g_object_unref(proxy);
govirt_mock_httpd_stop(httpd);
}
static void quit(int sig)
{
exit(1);
}
int
main(int argc, char **argv)
{
/* exit cleanly on ^C in case we're valgrinding. */
signal(SIGINT, quit);
/* prevents core generations as this could cause some issues/timeout
* depending on system configuration */
signal(SIGABRT, quit);
g_test_init(&argc, &argv, NULL);
g_test_add_func("/govirt/test-https-ca", test_govirt_https_ca);
g_test_add_func("/govirt/test-http", test_govirt_http);
g_test_add_func("/govirt/test-list-vms", test_govirt_list_vms);
g_test_add_func("/govirt/test-list-duplicate-vms", test_govirt_list_duplicate_vms);
return g_test_run();
}
Reference in New Issue View Git Blame Copy Permalink