load well statuses and well types with each request to the company licenses controller.

1 files changed, 10 insertions, 0 deletions

diff --git a/spec/controllers/v1/company_licenses_controller_spec.rb b/spec/controllers/v1/company_licenses_controller_spec.rb
index 2784a1d..c139f8b 100644
--- a/spec/controllers/v1/company_licenses_controller_spec.rb
+++ b/spec/controllers/v1/company_licenses_controller_spec.rb
@@ -27,5 +27,15 @@ describe V1::CompanyLicensesController do
       xhr :get, :index, company_id: company_id, township: "123"
       assigns(:active_licenses).should == active_licenses_in_township
     end
+
+    it "includes each well status" do
+      xhr :get, :index, company_id: company_id
+      assigns(:license_statuses).should =~ LicenseStatus::ALL
+    end
+
+    it "includes each well type" do
+      xhr :get, :index, company_id: company_id
+      assigns(:well_types).should =~ WellType::ALL
+    end
   end
 end