测试传递给NEST的参数（elasticsearch）

Question

我正在使用NEST来执行elasticsearch查询。

    public IReadOnlyCollection<IHit<Recommendation>> GetRecommendations(
        RecommenderQueryFields shoulds,
        RecommenderQueryFields musts, 
        RecommenderQueryFields mustNots)
    {
        var boolQuery = new BoolQuery();
        boolQuery.Should = GetQueryContainers(shoulds);
        boolQuery.Must = GetQueryContainers(musts);
        boolQuery.MustNot = GetQueryContainers(mustNots);

        var response = _elasticClient.Search<Recommendation>(s => s
            .AllTypes().From(0).Size(10)
            .Query(outerQuery => boolQuery));

        return response.Hits;
    }

我在GetQueryContainers方法中有逻辑，我想测试。有没有办法检查传递给boolQuery对象中的ElasticClient的内容？

我已经尝试了以下的东西，使用NUnit和NSubstitute。

    [Test]
    public void Test1()
    {
        // Arrange
        var searchResponse = Substitute.For<ISearchResponse<Recommendation>>();
        searchResponse.Hits.Returns(new List<IHit<Recommendation>>());
        var elasticClient = Substitute.For<IElasticClient>();

        var sut = new Recommender(elasticClient);

        // Act
        sut.GetRecommendations(null, null, null);

        // Assert
        elasticClient
            .Received(1)
            .Search(Arg.Is<Func<SearchDescriptor<Recommendation>, ISearchRequest>>(x => true));
    }

在Arg.Is<[...]>(x => true)我想替换true常数对boolQuery进行一些检查。但我不知道它是否可行或是如何完成的。或者有不同的方法来做到这一点？

Answer 1

TL; DR使用派生的QueryVisitor。见下面的Edit2。

发现问题已经得到解答。它与NEST无关，而是测试lambda表达式。这是不可能的：C# Moq Unit Testing with Lambda Expression or Func delegate

可以做的是测试将发送到elasticsearch的JSON请求，但是你需要实际的ElasticClient：ElasticSearch NEST 5.6.1 Query for unit test

可以做的是将逻辑放在自己的方法/类中。但是你只是为了测试而编写代码，我不喜欢它。喜欢：

public BoolQuery GetBoolQuery(RecommenderQueryFields shoulds, RecommenderQueryFields musts,
    RecommenderQueryFields mustNots)
{
    var boolQuery = new BoolQuery();
    boolQuery.Should = GetQueryContainers(shoulds);
    boolQuery.Must = GetQueryContainers(musts);
    boolQuery.MustNot = GetQueryContainers(mustNots);
    return boolQuery;
}

您正在暴露一种您不打算使用的public方法，仅用于测试。但是你可以这样在boolQuery上断言：

[Test]
public void GetRecommendations_CallsElasticSearch()
{
    // Arrange
    var elasticClient = Substitute.For<IElasticClient>();
    var sut = new Recommender(elasticClient);

    // Act
    var boolQuery = sut.GetBoolQuery(new RecommenderQueryFields{BlackListedFor = new List<string>{"asdf"}}, null, null);

    // Assert
    Assert.AreEqual(1, boolQuery.Should.Count());
}

在boolQuery.Should是一个QueryContainer列表，这是不可测试的，因为它是由lambdas生成的。虽然总比没有好，但它仍然不是测试NEST的干净方法。

Edit

@Russ Cam在评论中提到了IQueryContainer和QueryVisitor我得到了什么：

[Test]
public void test()
{
    // Arrange
    var fieldValue = "asdf";
    var elasticClient = Substitute.For<IElasticClient>();
    var sut = new Recommender(elasticClient);

    // Act
    var boolQuery = sut.GetBoolQuery(new RecommenderQueryFields { BlackListedFor = new List<string> { fieldValue } }, null, null);

    // Assert
    IQueryContainer qc = boolQuery.Should.First(); // Cast to IQueryContainer
    Assert.AreEqual(fieldValue, qc.Match.Query); // Assert value

    // Get "field name"
    var queryVisitor = new QueryVisitor();
    var prettyVisitor = new DslPrettyPrintVisitor(new ConnectionSettings(new InMemoryConnection()));
    qc.Accept(queryVisitor);
    qc.Accept(prettyVisitor);
    Assert.AreEqual(0, queryVisitor.Depth);
    Assert.AreEqual(VisitorScope.Query, queryVisitor.Scope);
    Assert.AreEqual("query: match (field: blacklistedfor.keyword)\r\n", prettyVisitor.PrettyPrint);
}

可以通过IQueryContainer访问该字段的值。

我尝试了QueryVisitor和DslPrettyPrintVisitor。第一个没有提供任何有用的信息。它有0深度，它是一个查询？我已经知道了。对于第二个，我可以断言一些额外的信息，比如字段名称（blacklistedfor）和后缀（keyword）。在字符串表示上断言并不完美，但总比没有好。

Edit2

@Russ Cam给了我一个我非常满意的解决方案。它使用派生的QueryVisitor：

public class MatchQueryVisitor : QueryVisitor
{
    public string Field { get; private set; }
    public string Value { get; private set; }

    public override void Visit(IMatchQuery query)
    {
        var inferrer = new Inferrer(new ConnectionSettings(new InMemoryConnection()));
        Field = inferrer.Field(query.Field);
        Value = query.Query;
    }
}

[Test]
public void test()
{
    // Arrange
    var fieldValue = "asdf";
    var elasticClient = Substitute.For<IElasticClient>();
    var sut = new Recommender(elasticClient);

    // Act
    var boolQuery = sut.GetBoolQuery(new RecommenderQueryFields { BlackListedFor = new List<string> { fieldValue } }, null,
        null);

    // Assert
    IQueryContainer qc = boolQuery.Should.First();
    var queryVisitor = new MatchQueryVisitor();
    qc.Accept(queryVisitor);
    Assert.AreEqual(fieldValue, queryVisitor.Value);
    Assert.AreEqual("blacklistedfor.keyword", queryVisitor.Field);
}

所以在MatchQueryVisitor中，它得到了Field和Value，然后在测试方法中断言。