summaryrefslogtreecommitdiff
path: root/extractor/mocks/mock_suffix_array.h
blob: 7018acc73db686bf91eb41ec70f6a649c92314b4 (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