filter by license status and township

1 files changed, 11 insertions, 3 deletions

diff --git a/spec/controllers/v1/company_licenses_controller_spec.rb b/spec/controllers/v1/company_licenses_controller_spec.rb
index fb0972a..2784a1d 100644
--- a/spec/controllers/v1/company_licenses_controller_spec.rb
+++ b/spec/controllers/v1/company_licenses_controller_spec.rb
@@ -4,20 +4,28 @@ describe V1::CompanyLicensesController do
   describe :index do
     let(:company) { Object.new }
     let(:company_id) { SecureRandom.uuid }
-    let(:active_licenses) { [] }
+    let(:active_licenses) { ["active"] }
+    let(:active_licenses_in_township) { ["active township"] }
 
     before :each do
       Company.stub(:find).with(company_id).and_return(company)
-      company.stub(:active_licenses).and_return(active_licenses)
-      xhr :get, :index, company_id: company_id
+      company.stub(:status).with(LicenseStatus::ACTIVE).and_return(active_licenses)
+      active_licenses.stub(:township).with("123").and_return(active_licenses_in_township)
     end
 
     it "returns the active licenses" do
+      xhr :get, :index, company_id: company_id
       assigns(:active_licenses).should == active_licenses
     end
 
     it "returns a json response" do
+      xhr :get, :index, company_id: company_id
       expect(-> { JSON.parse(response.body) }).not_to raise_error
     end
+
+    it "returns the active licenses for a given township" do
+      xhr :get, :index, company_id: company_id, township: "123"
+      assigns(:active_licenses).should == active_licenses_in_township
+    end
   end
 end