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

testing: supplement of file_upload

Browse Source
This commit is contained in:
zheiling
2026-06-29 16:56:28 +03:00
parent 795c8720e5
commit c4eeffd6a9
2 changed files with 89 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/file_p/file_upload.c
+67 -10
View File
@@ -13,7 +13,7 @@
#define FNAME "Test file"
#define ITERATIONS 100
#define FSIZE PACKAGE_SIZE * ITERATIONS
#define FSIZE PACKAGE_SIZE *ITERATIONS
dbuf_t *dbuf = NULL;
int read_ret_zero = 0;
@@ -30,9 +30,7 @@ void __wrap_session_send_string(struct session *sess, const char *fmt, ...) {
assert_string_equal(fmt, str2snd);
}
int __wrap_unlink(const char *__name) {
return 0;
}
int __wrap_unlink(const char *__name) { return 0; }
ssize_t __wrap_read(int __fd, void *__buf, size_t __nbytes) {
if (read_ret_zero) {
@@ -67,7 +65,7 @@ ssize_t __wrap_write(int __fd, const void *__buf, size_t __n) {
int32_t __wrap_db_save_file(session *s) { return 1; }
void test__file_upload__normal(void **state) {
void test__file_upload__success(void **state) {
dbuf = dbuf_init(INBUFSIZE);
session sess = {.state = OP_DOWNLOAD};
@@ -78,7 +76,6 @@ void test__file_upload__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];
str2snd = "finished\n";
const char *fmt_args[] = {"File %s is uploaded to the server\n", FNAME, NULL};
@@ -102,7 +99,6 @@ void test__file_upload__no_read(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];
read_ret_zero = true;
const char *fmt_args[] = {"Error uploading file %s!\n", FNAME, NULL};
@@ -126,17 +122,77 @@ void test__file_upload__canceled(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];
const char *fmt_args[] = {"Upload of %s is cancelled!\n", FNAME, NULL};
print_str_arr = fmt_args;
for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
if (i == 100) signal = sig_cancel;
if (i == 100)
signal = sig_cancel;
file_upload(&sess);
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
signal = sig_continue;
}
void test__file_upload__no_write_1(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_rest = PACKAGE_SIZE;
sess.file->path = malloc(sizeof(char) * INBUFSIZE);
sess.file->description = malloc(sizeof(char) * INBUFSIZE);
strcpy(sess.file->name, FNAME);
str2snd = "There is an error sending the file named \"%s\"\n";
const char *fmt_args[] = {"Error uploading the file \"%s\": %s\n", FNAME,
"No data available", NULL};
print_str_arr = fmt_args;
for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
if (i == 100) {
write_ret_m_1 = true;
errno = ENODATA;
}
file_upload(&sess);
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
}
void test__file_upload__no_write_2(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_rest = PACKAGE_SIZE;
sess.file->path = malloc(sizeof(char) * INBUFSIZE);
sess.file->description = malloc(sizeof(char) * INBUFSIZE);
strcpy(sess.file->name, FNAME);
str2snd = "There is an error sending the file named \"%s\"\n";
const char *fmt_args[] = {"Error uploading the file \"%s\": %s\n", FNAME,
"No data available", NULL};
print_str_arr = fmt_args;
for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
if (i == 50) {
signal = sig_cancel;
write_ret_m_1 = true;
errno = ENODATA;
}
file_upload(&sess);
}
assert_int_equal(sess.state, OP_WAIT);
assert_ptr_equal(sess.file, NULL);
}
int setup(void **state) { return 0; }
@@ -144,9 +200,10 @@ int tear_down(void **state) { return 0; }
int main(int argc, char **argv) {
const struct CMUnitTest tests[] = {
cmocka_unit_test(test__file_upload__normal),
cmocka_unit_test(test__file_upload__success),
cmocka_unit_test(test__file_upload__no_read),
cmocka_unit_test(test__file_upload__canceled),
cmocka_unit_test(test__file_upload__no_write_1),
};
return cmocka_run_group_tests(tests, setup, tear_down);