python: Use closest match for countries

[people/ms/libloc.git] / src / test-as.c

diff --git a/src/test-as.c b/src/test-as.c
index d545d9c39700708e6d3c422be3d01a19a60d6081..b531173c38e3812d08f96be3887e712bc268c487 100644 (file)
--- a/src/test-as.c
+++ b/src/test-as.c
@@ -34,7 +34,7 @@ int main(int argc, char** argv) {
 
        // Create a database
        struct loc_writer* writer;
-       err = loc_writer_new(ctx, &writer);
+       err = loc_writer_new(ctx, &writer, NULL);
        if (err < 0)
                exit(EXIT_FAILURE);
 
@@ -49,7 +49,7 @@ int main(int argc, char** argv) {
                loc_as_unref(as);
        }
 
-       FILE* f = fopen("test.db", "w");
+       FILE* f = fopen("test.db", "w+");
        if (!f) {
                fprintf(stderr, "Could not open file for writing: %s\n", strerror(errno));
                exit(EXIT_FAILURE);
@@ -106,11 +106,20 @@ int main(int argc, char** argv) {
 
        loc_database_enumerator_set_string(enumerator, "10");
 
-       as = loc_database_enumerator_next_as(enumerator);
+       err = loc_database_enumerator_next_as(enumerator, &as);
+       if (err) {
+               fprintf(stderr, "Could not enumerate next AS\n");
+               exit(EXIT_FAILURE);
+       }
+
        while (as) {
                printf("Found AS%d: %s\n", loc_as_get_number(as), loc_as_get_name(as));
 
-               as = loc_database_enumerator_next_as(enumerator);
+               err = loc_database_enumerator_next_as(enumerator, &as);
+               if (err) {
+                       fprintf(stderr, "Could not enumerate next AS\n");
+                       exit(EXIT_FAILURE);
+               }
        }
 
        loc_database_enumerator_unref(enumerator);