190 lines
5.0 KiB
C++
190 lines
5.0 KiB
C++
// Copyright (C) 2011, 2012 Strahinja Val Markovic <val@markovic.io>
|
|
//
|
|
// This file is part of YouCompleteMe.
|
|
//
|
|
// YouCompleteMe is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// YouCompleteMe is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with YouCompleteMe. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <gmock/gmock.h>
|
|
#include "Completer.h"
|
|
#include "Utils.h"
|
|
#include <boost/python.hpp>
|
|
|
|
using ::testing::ElementsAre;
|
|
using ::testing::WhenSorted;
|
|
|
|
namespace YouCompleteMe
|
|
{
|
|
|
|
namespace
|
|
{
|
|
|
|
std::vector<std::string> ToStringVector( const boost::python::list &pylist )
|
|
{
|
|
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 &c = std::string(),
|
|
const std::string &d = std::string(),
|
|
const std::string &e = std::string(),
|
|
const std::string &f = std::string(),
|
|
const std::string &g = std::string(),
|
|
const std::string &h = std::string(),
|
|
const std::string &i = std::string() )
|
|
{
|
|
Pylist candidates;
|
|
candidates.append( a );
|
|
if ( !b.empty() )
|
|
candidates.append( b );
|
|
if ( !c.empty() )
|
|
candidates.append( c );
|
|
if ( !d.empty() )
|
|
candidates.append( d );
|
|
if ( !e.empty() )
|
|
candidates.append( e );
|
|
if ( !f.empty() )
|
|
candidates.append( f );
|
|
if ( !g.empty() )
|
|
candidates.append( g );
|
|
if ( !h.empty() )
|
|
candidates.append( h );
|
|
if ( !i.empty() )
|
|
candidates.append( i );
|
|
|
|
return candidates;
|
|
}
|
|
|
|
} // unnamed namespace
|
|
|
|
class CompleterTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
virtual void SetUp()
|
|
{
|
|
Py_Initialize();
|
|
}
|
|
};
|
|
|
|
|
|
TEST_F( CompleterTest, OneCandidate )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates( "foobar" ) ).GetCandidatesForQuery( "fbr", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ), ElementsAre( "foobar" ) );
|
|
}
|
|
|
|
TEST_F( CompleterTest, ManyCandidateSimple )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates(
|
|
"foobar",
|
|
"foobartest",
|
|
"Foobartest" ) ).GetCandidatesForQuery( "fbr", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ),
|
|
WhenSorted( ElementsAre( "Foobartest",
|
|
"foobar",
|
|
"foobartest" ) ) );
|
|
}
|
|
|
|
TEST_F( CompleterTest, FirstCharSameAsQueryWins )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates(
|
|
"foobar",
|
|
"afoobar" ) ).GetCandidatesForQuery( "fbr", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ),
|
|
ElementsAre( "foobar",
|
|
"afoobar" ) );
|
|
}
|
|
|
|
TEST_F( CompleterTest, CompleteMatchForWordBoundaryCharsWins )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates(
|
|
"FooBarQux",
|
|
"FBaqux" ) ).GetCandidatesForQuery( "fbq", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ),
|
|
ElementsAre( "FooBarQux",
|
|
"FBaqux" ) );
|
|
|
|
Pylist results2;
|
|
Completer( Candidates(
|
|
"CompleterTest",
|
|
"CompleteMatchForWordBoundaryCharsWins"
|
|
) ).GetCandidatesForQuery( "ct", results2 );
|
|
|
|
EXPECT_THAT( ToStringVector( results2 ),
|
|
ElementsAre( "CompleterTest",
|
|
"CompleteMatchForWordBoundaryCharsWins" ) );
|
|
|
|
Pylist results3;
|
|
Completer( Candidates(
|
|
"FooBar",
|
|
"FooBarRux"
|
|
) ).GetCandidatesForQuery( "fbr", results3 );
|
|
|
|
EXPECT_THAT( ToStringVector( results3 ),
|
|
ElementsAre( "FooBarRux",
|
|
"FooBar" ) );
|
|
}
|
|
|
|
TEST_F( CompleterTest, RatioUtilizationTieBreak )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates(
|
|
"FooBarQux",
|
|
"FooBarQuxZaa" ) ).GetCandidatesForQuery( "fbq", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ),
|
|
ElementsAre( "FooBarQux",
|
|
"FooBarQuxZaa" ) );
|
|
}
|
|
|
|
TEST_F( CompleterTest, ShorterCandidateWins )
|
|
{
|
|
Pylist results;
|
|
Completer( Candidates(
|
|
"FooBarQux",
|
|
"FaBarQux" ) ).GetCandidatesForQuery( "fbq", results );
|
|
|
|
EXPECT_THAT( ToStringVector( results ),
|
|
ElementsAre( "FaBarQux",
|
|
"FooBarQux" ) );
|
|
|
|
Pylist results2;
|
|
Completer( Candidates(
|
|
"CompleterT",
|
|
"CompleterTest" ) ).GetCandidatesForQuery( "co", results2 );
|
|
|
|
EXPECT_THAT( ToStringVector( results2 ),
|
|
ElementsAre( "CompleterT",
|
|
"CompleterTest" ) );
|
|
}
|
|
|
|
} // namespace YouCompleteMe
|
|
|