From eeef0e0f07fd8f048b3d71991cb05e7bdcd978c5 Mon Sep 17 00:00:00 2001
From: Piotr Jastrzebski <piotr@scylladb.com>
Date: Wed, 17 Jan 2018 18:39:38 +0100
Subject: [PATCH] Clean up imports in tests

Signed-off-by: Piotr Jastrzebski <piotr@scylladb.com>
---
 tests/flat_mutation_reader_assertions.hh | 1 +
 tests/mutation_source_test.cc            | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/flat_mutation_reader_assertions.hh b/tests/flat_mutation_reader_assertions.hh
index 150635d4f4..1c296cbfb8 100644
--- a/tests/flat_mutation_reader_assertions.hh
+++ b/tests/flat_mutation_reader_assertions.hh
@@ -23,6 +23,7 @@
 
 #include <boost/test/unit_test.hpp>
 #include "flat_mutation_reader.hh"
+#include "mutation_assertions.hh"
 
 // Intended to be called in a seastar thread
 class flat_reader_assertions {
diff --git a/tests/mutation_source_test.cc b/tests/mutation_source_test.cc
index c35bb499f1..1093b823c3 100644
--- a/tests/mutation_source_test.cc
+++ b/tests/mutation_source_test.cc
@@ -20,10 +20,9 @@
  */
 
 #include <set>
+#include <boost/test/unit_test.hpp>
 #include "partition_slice_builder.hh"
 #include "schema_builder.hh"
-#include "mutation_reader_assertions.hh"
-#include "mutation_assertions.hh"
 #include "mutation_source_test.hh"
 #include "counters.hh"
 #include "simple_schema.hh"