diff options
| author | Teddy Wing | 2018-03-13 12:04:19 +0100 |
|---|---|---|
| committer | Teddy Wing | 2018-03-14 16:38:40 +0100 |
| commit | 11894d94b6698590a9ddb37729fdd90ea081159d (patch) | |
| tree | 513946bd479fcbf88d0d32ab9f39a4d1853a4070 | |
| parent | 87947a7b6da52305029954341c6dc40b8cde6f86 (diff) | |
| download | chouette-core-11894d94b6698590a9ddb37729fdd90ea081159d.tar.bz2 | |
Add `WayCostCollectionJSONSerializer`
A JSON serialiser for a collection of `WayCost`s. This will be used to
store `WayCost`s for a `Chouette::Route` in the `Route#costs` JSON
field.
Refs #6095
| -rw-r--r-- | lib/way_cost_collection_json_serializer.rb | 16 | ||||
| -rw-r--r-- | spec/lib/way_cost_collection_json_serializer_spec.rb | 37 |
2 files changed, 53 insertions, 0 deletions
diff --git a/lib/way_cost_collection_json_serializer.rb b/lib/way_cost_collection_json_serializer.rb new file mode 100644 index 000000000..191871cca --- /dev/null +++ b/lib/way_cost_collection_json_serializer.rb @@ -0,0 +1,16 @@ +class WayCostCollectionJSONSerializer + def self.dump(way_costs) + return if way_costs.nil? + + costs_by_id = {} + + way_costs.each do |way_cost| + costs_by_id[way_cost.id] = { + distance: way_cost.distance, + time: way_cost.time + } + end + + JSON.dump(costs_by_id) + end +end diff --git a/spec/lib/way_cost_collection_json_serializer_spec.rb b/spec/lib/way_cost_collection_json_serializer_spec.rb new file mode 100644 index 000000000..22faf88d3 --- /dev/null +++ b/spec/lib/way_cost_collection_json_serializer_spec.rb @@ -0,0 +1,37 @@ +RSpec.describe WayCostCollectionJSONSerializer do + describe ".dump" do + it "creates a JSON hash of hashed WayCost attributes" do + way_costs = [ + WayCost.new( + departure: Geokit::LatLng.new(48.85086, 2.36143), + arrival: Geokit::LatLng.new(47.91231, 1.87606), + distance: 1234, + time: 99, + id: '1-2' + ), + WayCost.new( + departure: Geokit::LatLng.new(47.91231, 1.87606), + arrival: Geokit::LatLng.new(52.50867, 13.42879), + distance: 5678, + time: 999, + id: '2-3' + ) + ] + + expect( + WayCostCollectionJSONSerializer.dump(way_costs) + ).to eq(<<-JSON.delete(" \n")) + { + "1-2": { + "distance": 1234, + "time": 99 + }, + "2-3": { + "distance": 5678, + "time": 999 + } + } + JSON + end + end +end |
