zheiling/BBS-Server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

testing: refactoring; fixes file_download

Browse Source
This commit is contained in:
zheiling
2026-06-26 13:51:11 +03:00
parent 8849637d45
commit 7e9487377f
11 changed files with 110 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/file_p/CMakeLists.txt
+6 -6
View File
@@ -2,13 +2,13 @@
add_executable(clear_list
clear_list.c
utils.c
)
target_link_libraries(clear_list
cmocka
file_p
main
test_utils
)
target_link_options(clear_list PRIVATE -Wl,--wrap=free)
@@ -19,13 +19,13 @@ add_test(NAME file_p_clear_list_test_1 COMMAND clear_list)
add_executable(file_list
file_list.c
utils.c
)
target_link_libraries(file_list
cmocka
file_p
main
test_utils
)
target_link_options(file_list PRIVATE -Wl,--wrap=db_get_files_data,--wrap=write)
@@ -36,13 +36,13 @@ add_test(NAME file_p_file_list_test_1 COMMAND file_list)
add_executable(extract_names_from_hash
extract_names_from_hash.c
utils.c
)
target_link_libraries(extract_names_from_hash
cmocka
file_p
main
test_utils
)
add_test(NAME file_p_extract_names_from_hash_test_1 COMMAND extract_names_from_hash)
@@ -51,7 +51,6 @@ add_test(NAME file_p_extract_names_from_hash_test_1 COMMAND extract_names_from_h
add_executable(file_send_prepare
file_send_prepare.c
utils.c
)
target_link_libraries(file_send_prepare
@@ -59,6 +58,7 @@ target_link_libraries(file_send_prepare
file_p
main
utils
test_utils
)
target_link_options(file_send_prepare PRIVATE -Wl,--wrap=write,--wrap=print_log,--wrap=lseek,--wrap=open,--wrap=db_get_file)
@@ -69,7 +69,6 @@ add_test(NAME file_p_file_send_prepare_test_1 COMMAND file_send_prepare)
add_executable(clear_file_from_sess
clear_file_from_sess.c
utils.c
)
target_link_libraries(clear_file_from_sess
@@ -77,6 +76,7 @@ target_link_libraries(clear_file_from_sess
file_p
main
utils
test_utils
)
target_link_options(clear_file_from_sess PRIVATE -Wl,--wrap=free)
@@ -87,7 +87,6 @@ add_test(NAME file_p_clear_file_from_sess_test_1 COMMAND clear_file_from_sess)
add_executable(file_download
file_download.c
utils.c
)
target_link_libraries(file_download
@@ -96,6 +95,7 @@ target_link_libraries(file_download
main
utils
client
test_utils
)
target_link_options(file_download PRIVATE -Wl,--wrap=print_log,--wrap=read,--wrap=write)
test/file_p/clear_list.c
+1 -1
View File
@@ -10,7 +10,7 @@
#include <file_p.h>
#include <main.h>
#include <string.h>
#include "utils.h"
#include <test_utils.h>
fl_t fl_sample = {
.name = "test_file", .description = "Hello test file", .owner = "user1234"};
test/file_p/file_download.c
+39 -29
View File
@@ -9,6 +9,7 @@
#include <stdlib.h>
#include <string.h>
#include <utils.h>
#include <test_utils.h>
dbuf_t *dbuf = NULL;
#define FNAME "Test file"
@@ -16,29 +17,8 @@ dbuf_t *dbuf = NULL;
#define FSIZE PACKAGE_SIZE *ITERATIONS
int read_ret_zero = 0;
int write_ret_m_1 = 0;
char *print_str = NULL;
void *print_str_arr = NULL;
int __wrap_print_log(FILE *output, enum pl_type type, const char *fmt, ...) {
if (print_str != NULL) {
assert_string_equal(print_str, fmt);
} else if (print_str_arr != NULL) {
char **ptrs = print_str_arr;
va_list args;
assert_string_equal(*ptrs, fmt);
ptrs++;
va_start(args, fmt);
while (*ptrs != NULL) {
char *out_str = va_arg(args, char *);
assert_string_equal(out_str, *ptrs);
ptrs++;
}
va_end(args);
}
return 0;
};
extern char *print_str;
extern void *print_str_arr;
ssize_t __wrap_read(int __fd, void *__buf, size_t __nbytes) {
if (read_ret_zero)
@@ -66,20 +46,52 @@ void test__file_download__normal(void **state) {
sess.file->path = malloc(sizeof(char) * INBUFSIZE);
sess.file->description = malloc(sizeof(char) * INBUFSIZE);
strcpy(sess.file->name, FNAME);
char cont_mes[64];
sprintf(cont_mes, "continue %d\n", PACKAGE_SIZE);
const char *fmt_args[] = {"File %s is downloaded from the server\n", FNAME,
NULL};
print_str_arr = fmt_args;
while (sess.state != OP_WAIT) {
for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
file_download(&sess);
if (sess.state == OP_DOWNLOAD_WAIT_CONFIRM_PACKAGE) {
download_confirm("continue", &sess, NULL);
download_confirm(cont_mes, &sess, NULL);
}
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
clear_file_from_sess(&sess);
}
void test__file_download__normal_alt_buf_size(void **state) {
dbuf = dbuf_init(INBUFSIZE);
session sess = {.state = OP_DOWNLOAD};
sess.file = malloc(sizeof(s_file_t));
sess.file->name = malloc(sizeof FNAME);
sess.file->size = sess.file->rest = FSIZE;
sess.file->package_rest = PACKAGE_SIZE;
sess.file->path = malloc(sizeof(char) * INBUFSIZE);
sess.file->description = malloc(sizeof(char) * INBUFSIZE);
strcpy(sess.file->name, FNAME);
char cont_mes[64];
sprintf(cont_mes, "continue %d\n", PACKAGE_SIZE / 3);
const char *fmt_args[] = {"File %s is downloaded from the server\n", FNAME,
NULL};
print_str_arr = fmt_args;
for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
file_download(&sess);
if (sess.state == OP_DOWNLOAD_WAIT_CONFIRM_PACKAGE) {
download_confirm(cont_mes, &sess, NULL);
}
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
print_str_arr = NULL;
clear_file_from_sess(&sess);
}
@@ -108,7 +120,6 @@ void test__file_download__cancel(void **state) {
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
print_str_arr = NULL;
}
void test__file_download__read_zero(void **state) {
@@ -129,7 +140,6 @@ void test__file_download__read_zero(void **state) {
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
read_ret_zero = false;
print_str_arr = NULL;
}
void test__file_download__read_write_m_1(void **state) {
@@ -154,7 +164,6 @@ void test__file_download__read_write_m_1(void **state) {
assert_int_equal(sess.state, ERR);
assert_ptr_equal(sess.file, NULL);
write_ret_m_1 = false;
print_str_arr = NULL;
}
int setup(void **state) { return 0; }
@@ -163,6 +172,7 @@ int tear_down(void **state) { return 0; }
int main(int argc, char **argv) {
const struct CMUnitTest tests[] = {
cmocka_unit_test(test__file_download__normal),
cmocka_unit_test(test__file_download__normal_alt_buf_size),
cmocka_unit_test(test__file_download__cancel),
cmocka_unit_test(test__file_download__read_zero),
cmocka_unit_test(test__file_download__read_write_m_1),
test/file_p/file_list.c
+1
View File
@@ -13,6 +13,7 @@
#include <string.h>
#include <sys/cdefs.h>
#include <unistd.h>
#include <test_utils.h>
#define COUNT 20
dbuf_t *dbuf;
test/file_p/file_receive_prepare.c
+1 -5
View File
@@ -17,6 +17,7 @@
#include <sys/cdefs.h>
#include <unistd.h>
#include <utils.h>
#include <test_utils.h>
#define FSIZE 10233
#define FSIZE_S "10233"
@@ -60,11 +61,6 @@ int __wrap_mkdir(const char *__path, __mode_t __mode) { return mock_type(int); }
int __wrap_chdir(const char *__path) { return 0; }
int __wrap_print_log(FILE *output, enum pl_type type,
const char *format_message, ...) {
return 0;
};
ssize_t __wrap_write(int __fd, const void *__buf, size_t __n) __wur {
dbuf_write(__buf, __n, &dbuf);
return __n;
test/file_p/file_send_prepare.c
+1 -5
View File
@@ -14,6 +14,7 @@
#include <sys/cdefs.h>
#include <unistd.h>
#include <utils.h>
#include <test_utils.h>
dbuf_t *dbuf;
@@ -23,11 +24,6 @@ dbuf_t *dbuf;
#define FID 19
#define FID_S "19"
int __wrap_print_log(FILE *output, enum pl_type type,
const char *format_message, ...) {
return 0;
};
ssize_t __wrap_write(int __fd, const void *__buf, size_t __n) __wur {
dbuf_write(__buf, __n, &dbuf);
return __n;
test/file_p/utils.c
-81
View File
@@ -1,81 +0,0 @@
#include "../../src/main.h"
#include <stddef.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "utils.h"
dbuf_t *dbuf_init(size_t init_sz) {
dbuf_t *dbuf = malloc(sizeof(dbuf_t));
dbuf->ptr = malloc(init_sz);
dbuf->len = 0;
dbuf->capacity = init_sz;
return dbuf;
}
static int it_times = 0;
int32_t dbuf_write(const char *text, size_t text_len, dbuf_t **__dbuf) {
dbuf_t *dbuf = *__dbuf;
if (text_len > (dbuf->capacity - dbuf->len)) {
if (text_len > (dbuf->capacity * 2 - dbuf->len)) {
dbuf->capacity = text_len * 2;
} else {
dbuf->capacity = dbuf->capacity * 2;
}
(*__dbuf)->ptr = realloc(dbuf->ptr, dbuf->capacity);
dbuf = *__dbuf;
}
it_times++;
int _it_times = it_times;
strncpy(dbuf->ptr + dbuf->len, text, text_len);
dbuf->ptr[dbuf->len + text_len] = '\0';
dbuf->len += text_len;
return 0;
}
int32_t dbuf_destroy(dbuf_t **dbuf) {
free((*dbuf)->ptr);
free(*dbuf);
*dbuf = NULL;
return 0;
}
void fill_list_with_samples(fl_t *fl_sample, fl_t **fl_st, fl_t **fl_cur,
int amount) {
fl_t *_fl_st = NULL, *_fl_cur = NULL;
if (fl_st == NULL)
return;
for (int i = 0; i < amount; i++) {
if (i == 0) {
_fl_st = _fl_cur = malloc(sizeof(fl_t));
memcpy(_fl_st, fl_sample, sizeof(fl_t));
} else {
_fl_cur->next = malloc(sizeof(fl_t));
memcpy(_fl_cur->next, fl_sample, sizeof(fl_t));
_fl_cur = _fl_cur->next;
}
_fl_cur->next = NULL;
_fl_cur->name = malloc(sizeof(char) * (strlen(fl_sample->name) + 32));
_fl_cur->description =
malloc(sizeof(char) * strlen(fl_sample->description) + 1);
_fl_cur->owner = malloc(sizeof(char) * strlen(fl_sample->owner) + 1);
sprintf(_fl_cur->name, "%s_%d", fl_sample->name, i + 1);
memcpy(_fl_cur->description, fl_sample->description,
sizeof(char) * strlen(fl_sample->description) + 1);
memcpy(_fl_cur->owner, fl_sample->owner,
sizeof(char) * strlen(fl_sample->owner) + 1);
}
*fl_st = _fl_st;
if (fl_cur != NULL) {
*fl_cur = _fl_cur;
}
}
test/file_p/utils.h
-13
View File
@@ -1,13 +0,0 @@
#include "../../src/main.h"
typedef struct {
char *ptr;
size_t len;
size_t capacity;
} dbuf_t;
void fill_list_with_samples(fl_t *fl_samples, fl_t **fl_st, fl_t **fl_cur,
int amount);
dbuf_t *dbuf_init(size_t init_sz);
int32_t dbuf_write(const char *text, size_t len, dbuf_t **dbuf);
int32_t dbuf_destroy(dbuf_t **dbuf);