Restart!

1 files changed, 16 insertions, 7 deletions

diff --git a/tests/piece-table-insert.c b/tests/piece-table-insert.c
index bed55bc..90a97a2 100644
--- a/tests/piece-table-insert.c
+++ b/tests/piece-table-insert.c
@@ -1,10 +1,19 @@
+#include <string.h>
 #include "../src/piece-table.h"
 
-int main() {
-    piece_table pt;
-    init_piece_table(&pt, "1234567890");
-    insert_piece_table(&pt, 4, "abcdefgh", 8);
-    print_piece_table(&pt);
-    free_piece_table(&pt);
-    return 0;
+int
+main()
+{
+  piece_table *pt = piece_table_create("1234567890");
+  piece_table_insert(pt, 10, "abcdefgh", 8);
+  piece_table_insert(pt, 18, "abcdefgh", 8);
+  if (strcmp(piece_table_to_string(pt), "1234567890abcdefghabcdefgh"))
+    return 1;
+  /* TODO: must have two pieces if optimized properly
+     1234567890
+     abcdefghabcdefgh
+
+     TODO: check other things too */
+  piece_table_destroy(pt);
+  return 0;
 }