fix vector write extend

Author: bbSnavy

src/vector.c

@@ -104,13 +104,13 @@ t_ion_result_code vector_write(t_ion_vector *self, void *src, size_t len) {
   len_total = self->unit * len;
   curr_total = self->unit * self->curr_w;
 
-  if ((self->curr_w + len_total) >= self->size) {
+  while ((self->curr_w + len) >= self->size) {
     result = vector_extend(self);
     if (result != RESULT_OK)
       return result;
   }
 
-  memcpy(self->body + (self->unit * self->curr_w), src, len_total);
+  memcpy(self->body + curr_total, src, len_total);
   self->curr_w += len;
 
   return RESULT_OK;

src/vector_test.c

@@ -12,6 +12,67 @@ Test(vector, vector_create) {
   cr_expect(vector->size > 0);
   cr_expect(vector->curr_r == 0);
   cr_expect(vector->curr_w == 0);
+
+  free(vector_consume(vector));
+}
+
+Test(vector, vector_consume) {
+  t_ion_vector *vector;
+  void *data;
+
+  vector = vector_new(sizeof(int32_t));
+  cr_expect(vector != NULL);
+
+  data = vector_consume(vector);
+  cr_expect(data != NULL);
+
+  free(data);
+}
+
+Test(vector, vector_write_loop) {
+  t_ion_result_code result;
+  t_ion_vector *vector;
+  void *data;
+  int32_t x;
+
+  vector = vector_new(sizeof(int32_t));
+  cr_expect(vector != NULL);
+
+  for (x = 0; x < 1024; x++) {
+    cr_expect(vector->curr_w == x);
+    cr_expect(vector->curr_r == 0);
+    result = vector_write(vector, &x, 1);
+    cr_expect(result == RESULT_OK);
+    cr_expect(vector->curr_w == x + 1);
+    cr_expect(vector->curr_r == 0);
+  }
+
+  free(vector_consume(vector));
+}
+
+Test(vector, vector_write_array) {
+  t_ion_result_code result;
+  t_ion_vector *vector;
+  int32_t t[1000000] = {0};
+  int32_t o[1000000] = {0};
+  int32_t x;
+
+  vector = vector_new(sizeof(int32_t));
+  cr_expect(vector != NULL);
+
+  for (x = 0; x < 1000000; x++) {
+    t[x] = x;
+  }
+
+  result = vector_write(vector, t, 1000000);
+  cr_expect(result == RESULT_OK);
+
+  result = vector_read(vector, o, 1000000);
+  cr_expect(result == RESULT_OK);
+
+  for (x = 0; x < 1000000; x++) {
+    cr_expect(t[x] == o[x]);
+  }
 }
 
 #endif