mirror of
https://github.com/libvirt/libvirt.git
synced 2025-02-25 18:55:26 -06:00
Add tests for virUSBDeviceFind functions
Mock the /sys/bus/usb directory and test the finding (and not finding) of some USB devices.
This commit is contained in:
Ján Tomko
190
tests/virusbtest.c
Normal file
190
tests/virusbtest.c
Normal file
@@ -0,0 +1,190 @@
|
||||
/*
|
||||
* Copyright (C) 2013-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: Jan Tomko <jtomko@redhat.com>
|
||||
*/
|
||||
|
||||
#include <config.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "viralloc.h"
|
||||
#include "virerror.h"
|
||||
#include "virfile.h"
|
||||
#include "virusb.h"
|
||||
|
||||
#include "testutils.h"
|
||||
|
||||
#define VIR_FROM_THIS VIR_FROM_NONE
|
||||
|
||||
typedef enum {
|
||||
FIND_BY_ALL,
|
||||
FIND_BY_VENDOR,
|
||||
FIND_BY_BUS
|
||||
} testUSBFindFlags;
|
||||
|
||||
struct findTestInfo {
|
||||
const char *name;
|
||||
unsigned int vendor;
|
||||
unsigned int product;
|
||||
unsigned int bus;
|
||||
unsigned int devno;
|
||||
const char *vroot;
|
||||
bool mandatory;
|
||||
int how;
|
||||
bool expectFailure;
|
||||
};
|
||||
|
||||
static int testDeviceFileActor(virUSBDevicePtr dev,
|
||||
const char *path,
|
||||
void *opaque ATTRIBUTE_UNUSED)
|
||||
{
|
||||
char *str = NULL;
|
||||
int ret = 0;
|
||||
|
||||
if (virAsprintf(&str, USB_DEVFS "%03d/%03d",
|
||||
virUSBDeviceGetBus(dev),
|
||||
virUSBDeviceGetDevno(dev)) < 0)
|
||||
return -1;
|
||||
|
||||
if (STRNEQ(path, str)) {
|
||||
virReportError(VIR_ERR_INTERNAL_ERROR,
|
||||
"Device path '%s' does not match expected '%s'",
|
||||
path, str);
|
||||
ret = -1;
|
||||
}
|
||||
VIR_FREE(str);
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int testDeviceFind(const void *opaque)
|
||||
{
|
||||
const struct findTestInfo *info = opaque;
|
||||
int ret = -1;
|
||||
virUSBDevicePtr dev = NULL;
|
||||
virUSBDeviceListPtr devs = NULL;
|
||||
int rv = 0;
|
||||
size_t i, ndevs = 0;
|
||||
|
||||
switch (info->how) {
|
||||
case FIND_BY_ALL:
|
||||
rv = virUSBDeviceFind(info->vendor, info->product,
|
||||
info->bus, info->devno,
|
||||
info->vroot, info->mandatory, &dev);
|
||||
break;
|
||||
case FIND_BY_VENDOR:
|
||||
rv = virUSBDeviceFindByVendor(info->vendor, info->product,
|
||||
info->vroot, info->mandatory, &devs);
|
||||
break;
|
||||
case FIND_BY_BUS:
|
||||
rv = virUSBDeviceFindByBus(info->bus, info->devno,
|
||||
info->vroot, info->mandatory, &dev);
|
||||
break;
|
||||
}
|
||||
|
||||
if (info->expectFailure) {
|
||||
if (rv == 0) {
|
||||
virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
|
||||
"unexpected success");
|
||||
} else {
|
||||
ret = 0;
|
||||
}
|
||||
goto cleanup;
|
||||
} else if (rv < 0) {
|
||||
goto cleanup;
|
||||
}
|
||||
|
||||
switch (info->how) {
|
||||
case FIND_BY_ALL:
|
||||
case FIND_BY_BUS:
|
||||
if (virUSBDeviceFileIterate(dev, testDeviceFileActor, NULL) < 0)
|
||||
goto cleanup;
|
||||
break;
|
||||
|
||||
case FIND_BY_VENDOR:
|
||||
ndevs = virUSBDeviceListCount(devs);
|
||||
|
||||
for (i = 0; i < ndevs; i++) {
|
||||
virUSBDevicePtr device = virUSBDeviceListGet(devs, i);
|
||||
if (virUSBDeviceFileIterate(device, testDeviceFileActor, NULL) < 0)
|
||||
goto cleanup;
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
ret = 0;
|
||||
|
||||
cleanup:
|
||||
virObjectUnref(devs);
|
||||
virUSBDeviceFree(dev);
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
mymain(void)
|
||||
{
|
||||
int rv = 0;
|
||||
|
||||
#define DO_TEST_FIND_FULL(name, vend, prod, bus, devno, vroot, mand, how, fail) \
|
||||
do { \
|
||||
struct findTestInfo data = { name, vend, prod, bus, \
|
||||
devno, vroot, mand, how, fail \
|
||||
}; \
|
||||
if (virtTestRun("USBDeviceFind " name, testDeviceFind, &data) < 0) \
|
||||
rv = -1; \
|
||||
} while (0)
|
||||
|
||||
#define DO_TEST_FIND(name, vend, prod, bus, devno) \
|
||||
DO_TEST_FIND_FULL(name, vend, prod, bus, devno, NULL, true, \
|
||||
FIND_BY_ALL, false)
|
||||
#define DO_TEST_FIND_FAIL(name, vend, prod, bus, devno) \
|
||||
DO_TEST_FIND_FULL(name, vend, prod, bus, devno, NULL, true, \
|
||||
FIND_BY_ALL, true)
|
||||
|
||||
#define DO_TEST_FIND_BY_BUS(name, bus, devno) \
|
||||
DO_TEST_FIND_FULL(name, 101, 202, bus, devno, NULL, true, \
|
||||
FIND_BY_BUS, false)
|
||||
#define DO_TEST_FIND_BY_BUS_FAIL(name, bus, devno) \
|
||||
DO_TEST_FIND_FULL(name, 101, 202, bus, devno, NULL, true, \
|
||||
FIND_BY_BUS, true)
|
||||
|
||||
#define DO_TEST_FIND_BY_VENDOR(name, vend, prod) \
|
||||
DO_TEST_FIND_FULL(name, vend, prod, 123, 456, NULL, true, \
|
||||
FIND_BY_VENDOR, false)
|
||||
#define DO_TEST_FIND_BY_VENDOR_FAIL(name, vend, prod) \
|
||||
DO_TEST_FIND_FULL(name, vend, prod, 123, 456, NULL, true, \
|
||||
FIND_BY_VENDOR, true)
|
||||
|
||||
DO_TEST_FIND("Nexus", 0x18d1, 0x4e22, 1, 20);
|
||||
DO_TEST_FIND_FAIL("Nexus wrong devnum", 0x18d1, 0x4e22, 1, 25);
|
||||
DO_TEST_FIND_FAIL("Bogus", 0xf00d, 0xbeef, 1024, 768);
|
||||
|
||||
DO_TEST_FIND_BY_BUS("integrated camera", 1, 5);
|
||||
DO_TEST_FIND_BY_BUS_FAIL("wrong bus/devno combination", 2, 20);
|
||||
DO_TEST_FIND_BY_BUS_FAIL("missing bus", 5, 20);
|
||||
DO_TEST_FIND_BY_BUS_FAIL("missing devnum", 1, 158);
|
||||
|
||||
DO_TEST_FIND_BY_VENDOR("Nexus (multiple results)", 0x18d1, 0x4e22);
|
||||
DO_TEST_FIND_BY_VENDOR_FAIL("Bogus vendor and product", 0xf00d, 0xbeef);
|
||||
DO_TEST_FIND_BY_VENDOR_FAIL("Valid vendor", 0x1d6b, 0xbeef);
|
||||
|
||||
if (rv < 0)
|
||||
return EXIT_FAILURE;
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
VIRT_TEST_MAIN_PRELOAD(mymain, abs_builddir "/.libs/virusbmock.so")
|
||||
Reference in New Issue
Block a user