summaryrefslogtreecommitdiff
path: root/extractor/mocks/mock_vocabulary.h
diff options
context:
space:
mode:
Diffstat (limited to 'extractor/mocks/mock_vocabulary.h')
-rw-r--r--extractor/mocks/mock_vocabulary.h13
1 files changed, 13 insertions, 0 deletions
diff --git a/extractor/mocks/mock_vocabulary.h b/extractor/mocks/mock_vocabulary.h
new file mode 100644
index 00000000..042c9ce2
--- /dev/null
+++ b/extractor/mocks/mock_vocabulary.h
@@ -0,0 +1,13 @@
+#include <gmock/gmock.h>
+
+#include "vocabulary.h"
+
+namespace extractor {
+
+class MockVocabulary : public Vocabulary {
+ public:
+ MOCK_METHOD1(GetTerminalValue, string(int word_id));
+ MOCK_METHOD1(GetTerminalIndex, int(const string& word));
+};
+
+} // namespace extractor