Raise UnlicensedFeatureError when we receive a HTTP 402

lib/ansible_tower_client/base_models/job_template.rb

@@ -11,6 +11,7 @@ def survey_spec
       spec_url = related['survey_spec']
       return nil unless spec_url
       api.get(spec_url).body
+    rescue AnsibleTowerClient::UnlicensedFeatureError
     end
 
     def survey_spec_hash

lib/ansible_tower_client/exception.rb

@@ -1,6 +1,7 @@
 module AnsibleTowerClient
   class Error < Exception; end
+  class ClientError < Error; end
   class ConnectionError < Error; end
   class NoMethodError < Error; end
-  class ClientError < Error; end
+  class UnlicensedFeatureError < Error; end
 end

lib/ansible_tower_client/middleware/raise_tower_error.rb

@@ -5,6 +5,8 @@ class RaiseTowerError < Faraday::Response::Middleware
 
       def on_complete(env)
         case env[:status]
+        when 402
+          raise AnsibleTowerClient::UnlicensedFeatureError
         when 404
           raise Faraday::Error::ResourceNotFound, response_values(env)
         when 407

spec/middleware/raise_tower_error_spec.rb

@@ -5,15 +5,20 @@
 describe AnsibleTowerClient::Middleware::RaiseTowerError do
   context "Faraday::Error" do
     let(:env_400) { MockEnv.new('missing these attributes', 400) }
+    let(:env_402) { MockEnv.new('missing these attributes', 402) }
     let(:env_404) { MockEnv.new('missing these attributes', 404) }
     let(:env_407) { MockEnv.new('missing these attributes', 407) }
 
     let(:error) { AnsibleTowerClient::Middleware::RaiseTowerError.new }
 
-    it "raises ClientError and returns the body message with a status 404" do
+    it "raises ClientError and returns the body message with a status 400" do
       expect { error.on_complete(env_400) }.to raise_error(AnsibleTowerClient::ClientError, "missing these attributes")
     end
 
+    it "raises UnlicensedFeatureError with a status 402" do
+      expect { error.on_complete(env_402) }.to raise_error(AnsibleTowerClient::UnlicensedFeatureError)
+    end
+
     it "raises ResourceNotFound exception with a status 404" do
       expect { error.on_complete(env_404) }.to raise_error(Faraday::Error::ResourceNotFound)
     end

spec/support/shared_examples/job_template.rb

@@ -25,6 +25,13 @@
       expect(described_class.new(api, raw_instance_no_survey).survey_spec).to be_nil
     end
   end
+
+  describe "unlicensed" do
+    it "returns a survey spec" do
+      expect(api).to receive(:get).and_raise(AnsibleTowerClient::UnlicensedFeatureError)
+      expect(described_class.new(api, raw_instance).survey_spec).to be_nil
+    end
+  end
 end
 
 shared_examples_for "JobTemplate#survey_spec_hash" do