queryContext.setQueryBuilder(queryBuilder);
        assertEquals(queryBuilder, queryContext.getQueryBuilder());
    }

    // Test addFunctionScore
    @Test
    public void test_addFunctionScore() {
        queryContext = new QueryContext("test", false);
        QueryBuilder initialQuery = QueryBuilders.termQuery("field", "value");
        queryContext.setQueryBuilder(initialQuery);