2 files changed, 22 insertions, 3 deletions
diff --git a/tests/sha256/buildfile b/tests/sha256/buildfile
index 627b78f..fb490d0 100644
--- a/tests/sha256/buildfile
+++ b/tests/sha256/buildfile
@@ -4,4 +4,10 @@
 
 import libs = libbutl%lib{butl}
 
+if ($cxx.features.modules && ($force_std_modules == true || $cxx.id != 'msvc'))
+{
+  import libs += libstd-modules%liba{std-modules}
+  cxx.poptions += -D__cpp_lib_modules
+}
+
 exe{driver}: {hxx cxx}{*} $libs
diff --git a/tests/sha256/driver.cxx b/tests/sha256/driver.cxx
index 2d2f963..191756c 100644
--- a/tests/sha256/driver.cxx
+++ b/tests/sha256/driver.cxx
@@ -1,12 +1,25 @@
-// file      : tests/triplet/driver.cxx -*- C++ -*-
+// file      : tests/sha256/driver.cxx -*- C++ -*-
 // copyright : Copyright (c) 2014-2017 Code Synthesis Ltd
 // license   : MIT; see accompanying LICENSE file
 
-#include <string>
 #include <cassert>
+
+#ifndef __cpp_lib_modules
+#include <string>
 #include <iostream>
+#endif
+
+// Other includes.
 
-#include <libbutl/sha256.hxx>
+#ifdef __cpp_modules
+#ifdef __cpp_lib_modules
+import std.core;
+import std.io;
+#endif
+import butl.sha256;
+#else
+#include <libbutl/sha256.mxx>
+#endif
 
 using namespace std;
 using namespace butl;