summaryrefslogtreecommitdiff
path: root/extractor/mocks/mock_suffix_array.h
blob: 6886232afc81c2f04484f4b6bc80369b4f607c6e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#include <gmock/gmock.h>

#include <memory>
#include <string>

#include "../data_array.h"
#include "../phrase_location.h"
#include "../suffix_array.h"

using namespace std;

namespace extractor {

class MockSuffixArray : public SuffixArray {
 public:
  MOCK_CONST_METHOD0(GetSize, int());
  MOCK_CONST_METHOD0(GetData, shared_ptr<DataArray>());
  MOCK_CONST_METHOD0(BuildLCPArray, vector<int>());
  MOCK_CONST_METHOD1(GetSuffix, int(int));
  MOCK_CONST_METHOD4(Lookup, PhraseLocation(int, int, const string& word, int));
};

} // namespace extractor