[21_5] make string_u16 iterable with range based for loop

XmacsLabs · May 20, 2024 · 2e89509 · 2e89509
1 parent f75cd8c
commit 2e89509
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lolly/data/string_u16.hpp b/lolly/data/string_u16.hpp
@@ -80,6 +80,8 @@ class string_u16 {
     rep->resize (size);
     return rep->a;
   }
+  char16_t* begin () { return rep->a; }
+  char16_t* end () { return rep->a + rep->n; }
 
   inline operator string_u16_view () {
     return string_u16_view (rep->a, rep->n);

diff --git a/tests/lolly/data/string_u16_test.cpp b/tests/lolly/data/string_u16_test.cpp
@@ -158,4 +158,14 @@ TEST_CASE ("test copy") {
 
   string_u16 str3= copy (str1 (0, 3));
   CHECK (str3 == u"str");
+}
+
+TEST_CASE ("test iteration") {
+  string_u16 str1         = u"s\0tr1";
+  int        expectedIndex= 0;
+  for (const auto element : str1) {
+    CHECK_EQ (element, str1[expectedIndex]);
+    ++expectedIndex;
+  }
+  CHECK_EQ (expectedIndex, N (str1));
 }