Ticket #10976: QuotedFilenameQueryTest.cpp

#include <stdio.h>

#include <File.h>
#include <Path.h>
#include <Query.h>
#include <String.h>
#include <Volume.h>
#include <VolumeRoster.h>


int main(int, char **)
{
    BPath filePath("/boot/home", "querytest\".cpp");
    status_t error = filePath.InitCheck();
    if (error != B_OK) {
        fprintf(stderr, "Failed to initialize path: %s\n", strerror(error));
        return error;
    }

    BFile file(filePath.Path(), B_CREATE_FILE | B_WRITE_ONLY);

    error = file.InitCheck();
    if (error != B_OK) {
        fprintf(stderr, "Failed to create test file: %s\n", strerror(error));
        return error;
    }

    BVolumeRoster roster;
    BVolume volume;
    while (roster.GetNextVolume(&volume) == B_OK) {
        if (!volume.KnowsQuery())
            continue;

        char volumeName[B_FILE_NAME_LENGTH];
        volume.GetName(volumeName);
        printf("Evaluating volume: %s\n", volumeName);

        BQuery query;
        error = query.SetVolume(&volume);
        if (error != B_OK) {
            fprintf(stderr, "Failed to set query volume: %s\n",
                strerror(error));
            return error;
        }

        BString predicate;
        predicate.SetToFormat("name==%s", filePath.Leaf());
        error = query.SetPredicate(predicate);
        if (error != B_OK) {
            fprintf(stderr, "Failed to set explicit predicate: %s\n",
                strerror(error));
            return error;
        }

        error = query.Fetch();
        if (error != B_OK) {
            fprintf(stderr, "Failed to fetch explicit query results: %s\n",
                strerror(error));
            return error;
        }

        int32 count = 0;
        entry_ref ref;
        for (; query.GetNextRef(&ref) == B_OK; count++);

        printf("Explicit predicate '%s' returned %" B_PRId32 " entries.\n",
            predicate.String(), count);

        query.Clear();
        query.SetVolume(&volume);
        query.PushAttr("name");
        query.PushString(filePath.Leaf());
        query.PushOp(B_EQ);

        error = query.GetPredicate(&predicate);
        if (error != B_OK) {
            fprintf(stderr, "Failed to get RPN predicate: %s\n",
                strerror(error));
            return error;
        }

        printf("RPN-generated predicate: %s\n", predicate.String());

        error = query.Fetch();
        if (error != B_OK) {
            fprintf(stderr, "Failed to fetch RPN query results: %s\n",
                strerror(error));
            return error;
        }

        count = 0;
        for (; query.GetNextRef(&ref) == B_OK; count++);

        printf("RPN predicate returned %" B_PRId32 " entries.\n\n", count);
        query.Clear();
    }

    return 0;
}