testing: file_upload

test/file_p/file_upload.c

@@ -0,0 +1,153 @@
+#include "main.h"
+#include <client.h>
+#include <cmocka.h>
+#include <errno.h>
+#include <file_p.h>
+#include <stdarg.h>
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <test_utils.h>
+#include <utils.h>
+
+#define FNAME "Test file"
+#define ITERATIONS 100
+#define FSIZE PACKAGE_SIZE * ITERATIONS
+
+dbuf_t *dbuf = NULL;
+int read_ret_zero = 0;
+int write_ret_m_1 = 0;
+size_t package_rest = 0; /* indicator for the write function */
+enum package_signal signal;
+extern char *print_str;
+extern void *print_str_arr;
+char *str2snd;
+
+/* TODO: complete tasks in function and write new tests for them */
+
+void __wrap_session_send_string(struct session *sess, const char *fmt, ...) {
+  assert_string_equal(fmt, str2snd);
+}
+
+int __wrap_unlink(const char *__name) {
+  return 0;
+}
+
+ssize_t __wrap_read(int __fd, void *__buf, size_t __nbytes) {
+  if (read_ret_zero) {
+    read_ret_zero = 0;
+    return 0;
+  }
+
+  if (package_rest < __nbytes) {
+    size_t cp_rest = __nbytes - package_rest;
+    memset(__buf, 'e', package_rest);
+    s_file_pd_t fpd = {.signal = signal, .package_size = PACKAGE_SIZE};
+    memcpy(__buf + package_rest, &fpd, sizeof(s_file_pd_t));
+    memset(__buf + package_rest + sizeof(s_file_pd_t), 'e', cp_rest);
+    package_rest = PACKAGE_SIZE;
+    package_rest -= cp_rest - sizeof(s_file_pd_t);
+    size_t _package_rest = package_rest;
+    return __nbytes;
+  }
+
+  package_rest -= __nbytes;
+  memset(__buf, 'e', __nbytes);
+
+  return __nbytes;
+}
+
+ssize_t __wrap_write(int __fd, const void *__buf, size_t __n) {
+  if (write_ret_m_1)
+    return -1;
+  dbuf_write(__buf, __n, &dbuf);
+  return __n;
+}
+
+int32_t __wrap_db_save_file(session *s) { return 1; }
+
+void test__file_upload__normal(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);
+  char cont_mes[64];
+  str2snd = "finished\n";
+
+  const char *fmt_args[] = {"File %s is uploaded to the server\n", FNAME, NULL};
+  print_str_arr = fmt_args;
+
+  for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
+    file_upload(&sess);
+  }
+  assert_int_equal(sess.state, OP_WAIT);
+  assert_ptr_equal(sess.file, NULL);
+}
+
+void test__file_upload__no_read(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);
+  char cont_mes[64];
+  read_ret_zero = true;
+
+  const char *fmt_args[] = {"Error uploading file %s!\n", FNAME, NULL};
+  print_str_arr = fmt_args;
+
+  for (int i = 0; (sess.state != OP_WAIT) && (i < 2000000000); i++) {
+    file_upload(&sess);
+  }
+  assert_int_equal(sess.state, OP_WAIT);
+  assert_ptr_equal(sess.file, NULL);
+}
+
+void test__file_upload__canceled(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);
+  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;
+    file_upload(&sess);
+  }
+  assert_int_equal(sess.state, OP_WAIT);
+  assert_ptr_equal(sess.file, NULL);
+}
+
+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__file_upload__normal),
+      cmocka_unit_test(test__file_upload__no_read),
+      cmocka_unit_test(test__file_upload__canceled),
+  };
+
+  return cmocka_run_group_tests(tests, setup, tear_down);
+}