Minimized the test code
Also, modified the Completer interface to facilitate the minimization of the test code.
This commit is contained in:
parent
6430677f4f
commit
0110611996
@ -53,14 +53,14 @@ void ThreadMain( TaskStack &task_stack )
|
|||||||
} // unnamed namespace
|
} // unnamed namespace
|
||||||
|
|
||||||
|
|
||||||
Completer::Completer( const Pylist &candidates )
|
Completer::Completer( const std::vector< std::string > &candidates )
|
||||||
: threading_enabled_( false )
|
: threading_enabled_( false )
|
||||||
{
|
{
|
||||||
AddCandidatesToDatabase( candidates, "", "" );
|
AddCandidatesToDatabase( candidates, "", "" );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
Completer::Completer( const Pylist &candidates,
|
Completer::Completer( const std::vector< std::string > &candidates,
|
||||||
const std::string &filetype,
|
const std::string &filetype,
|
||||||
const std::string &filepath )
|
const std::string &filepath )
|
||||||
: threading_enabled_( false )
|
: threading_enabled_( false )
|
||||||
@ -91,18 +91,35 @@ void Completer::EnableThreading()
|
|||||||
void Completer::AddCandidatesToDatabase( const Pylist &new_candidates,
|
void Completer::AddCandidatesToDatabase( const Pylist &new_candidates,
|
||||||
const std::string &filetype,
|
const std::string &filetype,
|
||||||
const std::string &filepath )
|
const std::string &filepath )
|
||||||
|
{
|
||||||
|
int num_candidates = len( new_candidates );
|
||||||
|
std::vector< std::string > candidates;
|
||||||
|
candidates.reserve( num_candidates );
|
||||||
|
|
||||||
|
for (int i = 0; i < num_candidates; ++i)
|
||||||
|
{
|
||||||
|
candidates.push_back( extract< std::string >( new_candidates[ i ] ) );
|
||||||
|
}
|
||||||
|
|
||||||
|
AddCandidatesToDatabase( candidates, filetype, filepath );
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void Completer::AddCandidatesToDatabase(
|
||||||
|
const std::vector< std::string > &new_candidates,
|
||||||
|
const std::string &filetype,
|
||||||
|
const std::string &filepath )
|
||||||
{
|
{
|
||||||
std::vector< Candidate *> &candidates =
|
std::vector< Candidate *> &candidates =
|
||||||
GetCandidateVector( filetype, filepath );
|
GetCandidateVector( filetype, filepath );
|
||||||
|
|
||||||
int num_candidates = len( new_candidates );
|
int num_candidates = new_candidates.size();
|
||||||
candidates.clear();
|
candidates.clear();
|
||||||
candidates.reserve( num_candidates );
|
candidates.reserve( num_candidates );
|
||||||
|
|
||||||
std::string candidate_text;
|
|
||||||
for (int i = 0; i < num_candidates; ++i)
|
for (int i = 0; i < num_candidates; ++i)
|
||||||
{
|
{
|
||||||
candidate_text = extract< std::string >( new_candidates[ i ] );
|
const std::string &candidate_text = new_candidates[ i ];
|
||||||
Candidate *&candidate = GetValueElseInsert( candidate_repository_,
|
Candidate *&candidate = GetValueElseInsert( candidate_repository_,
|
||||||
candidate_text, NULL );
|
candidate_text, NULL );
|
||||||
if ( !candidate )
|
if ( !candidate )
|
||||||
@ -113,24 +130,26 @@ void Completer::AddCandidatesToDatabase( const Pylist &new_candidates,
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void Completer::CandidatesForQuery( const std::string &query,
|
std::vector< std::string > Completer::CandidatesForQuery(
|
||||||
Pylist &candidates ) const
|
const std::string &query ) const
|
||||||
{
|
{
|
||||||
CandidatesForQueryAndType( query, "", candidates );
|
return CandidatesForQueryAndType( query, "" );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void Completer::CandidatesForQueryAndType( const std::string &query,
|
std::vector< std::string > Completer::CandidatesForQueryAndType(
|
||||||
const std::string &filetype,
|
const std::string &query,
|
||||||
Pylist &candidates ) const
|
const std::string &filetype ) const
|
||||||
{
|
{
|
||||||
std::vector< Result > results;
|
std::vector< Result > results;
|
||||||
ResultsForQueryAndType( query, filetype, results );
|
ResultsForQueryAndType( query, filetype, results );
|
||||||
|
|
||||||
|
std::vector< std::string > candidates;
|
||||||
foreach ( const Result& result, results )
|
foreach ( const Result& result, results )
|
||||||
{
|
{
|
||||||
candidates.append( *result.Text() );
|
candidates.push_back( *result.Text() );
|
||||||
}
|
}
|
||||||
|
return candidates;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@ -156,6 +175,7 @@ Future Completer::CandidatesForQueryAndTypeAsync(
|
|||||||
return Future( move( future ) );
|
return Future( move( future ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
AsyncResults Completer::ResultsForQueryAndType(
|
AsyncResults Completer::ResultsForQueryAndType(
|
||||||
const std::string &query,
|
const std::string &query,
|
||||||
const std::string &filetype ) const
|
const std::string &filetype ) const
|
||||||
@ -165,6 +185,7 @@ AsyncResults Completer::ResultsForQueryAndType(
|
|||||||
return results;
|
return results;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void Completer::ResultsForQueryAndType( const std::string &query,
|
void Completer::ResultsForQueryAndType( const std::string &query,
|
||||||
const std::string &filetype,
|
const std::string &filetype,
|
||||||
std::vector< Result > &results ) const
|
std::vector< Result > &results ) const
|
||||||
|
@ -56,25 +56,30 @@ class Completer : boost::noncopyable
|
|||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
Completer() {}
|
Completer() {}
|
||||||
Completer( const Pylist &candidates );
|
Completer( const std::vector< std::string > &candidates );
|
||||||
Completer( const Pylist &candidates,
|
Completer( const std::vector< std::string > &candidates,
|
||||||
const std::string &filetype,
|
const std::string &filetype,
|
||||||
const std::string &filepath );
|
const std::string &filepath );
|
||||||
~Completer();
|
~Completer();
|
||||||
|
|
||||||
void EnableThreading();
|
void EnableThreading();
|
||||||
|
|
||||||
|
void AddCandidatesToDatabase(
|
||||||
|
const std::vector< std::string > &new_candidates,
|
||||||
|
const std::string &filetype,
|
||||||
|
const std::string &filepath );
|
||||||
|
|
||||||
void AddCandidatesToDatabase( const Pylist &new_candidates,
|
void AddCandidatesToDatabase( const Pylist &new_candidates,
|
||||||
const std::string &filetype,
|
const std::string &filetype,
|
||||||
const std::string &filepath );
|
const std::string &filepath );
|
||||||
|
|
||||||
// Only provided for tests!
|
// Only provided for tests!
|
||||||
void CandidatesForQuery( const std::string &query,
|
std::vector< std::string > CandidatesForQuery(
|
||||||
Pylist &candidates ) const;
|
const std::string &query ) const;
|
||||||
|
|
||||||
void CandidatesForQueryAndType( const std::string &query,
|
std::vector< std::string > CandidatesForQueryAndType(
|
||||||
const std::string &filetype,
|
const std::string &query,
|
||||||
Pylist &candidates ) const;
|
const std::string &filetype ) const;
|
||||||
|
|
||||||
Future CandidatesForQueryAndTypeAsync( const std::string &query,
|
Future CandidatesForQueryAndTypeAsync( const std::string &query,
|
||||||
const std::string &filetype ) const;
|
const std::string &filetype ) const;
|
||||||
|
@ -30,9 +30,14 @@ BOOST_PYTHON_MODULE(indexer)
|
|||||||
.def( "ResultsReady", &Future::ResultsReady )
|
.def( "ResultsReady", &Future::ResultsReady )
|
||||||
.def( "GetResults", &Future::GetResults );
|
.def( "GetResults", &Future::GetResults );
|
||||||
|
|
||||||
|
void (Completer::*actd) (const Pylist&,
|
||||||
|
const std::string&,
|
||||||
|
const std::string&) =
|
||||||
|
&Completer::AddCandidatesToDatabase;
|
||||||
|
|
||||||
class_< Completer, boost::noncopyable >( "Completer" )
|
class_< Completer, boost::noncopyable >( "Completer" )
|
||||||
.def( "EnableThreading", &Completer::EnableThreading )
|
.def( "EnableThreading", &Completer::EnableThreading )
|
||||||
.def( "AddCandidatesToDatabase", &Completer::AddCandidatesToDatabase )
|
.def( "AddCandidatesToDatabase", actd )
|
||||||
.def( "CandidatesForQueryAndTypeAsync",
|
.def( "CandidatesForQueryAndTypeAsync",
|
||||||
&Completer::CandidatesForQueryAndTypeAsync );
|
&Completer::CandidatesForQueryAndTypeAsync );
|
||||||
}
|
}
|
||||||
|
@ -30,19 +30,7 @@ namespace YouCompleteMe
|
|||||||
namespace
|
namespace
|
||||||
{
|
{
|
||||||
|
|
||||||
std::vector<std::string> ToStringVector( const boost::python::list &pylist )
|
std::vector< std::string > Candidates( const std::string &a,
|
||||||
{
|
|
||||||
std::vector<std::string> values;
|
|
||||||
for (int i = 0; i < boost::python::len( pylist ); ++i)
|
|
||||||
{
|
|
||||||
values.push_back(
|
|
||||||
boost::python::extract< std::string >( pylist[ i ] ) );
|
|
||||||
}
|
|
||||||
|
|
||||||
return values;
|
|
||||||
}
|
|
||||||
|
|
||||||
Pylist Candidates( const std::string &a,
|
|
||||||
const std::string &b = std::string(),
|
const std::string &b = std::string(),
|
||||||
const std::string &c = std::string(),
|
const std::string &c = std::string(),
|
||||||
const std::string &d = std::string(),
|
const std::string &d = std::string(),
|
||||||
@ -52,167 +40,124 @@ Pylist Candidates( const std::string &a,
|
|||||||
const std::string &h = std::string(),
|
const std::string &h = std::string(),
|
||||||
const std::string &i = std::string() )
|
const std::string &i = std::string() )
|
||||||
{
|
{
|
||||||
Pylist candidates;
|
std::vector< std::string > candidates;
|
||||||
candidates.append( a );
|
candidates.push_back( a );
|
||||||
if ( !b.empty() )
|
if ( !b.empty() )
|
||||||
candidates.append( b );
|
candidates.push_back( b );
|
||||||
if ( !c.empty() )
|
if ( !c.empty() )
|
||||||
candidates.append( c );
|
candidates.push_back( c );
|
||||||
if ( !d.empty() )
|
if ( !d.empty() )
|
||||||
candidates.append( d );
|
candidates.push_back( d );
|
||||||
if ( !e.empty() )
|
if ( !e.empty() )
|
||||||
candidates.append( e );
|
candidates.push_back( e );
|
||||||
if ( !f.empty() )
|
if ( !f.empty() )
|
||||||
candidates.append( f );
|
candidates.push_back( f );
|
||||||
if ( !g.empty() )
|
if ( !g.empty() )
|
||||||
candidates.append( g );
|
candidates.push_back( g );
|
||||||
if ( !h.empty() )
|
if ( !h.empty() )
|
||||||
candidates.append( h );
|
candidates.push_back( h );
|
||||||
if ( !i.empty() )
|
if ( !i.empty() )
|
||||||
candidates.append( i );
|
candidates.push_back( i );
|
||||||
|
|
||||||
return candidates;
|
return candidates;
|
||||||
}
|
}
|
||||||
|
|
||||||
} // unnamed namespace
|
} // unnamed namespace
|
||||||
|
|
||||||
class CompleterTest : public ::testing::Test
|
|
||||||
|
TEST( CompleterTest, OneCandidate )
|
||||||
{
|
{
|
||||||
protected:
|
EXPECT_THAT( Completer( Candidates(
|
||||||
virtual void SetUp()
|
"foobar" ) ).CandidatesForQuery( "fbr" ),
|
||||||
{
|
ElementsAre( "foobar" ) );
|
||||||
Py_Initialize();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
TEST_F( CompleterTest, OneCandidate )
|
|
||||||
{
|
|
||||||
Pylist results;
|
|
||||||
Completer( Candidates( "foobar" ) ).CandidatesForQuery( "fbr", results );
|
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ), ElementsAre( "foobar" ) );
|
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, ManyCandidateSimple )
|
TEST( CompleterTest, ManyCandidateSimple )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"foobar",
|
"foobar",
|
||||||
"foobartest",
|
"foobartest",
|
||||||
"Foobartest" ) ).CandidatesForQuery( "fbr", results );
|
"Foobartest" ) ).CandidatesForQuery( "fbr" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
WhenSorted( ElementsAre( "Foobartest",
|
WhenSorted( ElementsAre( "Foobartest",
|
||||||
"foobar",
|
"foobar",
|
||||||
"foobartest" ) ) );
|
"foobartest" ) ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, FirstCharSameAsQueryWins )
|
TEST( CompleterTest, FirstCharSameAsQueryWins )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"foobar",
|
"foobar",
|
||||||
"afoobar" ) ).CandidatesForQuery( "fbr", results );
|
"afoobar" ) ).CandidatesForQuery( "fbr" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "foobar",
|
ElementsAre( "foobar",
|
||||||
"afoobar" ) );
|
"afoobar" ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, CompleteMatchForWordBoundaryCharsWins )
|
TEST( CompleterTest, CompleteMatchForWordBoundaryCharsWins )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"FooBarQux",
|
"FooBarQux",
|
||||||
"FBaqux" ) ).CandidatesForQuery( "fbq", results );
|
"FBaqux" ) ).CandidatesForQuery( "fbq" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "FooBarQux",
|
ElementsAre( "FooBarQux",
|
||||||
"FBaqux" ) );
|
"FBaqux" ) );
|
||||||
|
|
||||||
Pylist results2;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"CompleterTest",
|
"CompleterTest",
|
||||||
"CompleteMatchForWordBoundaryCharsWins"
|
"CompleteMatchForWordBoundaryCharsWins" ) )
|
||||||
) ).CandidatesForQuery( "ct", results2 );
|
.CandidatesForQuery( "ct" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results2 ),
|
|
||||||
ElementsAre( "CompleterTest",
|
ElementsAre( "CompleterTest",
|
||||||
"CompleteMatchForWordBoundaryCharsWins" ) );
|
"CompleteMatchForWordBoundaryCharsWins" ) );
|
||||||
|
|
||||||
Pylist results3;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"FooBar",
|
"FooBar",
|
||||||
"FooBarRux" ) ).CandidatesForQuery( "fbr", results3 );
|
"FooBarRux" ) ).CandidatesForQuery( "fbr" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results3 ),
|
|
||||||
ElementsAre( "FooBarRux",
|
ElementsAre( "FooBarRux",
|
||||||
"FooBar" ) );
|
"FooBar" ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, RatioUtilizationTieBreak )
|
TEST( CompleterTest, RatioUtilizationTieBreak )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"FooBarQux",
|
"FooBarQux",
|
||||||
"FooBarQuxZaa" ) ).CandidatesForQuery( "fbq", results );
|
"FooBarQuxZaa" ) ).CandidatesForQuery( "fbq" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "FooBarQux",
|
ElementsAre( "FooBarQux",
|
||||||
"FooBarQuxZaa" ) );
|
"FooBarQuxZaa" ) );
|
||||||
|
|
||||||
Pylist results2;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"FooBar",
|
"FooBar",
|
||||||
"FooBarRux" ) ).CandidatesForQuery( "fba", results2 );
|
"FooBarRux" ) ).CandidatesForQuery( "fba" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results2 ),
|
|
||||||
ElementsAre( "FooBar",
|
ElementsAre( "FooBar",
|
||||||
"FooBarRux" ) );
|
"FooBarRux" ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, QueryPrefixOfCandidateWins )
|
TEST( CompleterTest, QueryPrefixOfCandidateWins )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"foobar",
|
"foobar",
|
||||||
"fbaroo" ) ).CandidatesForQuery( "foo", results );
|
"fbaroo" ) ).CandidatesForQuery( "foo" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "foobar",
|
ElementsAre( "foobar",
|
||||||
"fbaroo" ) );
|
"fbaroo" ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, ShorterCandidateWins )
|
TEST( CompleterTest, ShorterCandidateWins )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"FooBarQux",
|
"FooBarQux",
|
||||||
"FaBarQux" ) ).CandidatesForQuery( "fbq", results );
|
"FaBarQux" ) ).CandidatesForQuery( "fbq" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "FaBarQux",
|
ElementsAre( "FaBarQux",
|
||||||
"FooBarQux" ) );
|
"FooBarQux" ) );
|
||||||
|
|
||||||
Pylist results2;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"CompleterT",
|
"CompleterT",
|
||||||
"CompleterTest" ) ).CandidatesForQuery( "co", results2 );
|
"CompleterTest" ) ).CandidatesForQuery( "co" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results2 ),
|
|
||||||
ElementsAre( "CompleterT",
|
ElementsAre( "CompleterT",
|
||||||
"CompleterTest" ) );
|
"CompleterTest" ) );
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F( CompleterTest, SameLowercaseCandidateWins )
|
TEST( CompleterTest, SameLowercaseCandidateWins )
|
||||||
{
|
{
|
||||||
Pylist results;
|
EXPECT_THAT( Completer( Candidates(
|
||||||
Completer( Candidates(
|
|
||||||
"foobar",
|
"foobar",
|
||||||
"Foobar" ) ).CandidatesForQuery( "foo", results );
|
"Foobar" ) ).CandidatesForQuery( "foo" ),
|
||||||
|
|
||||||
EXPECT_THAT( ToStringVector( results ),
|
|
||||||
ElementsAre( "foobar",
|
ElementsAre( "foobar",
|
||||||
"Foobar" ) );
|
"Foobar" ) );
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user