| 1 | #include "wvtest.h" | ||
| 2 | #include "tripstop.h" | ||
| 3 | #include <stdio.h> | ||
| 4 | |||
| 5 | using namespace std; | ||
| 6 | using namespace tr1; | ||
| 7 | |||
| 8 | |||
| 9 | WVTEST_MAIN("save/load") | ||
| 10 | { | ||
| 11 | TripStop t1(1, TripStop::OSM, 44.5f, 54.4f); | ||
| 12 | t1.add_triphop(500, 550, 0, 0, 0, 0); | ||
| 13 | t1.add_triphop(550, 600, 0, 0, 0, 0); | ||
| 14 | |||
| 15 | char *tmpname = tmpnam(NULL); // security issues in unit tests? bah. | ||
| 16 | unlink(tmpname); | ||
| 17 | FILE *fp1 = fopen(tmpname, "w"); | ||
| 18 | t1.write(fp1); | ||
| 19 | fclose(fp1); | ||
| 20 | |||
| 21 | FILE *fp2 = fopen(tmpname, "r"); | ||
| 22 | TripStop t2(fp2); | ||
| 23 | |||
| 24 | WVPASSEQ(t2.id, t1.id); | ||
| 25 | WVPASSEQ(t2.type, t1.type); | ||
| 26 | WVPASSEQ(t2.lat, t1.lat); | ||
| 27 | WVPASSEQ(t2.lng, t1.lng); | ||
| 28 | shared_ptr<TripStop::ServiceDict> tdict = t2.tdict; | ||
| 29 | WVPASSEQ(tdict->size(), 1); | ||
| 30 | WVPASSEQ(((*tdict))[0].size(), 1); | ||
| 31 | WVPASSEQ(((*tdict))[0][0].size(), 2); | ||
| 32 | WVPASSEQ(((*tdict))[0][0][0].start_time, 500); | ||
| 33 | WVPASSEQ(((*tdict))[0][0][1].start_time, 550); | ||
| 34 | |||
| 35 | fclose(fp2); | ||
| 36 | } | ||
| 37 | |||
| 38 | |||
| 39 | WVTEST_MAIN("get_multiple_triphops") | ||
| 40 | { | ||
| 41 | TripStop t; | ||
| 42 | t.add_triphop(500, 550, 0, 0, 0, 0); | ||
| 43 | t.add_triphop(550, 600, 0, 0, 0, 0); | ||
| 44 | t.add_triphop(600, 650, 0, 0, 0, 0); | ||
| 45 | t.add_triphop(600, 650, 0, 0, 0, 1); | ||
| 46 | |||
| 47 | // Ask for different amounts... | ||
| 48 | |||
| 49 | vector<TripHop> v = t.find_triphops(499, 0, 0, 3); | ||
| 50 | WVPASSEQ(v.size(), 3); | ||
| 51 | WVPASSEQ(v[0].start_time, 500); | ||
| 52 | WVPASSEQ(v[1].start_time, 550); | ||
| 53 | WVPASSEQ(v[2].start_time, 600); | ||
| 54 | |||
| 55 | v = t.find_triphops(499, 0, 0, 2); | ||
| 56 | WVPASSEQ(v.size(), 2); | ||
| 57 | WVPASSEQ(v[0].start_time, 500); | ||
| 58 | WVPASSEQ(v[1].start_time, 550); | ||
| 59 | |||
| 60 | v = t.find_triphops(499, 0, 0, 4); | ||
| 61 | WVPASSEQ(v.size(), 3); | ||
| 62 | WVPASSEQ(v[0].start_time, 500); | ||
| 63 | WVPASSEQ(v[1].start_time, 550); | ||
| 64 | WVPASSEQ(v[2].start_time, 600); | ||
| 65 | |||
| 66 | v = t.find_triphops(551, 0, 0, 2); | ||
| 67 | WVPASSEQ(v.size(), 1); | ||
| 68 | WVPASSEQ(v[0].start_time, 600); | ||
| 69 | |||
| 70 | } | ||
| 71 |