geolite 0.0.3

open source geocode solution
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

use super::*;
use prost::Message;

use crate::extract::pbf_fixtures::{blob_compression, make_blob};

fn decode(bytes: &[u8]) -> osm_data::blob_msg {
  osm_data::blob_msg::decode(bytes).expect("failed to decode blob")
}

// 00.00: blob sem compressao devolve os bytes originais intactos
#[test]
fn _00_00_returns_raw_bytes_when_blob_is_uncompressed() {
  let payload = b"conteudo do bloco".to_vec();
  let blob = decode(&make_blob(&payload, blob_compression::raw));

  assert_eq!(decompress_blob(&blob), payload);
}

// 00.01: blob zlib e inflado de volta ao payload original
#[test]
fn _00_01_inflates_zlib_blob_payload() {
  // repeating payload so that zlib actually compresses it
  let payload = b"osm".repeat(500);
  let bytes = make_blob(&payload, blob_compression::zlib);
  let blob = decode(&bytes);

  assert_eq!(decompress_blob(&blob), payload);
  assert!(
    bytes.len() < payload.len(),
    "o fixture deveria estar comprimido de fato"
  );
}

// 00.02: quando raw e zlib_data coexistem, raw tem precedencia
#[test]
fn _00_02_prefers_raw_over_zlib_when_both_are_present() {
  use std::io::Write;
  let mut encoder = flate2::write::ZlibEncoder::new(Vec::new(), flate2::Compression::default());
  encoder.write_all(b"do zlib").expect("failed to deflate");

  let blob = decode(
    &pbf_fixtures::blob_wire {
      raw: Some(b"do raw".to_vec()),
      raw_size: Some(6),
      zlib_data: Some(encoder.finish().expect("failed to finish deflate")),
    }
    .encode_to_vec(),
  );

  assert_eq!(decompress_blob(&blob), b"do raw");
}

// 00.03: blob sem raw e sem zlib_data nao tem como ser decodificado
#[test]
#[should_panic(expected = "unsupported blob compression")]
fn _00_03_panics_when_blob_has_neither_raw_nor_zlib() {
  let blob = decode(&make_blob(b"ignorado", blob_compression::none));
  decompress_blob(&blob);
}