Add class method for icao codes, spec

timrogers · Nov 28, 2024 · bd96675 · bd96675
1 parent 719dc79
commit bd96675
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/airports.rb b/lib/airports.rb
@@ -29,6 +29,10 @@ def self.iata_codes
     parsed_data.keys
   end
 
+  def self.icao_codes
+    parsed_data.values.map { |airport_data| airport_data["icao"] }
+  end
+
   def self.all
     @all ||= parsed_data.values.map do |airport_data|
       airport_from_parsed_data_element(airport_data)

diff --git a/spec/airports_spec.rb b/spec/airports_spec.rb
@@ -60,6 +60,13 @@
     it { is_expected.to include("LHR") }
   end
 
+  describe ".icao_codes" do
+    subject { described_class.icao_codes }
+
+    it { is_expected.to be_a(Array) }
+    it { is_expected.to include("EGLL") }
+  end
+
   describe ".all" do
     subject { described_class.all }