sched_api_spec.rb in td-client-0.8.63

- old
+ new
@@ -14,23 +14,56 @@
       {'cron' => cron, 'query' => query, 'database' => db_name}
     end
 
     it 'should create a new schedule' do
       start = Time.now
-      stub_api_request(:post, "/v3/schedule/create/#{e(sched_name)}").with(opts.merge('type' => 'hive'))
+      stub_api_request(:post, "/v3/schedule/create/#{e(sched_name)}")
+        .with(opts.merge('type' => 'hive'))
         .to_return(:body => {'name' => sched_name, 'start' => start.to_s}.to_json)
 
       api.create_schedule(sched_name, opts).should == start.to_s
     end
 
     it 'should return 422 error with invalid name' do
       name = '1'
       err_msg = "Validation failed: Name is too short" # " (minimum is 3 characters)"
-      stub_api_request(:post, "/v3/schedule/create/#{e(name)}").with(opts.merge('type' => 'hive'))
+      stub_api_request(:post, "/v3/schedule/create/#{e(name)}")
+        .with(opts.merge('type' => 'hive'))
         .to_return(:status => 422, :body => {'message' => err_msg}.to_json)
 
       expect {
         api.create_schedule(name, opts)
-      }.to raise_error(TreasureData::APIError, /^#{err_msg}/)
+      }.to raise_error(TreasureData::APIError, /#{err_msg}/)
+    end
+  end
+
+  describe 'update_schedule' do
+    let :pig_query do
+      "OUT = FOREACH (GROUP plt364 ALL) GENERATE COUNT(plt364);\n" * 200
+    end
+    let :opts do
+      {'cron' => cron, 'query' => pig_query, 'database' => db_name}
+    end
+
+    it 'should not return 414 even if the query text is very long' do
+      stub_api_request(:post, "/v3/schedule/update/#{e(sched_name)}")
+        .with(opts.merge('type' => 'pig'))
+        .to_return(:body => {'name' => sched_name, 'query' => pig_query}.to_json)
+
+      err_msg = "Update schedule failed: Request-URI Too Long"
+      expect {
+        api.update_schedule(sched_name, opts)
+      }.not_to raise_error
+    end
+
+    it 'should update the schedule with the new query' do
+      stub_api_request(:post, "/v3/schedule/update/#{e(sched_name)}")
+        .with(opts.merge('type' => 'pig'))
+        .to_return(:body => {'name' => sched_name, 'query' => pig_query}.to_json)
+
+      stub_api_request(:get, "/v3/schedule/list")
+        .to_return(:body => {'schedules' => [{'name' => sched_name, 'query' => pig_query}]}.to_json)
+
+      expect(api.list_schedules.first[2]).to eq(pig_query)
     end
   end
 end