summaryrefslogtreecommitdiffstats
path: root/searchlib/src/tests/common/location/geo_location_test.cpp
blob: 8093ea61697514d3cd362c095d2610ab8ca510d6 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

#include <stdio.h>
#include <vespa/searchlib/common/geo_location.h>
#include <vespa/searchlib/common/geo_location_spec.h>
#include <vespa/searchlib/common/geo_location_parser.h>
#include <vespa/vespalib/gtest/gtest.h>

using search::common::GeoLocation;
using search::common::GeoLocationParser;

bool is_parseable(const char *str) {
    GeoLocationParser parser;
    return parser.parseOldFormat(str);
}

GeoLocation parse(const char *str) {
    GeoLocationParser parser;
    EXPECT_TRUE(parser.parseOldFormat(str));
    return parser.getGeoLocation();
}

TEST(GeoLocationTest, malformed_bounding_boxes_are_not_parseable) {
    EXPECT_TRUE(is_parseable("[2,10,20,30,40]"));
    EXPECT_FALSE(is_parseable("[2,10,20,30,40][2,10,20,30,40]"));
    EXPECT_FALSE(is_parseable("[1,10,20,30,40]"));
    EXPECT_FALSE(is_parseable("[3,10,20,30,40]"));
    EXPECT_FALSE(is_parseable("[2, 10, 20, 30, 40]"));
    EXPECT_FALSE(is_parseable("[2,10,20,30,40"));
    EXPECT_FALSE(is_parseable("[2,10,20,30]"));
    EXPECT_FALSE(is_parseable("[10,20,30,40]"));
}

TEST(GeoLocationTest, malformed_circles_are_not_parseable) {
    EXPECT_TRUE(is_parseable("(2,10,20,5,0,0,0)"));
    EXPECT_FALSE(is_parseable("(2,10,20,5,0,0,0)(2,10,20,5,0,0,0)"));
    EXPECT_FALSE(is_parseable("(1,10,20,5,0,0,0)"));
    EXPECT_FALSE(is_parseable("(3,10,20,5,0,0,0)"));
    EXPECT_FALSE(is_parseable("(2, 10, 20, 5, 0, 0, 0)"));
    EXPECT_FALSE(is_parseable("(2,10,20,5)"));
    EXPECT_FALSE(is_parseable("(2,10,20,5,0,0,0"));
    EXPECT_FALSE(is_parseable("(2,10,20,5,0,0,0,1000"));
    EXPECT_FALSE(is_parseable("(10,20,5)"));
}

TEST(GeoLocationTest, bounding_boxes_can_be_parsed) {
    auto loc = parse("[2,10,20,30,40]");
    EXPECT_EQ(false, loc.has_point);
    EXPECT_EQ(true, loc.bounding_box.active());
    EXPECT_EQ(0u, loc.x_aspect.multiplier);
    EXPECT_EQ(0, loc.point.x);
    EXPECT_EQ(0, loc.point.y);
    EXPECT_EQ(std::numeric_limits<uint32_t>::max(), loc.radius);
    EXPECT_EQ(10, loc.bounding_box.x.low);
    EXPECT_EQ(20, loc.bounding_box.y.low);
    EXPECT_EQ(30, loc.bounding_box.x.high);
    EXPECT_EQ(40, loc.bounding_box.y.high);
}

TEST(GeoLocationTest, circles_can_be_parsed) {
    auto loc = parse("(2,10,20,5,0,0,0)");
    EXPECT_EQ(true, loc.has_point);
    EXPECT_EQ(true, loc.bounding_box.active());
    EXPECT_EQ(0u, loc.x_aspect.multiplier);
    EXPECT_EQ(10, loc.point.x);
    EXPECT_EQ(20, loc.point.y);
    EXPECT_EQ(5u, loc.radius);
    EXPECT_EQ(5, loc.bounding_box.x.low);
    EXPECT_EQ(15, loc.bounding_box.y.low);
    EXPECT_EQ(15, loc.bounding_box.x.high);
    EXPECT_EQ(25, loc.bounding_box.y.high);    
}

TEST(GeoLocationTest, circles_can_have_aspect_ratio) {
    auto loc = parse("(2,10,20,5,0,0,0,2147483648)");
    EXPECT_EQ(true, loc.has_point);
    EXPECT_EQ(true, loc.bounding_box.active());
    EXPECT_EQ(2147483648u, loc.x_aspect.multiplier);
    EXPECT_EQ(10, loc.point.x);
    EXPECT_EQ(20, loc.point.y);
    EXPECT_EQ(5u, loc.radius);
    EXPECT_EQ(-1, loc.bounding_box.x.low);
    EXPECT_EQ(15, loc.bounding_box.y.low);
    EXPECT_EQ(21, loc.bounding_box.x.high);
    EXPECT_EQ(25, loc.bounding_box.y.high);
}

TEST(GeoLocationTest, bounding_box_can_be_specified_after_circle) {
    auto loc = parse("(2,10,20,5,0,0,0)[2,10,20,30,40]");
    EXPECT_EQ(true, loc.has_point);
    EXPECT_EQ(true, loc.bounding_box.active());
    EXPECT_EQ(0u, loc.x_aspect.multiplier);
    EXPECT_EQ(10, loc.point.x);
    EXPECT_EQ(20, loc.point.y);
    EXPECT_EQ(5u, loc.radius);
    EXPECT_EQ(10, loc.bounding_box.x.low);
    EXPECT_EQ(20, loc.bounding_box.y.low);
    EXPECT_EQ(15, loc.bounding_box.x.high);
    EXPECT_EQ(25, loc.bounding_box.y.high);
}

TEST(GeoLocationTest, circles_can_be_specified_after_bounding_box) {
    auto loc = parse("[2,10,20,30,40](2,10,20,5,0,0,0)");
    EXPECT_EQ(true, loc.has_point);
    EXPECT_EQ(true, loc.bounding_box.active());
    EXPECT_EQ(0u, loc.x_aspect.multiplier);
    EXPECT_EQ(10, loc.point.x);
    EXPECT_EQ(20, loc.point.y);
    EXPECT_EQ(5u, loc.radius);
    EXPECT_EQ(10, loc.bounding_box.x.low);
    EXPECT_EQ(20, loc.bounding_box.y.low);
    EXPECT_EQ(15, loc.bounding_box.x.high);
    EXPECT_EQ(25, loc.bounding_box.y.high);    
}

TEST(GeoLocationTest, santa_search_gives_non_wrapped_bounding_box) {
    auto loc = parse("(2,122163600,89998536,290112,4,2000,0,109704)");
    EXPECT_GE(loc.bounding_box.x.high, loc.bounding_box.x.low);
    EXPECT_GE(loc.bounding_box.y.high, loc.bounding_box.y.low);
}

TEST(GeoLocationTest, near_boundary_search_gives_non_wrapped_bounding_box) {
    auto loc1 = parse("(2,2000000000,2000000000,3000000000,0,1,0)");
    EXPECT_GE(loc1.bounding_box.x.high, loc1.bounding_box.x.low);
    EXPECT_GE(loc1.bounding_box.y.high, loc1.bounding_box.y.low);
    EXPECT_EQ(std::numeric_limits<int32_t>::max(), loc1.bounding_box.y.high);
    EXPECT_EQ(std::numeric_limits<int32_t>::max(), loc1.bounding_box.y.high);    

    auto loc2 = parse("(2,-2000000000,-2000000000,3000000000,0,1,0)");
    EXPECT_GE(loc2.bounding_box.x.high, loc2.bounding_box.x.low);
    EXPECT_GE(loc2.bounding_box.y.high, loc2.bounding_box.y.low);
    EXPECT_EQ(std::numeric_limits<int32_t>::min(), loc2.bounding_box.x.low);
    EXPECT_EQ(std::numeric_limits<int32_t>::min(), loc2.bounding_box.y.low);
}

GTEST_MAIN_RUN_ALL_TESTS()