bytestring: Implement bytestring_spn() and bytestring_cspn()

[ccan] / ccan / bytestring / test / run.c

diff --git a/ccan/bytestring/test/run.c b/ccan/bytestring/test/run.c
index ef149c56ba7c27a1a370ddc101e0833cd102e52b..23415df58c3f4c5fa11d631f8951c6eb47a02c8f 100644 (file)
--- a/ccan/bytestring/test/run.c
+++ b/ccan/bytestring/test/run.c
@@ -3,6 +3,8 @@
 #include <ccan/bytestring/bytestring.h>
 #include <ccan/tap/tap.h>
 
+#include <ccan/bytestring/bytestring.c>
+
 #define TEST_STRING    "test string"
 #define TEST_STRING_2  "abc\0def"
 
@@ -14,7 +16,7 @@ int main(void)
        struct bytestring bs, bs1, bs2, bs3, bs4, bs5, bs6;
 
        /* This is how many tests you plan to run */
-       plan_tests(47);
+       plan_tests(53);
 
        bs = bytestring(str1, sizeof(str1) - 1);
        ok1(bs.ptr == str1);
@@ -88,6 +90,16 @@ int main(void)
        ok1(bytestring_eq(bytestring_bytestring(bs2, bytestring_NULL),
                          bytestring(bs2.ptr, 0)));
 
+
+       ok1(bytestring_spn(bs1, BYTESTRING("est")) == 4);
+       ok1(bytestring_cspn(bs1, BYTESTRING(" ")) == 4);
+
+       ok1(bytestring_spn(bs2, BYTESTRING("z")) == 0);
+       ok1(bytestring_cspn(bs2, BYTESTRING("\0")) == 3);
+
+       ok1(bytestring_spn(bs1, BYTESTRING("eginrst ")) == bs1.len);
+       ok1(bytestring_cspn(bs2, BYTESTRING("z")) == bs2.len);
+
        /* This exits depending on whether all tests passed */
        return exit_status();
 }