From 757541f9d2f22faebf18665c3b6fe03203b428cf Mon Sep 17 00:00:00 2001
From: Kohei Takahashi <flast@flast.jp>
Date: Wed, 18 Apr 2018 22:39:55 +0900
Subject: [PATCH] Updated c-style array test for associative container

---
 test/sequence/github-176.cpp | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/test/sequence/github-176.cpp b/test/sequence/github-176.cpp
index 553e83c2..82960e9a 100644
--- a/test/sequence/github-176.cpp
+++ b/test/sequence/github-176.cpp
@@ -6,9 +6,12 @@
 ==============================================================================*/
 
 #include <boost/fusion/sequence/intrinsic/at.hpp>
+#include <boost/fusion/sequence/intrinsic/at_key.hpp>
 #include <boost/fusion/container/vector.hpp>
 #include <boost/fusion/container/list.hpp>
 #include <boost/fusion/container/deque.hpp>
+#include <boost/fusion/container/map.hpp>
+#include <boost/fusion/container/set.hpp>
 #include <boost/fusion/tuple/tuple.hpp>
 #include <boost/core/lightweight_test.hpp>
 
@@ -39,6 +42,36 @@ void test_at()
     BOOST_TEST(boost::fusion::at_c<0>(seq)[2] == 6);
 }
 
+template <typename T> inline T& value(T& v) { return v; }
+template <typename K, typename T> inline T& value(boost::fusion::pair<K, T>& v) { return v.second; }
+
+template <typename Sequence>
+void test_at_key()
+{
+    Sequence seq;
+
+    // zero initialized
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[0] == 0);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[1] == 0);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[2] == 0);
+
+    int (&arr)[3] = value(boost::fusion::deref(boost::fusion::begin(seq)));
+
+    arr[0] = 2;
+    arr[1] = 4;
+    arr[2] = 6;
+
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[0] == 2);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[1] == 4);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[2] == 6);
+
+    boost::fusion::at_key<int[3]>(seq)[1] = 42;
+
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[0] == 2);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[1] == 42);
+    BOOST_TEST(boost::fusion::at_key<int[3]>(seq)[2] == 6);
+}
+
 int main()
 {
     using namespace boost::fusion;
@@ -47,4 +80,7 @@ int main()
     test_at<deque<int[3]> >();
     test_at<list<int[3]> >();
     test_at<tuple<int[3]> >();
+
+    test_at_key<set<int[3]> >();
+    test_at_key<map<pair<int[3], int[3]> > >();
 }