mirror of
https://github.com/pumpitupdev/pumptools.git
synced 2026-09-22 22:28:03 +03:00
icex2
eafb2909d1
Use a similar code style to bemanitools though reduce indentation to 2 spaces. Also some other tweaks to be more similar to google style.
143 lines
3.3 KiB
C
143 lines
3.3 KiB
C
#include <stdlib.h>
|
|
|
|
#include <cmocka/cmocka.h>
|
|
|
|
#include "capnhook/hook/lib.h"
|
|
|
|
#include "hook/propatch/usb-fix.h"
|
|
|
|
#include "test-util/mem.h"
|
|
|
|
#include "util/log.h"
|
|
#include "util/mem.h"
|
|
|
|
static const char *readlink_return_value =
|
|
"../devices/pci0000:00/0000:00:14.0/usb2/2-2/2-2:1.0/host7/target7:0:0/"
|
|
"7:0:0:0/block/sde";
|
|
static const char *player1_match_buf =
|
|
"../devices/pci0000:00/0000:00:14.0/usb2/1-1/1-1:1.0";
|
|
static const char *player2_match_buf =
|
|
"../devices/pci0000:00/0000:00:14.0/usb2/1-2/1-2:1.0";
|
|
|
|
extern ssize_t
|
|
propatch_usb_fix_readlink(const char *path, char *buf, size_t len);
|
|
|
|
static uint32_t readlink_call_count = 0;
|
|
|
|
ssize_t readlink_mock(const char *path, char *buf, size_t len)
|
|
{
|
|
ssize_t len_return;
|
|
|
|
readlink_call_count++;
|
|
|
|
len_return = strlen(readlink_return_value);
|
|
|
|
assert_true(len >= len_return);
|
|
|
|
strcpy(buf, readlink_return_value);
|
|
|
|
return len_return;
|
|
}
|
|
|
|
static int setup(void **state)
|
|
{
|
|
struct cnh_lib_unit_test_func_mocks *func_mocks;
|
|
size_t func_mocks_cnt;
|
|
|
|
test_util_mem_install_mem_interface();
|
|
|
|
func_mocks_cnt = 1;
|
|
func_mocks = cnh_lib_allocate_func_mocks(func_mocks_cnt);
|
|
|
|
func_mocks[0].name = "readlink";
|
|
func_mocks[0].func = readlink_mock;
|
|
|
|
cnh_lib_init_unit_test(func_mocks, func_mocks_cnt);
|
|
|
|
readlink_call_count = 0;
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int teardown(void **state)
|
|
{
|
|
propatch_usb_fix_shutdown();
|
|
|
|
cnh_lib_shutdown_unit_test();
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void test_usb_fix_disabled(void **state)
|
|
{
|
|
ssize_t res;
|
|
char buffer[256];
|
|
|
|
res = propatch_usb_fix_readlink(
|
|
"/sys/block/sde/device", buffer, sizeof(buffer));
|
|
|
|
assert_int_equal(res, strlen(readlink_return_value));
|
|
assert_string_equal(buffer, readlink_return_value);
|
|
|
|
assert_int_equal(readlink_call_count, 1);
|
|
}
|
|
|
|
static void test_no_match_usb_fix(void **state)
|
|
{
|
|
ssize_t res;
|
|
char buffer[256];
|
|
|
|
propatch_usb_fix_init("sdd,sdf", "2-1", "2-2");
|
|
|
|
res = propatch_usb_fix_readlink(
|
|
"/sys/block/sde/device", buffer, sizeof(buffer));
|
|
|
|
assert_int_equal(res, strlen(readlink_return_value));
|
|
|
|
assert_int_equal(readlink_call_count, 1);
|
|
}
|
|
|
|
static void test_match_player1_usb_fix(void **state)
|
|
{
|
|
ssize_t res;
|
|
char buffer[256];
|
|
|
|
propatch_usb_fix_init("sde,sdf", "2-1", "2-2");
|
|
|
|
res = propatch_usb_fix_readlink(
|
|
"/sys/block/sde/device", buffer, sizeof(buffer));
|
|
|
|
assert_int_equal(res, strlen(player1_match_buf));
|
|
assert_string_equal(buffer, player1_match_buf);
|
|
|
|
assert_int_equal(readlink_call_count, 1);
|
|
}
|
|
|
|
static void test_match_player2_usb_fix(void **state)
|
|
{
|
|
ssize_t res;
|
|
char buffer[256];
|
|
|
|
propatch_usb_fix_init("sde,sdf", "2-2", "2-1");
|
|
|
|
res = propatch_usb_fix_readlink(
|
|
"/sys/block/sde/device", buffer, sizeof(buffer));
|
|
|
|
assert_int_equal(res, strlen(player2_match_buf));
|
|
assert_string_equal(buffer, player2_match_buf);
|
|
|
|
assert_int_equal(readlink_call_count, 1);
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test_setup_teardown(test_usb_fix_disabled, setup, teardown),
|
|
cmocka_unit_test_setup_teardown(test_no_match_usb_fix, setup, teardown),
|
|
cmocka_unit_test_setup_teardown(
|
|
test_match_player1_usb_fix, setup, teardown),
|
|
cmocka_unit_test_setup_teardown(
|
|
test_match_player2_usb_fix, setup, teardown)};
|
|
|
|
return cmocka_run_group_tests(tests, NULL, NULL);
|
|
} |