60 lines
1.6 KiB
C
60 lines
1.6 KiB
C
#include <cmocka.h>
|
|
#include <stdarg.h>
|
|
#include <stddef.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include <main.h>
|
|
#include <session.h>
|
|
#include <string.h>
|
|
|
|
#define TEST_BUF_USED "Voluptas voluptate reprehenderit eum voluptatibus.\n"
|
|
#define TEST_BUF_UNUSED \
|
|
"Impedit et rerum ut aut vel nisi. Animi et vero nihil aut possimus " \
|
|
"commodi. Autem dolorem et officia provident.\n"
|
|
#define TEST_BUF TEST_BUF_USED TEST_BUF_UNUSED
|
|
|
|
int __wrap_read(int fd, void *buf, size_t nbytes) {
|
|
return mock_type(int);
|
|
}
|
|
|
|
void test_session_do_read__regular(void **state) {
|
|
session sess = {
|
|
.buf = TEST_BUF,
|
|
.buf_used = sizeof TEST_BUF,
|
|
.sd = 2,
|
|
};
|
|
char *line;
|
|
will_return(__wrap_read, sizeof TEST_BUF - 1);
|
|
int ret = session_do_read(&sess, &line);
|
|
assert_memory_equal(line, TEST_BUF_USED, sizeof(TEST_BUF_USED));
|
|
assert_int_equal(ret, 1);
|
|
free(line);
|
|
}
|
|
|
|
void test_session_do_read__no_read(void **state) {
|
|
session sess = {
|
|
.buf = TEST_BUF,
|
|
.buf_used = sizeof TEST_BUF,
|
|
.sd = 2,
|
|
.state = OP_WAIT,
|
|
};
|
|
char *line;
|
|
will_return(__wrap_read, -1);
|
|
int ret = session_do_read(&sess, &line);
|
|
assert_int_equal(ret, 0);
|
|
assert_int_equal(sess.state, ERR);
|
|
}
|
|
|
|
int setup(void **state) { return 0; }
|
|
int tear_down(void **state) { return 0; }
|
|
|
|
int main(int argc, char **argv) {
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test(test_session_do_read__regular),
|
|
cmocka_unit_test(test_session_do_read__no_read)
|
|
};
|
|
|
|
return cmocka_run_group_tests(tests, setup, tear_down);
|
|
}
|