1 files changed, 17 insertions, 0 deletions
diff --git a/extractor/mocks/mock_suffix_array.h b/extractor/mocks/mock_suffix_array.h
new file mode 100644
index 00000000..38d8bad6
--- /dev/null
+++ b/extractor/mocks/mock_suffix_array.h
@@ -0,0 +1,17 @@
+#include <gmock/gmock.h>
+
+#include <string>
+
+#include "../data_array.h"
+#include "../phrase_location.h"
+#include "../suffix_array.h"
+
+using namespace std;
+
+class MockSuffixArray : public SuffixArray {
+ public:
+  MockSuffixArray() : SuffixArray(make_shared<DataArray>()) {}
+
+  MOCK_CONST_METHOD0(GetSize, int());
+  MOCK_CONST_METHOD4(Lookup, PhraseLocation(int, int, const string& word, int));
+};