improve code

spec/factories/inventory.rb

@@ -5,15 +5,11 @@
     name { "Inventory#{id}" }
     owner { create(:user) }
 
-    trait :with_host do
-      hosts { create_list(:host, 1, inventory_id: id) }
-    end
-
     trait :with_hosts do
-      hosts { create_list(:host, 12, inventory_id: id) }
+      hosts { create_list(:host, 3, inventory_id: id) }
     end
 
-    factory :small_inventory, traits: [:with_host]
+    factory :small_inventory, traits: [:with_hosts]
     factory :big_inventory, traits: [:with_hosts]
   end
 

spec/requests/apis_spec.rb

@@ -31,16 +31,18 @@
         expect(response.body).to eq @expected
       end
 
-      it "returns a inventory with host" do
+      it "returns a inventory with hosts" do
         @expected = {
-          all: [small_inventory.hosts[0].alias],
+          all: small_inventory.hosts.map(&:alias),
         }.to_json
 
         get "/api/v1/inventory/#{small_inventory.key}?token=#{admin.api_key}"
 
         expect(response.status).to eq 200
         expect(response.body).to eq @expected
       end
+
+      it "returns an inventory with groups"
     end
 
     describe "GET host" do