td-client 1.0.0-java

data/spec/td/client/api_ssl_connection_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+require 'td/client/api'
+require 'logger'
+require 'webrick'
+require 'webrick/https'
+
+# Workaround for https://github.com/jruby/jruby-openssl/issues/78
+# With recent JRuby + jruby-openssl, X509CRL#extentions_to_text causes
+# StringIndexOOBException when we try to dump SSL Server Certificate.
+# when one of extensions has "" as value.
+# This hack is from httpclient https://github.com/nahi/httpclient/blob/master/lib/httpclient/util.rb#L27-L46
+if defined? JRUBY_VERSION
+  require 'openssl'
+  require 'java'
+  module OpenSSL
+    module X509
+      class Certificate
+        java_import 'java.security.cert.Certificate'
+        java_import 'java.security.cert.CertificateFactory'
+        java_import 'java.io.ByteArrayInputStream'
+        def to_text
+          cf = CertificateFactory.getInstance('X.509')
+          cf.generateCertificate(ByteArrayInputStream.new(self.to_der.to_java_bytes)).toString
+        end
+      end
+    end
+  end
+end
+
+describe 'API SSL connection' do
+  DIR = File.dirname(File.expand_path(__FILE__))
+
+  after :each do
+    @server.shutdown if @server
+  end
+
+  it 'should fail to connect SSLv3 only server' do
+    @server = setup_server(:SSLv3)
+    api = API.new(nil, :endpoint => "https://localhost:#{@serverport}", :retry_post_requests => false)
+    api.ssl_ca_file = File.join(DIR, 'ca-all.cert')
+    expect {
+      api.delete_database('no_such_database')
+    }.to raise_error OpenSSL::SSL::SSLError
+  end
+
+  # 1.8.7's net/http does not call SSLContext#set_params. Use 1.8.7 where you don't care security.
+  if RUBY_VERSION >= '1.9.0'
+    it 'should success to connect TLSv1 only server' do
+      @server = setup_server(:TLSv1)
+      api = API.new(nil, :endpoint => "https://localhost:#{@serverport}", :retry_post_requests => false)
+      api.ssl_ca_file = File.join(DIR, 'ca-all.cert')
+      expect {
+        api.delete_database('no_such_database')
+      }.to raise_error TreasureData::NotFoundError
+    end
+  end
+
+  def setup_server(ssl_version)
+    logger = Logger.new(STDERR)
+    logger.level = Logger::Severity::FATAL  # avoid logging SSLError (ERROR level)
+    @server = WEBrick::HTTPServer.new(
+      :BindAddress => "localhost",
+      :Logger => logger,
+      :Port => 0,
+      :AccessLog => [],
+      :DocumentRoot => '.',
+      :SSLEnable => true,
+      :SSLCACertificateFile => File.join(DIR, 'ca.cert'),
+      :SSLCertificate => cert('server.cert'),
+      :SSLPrivateKey => key('server.key')
+    )
+    @serverport = @server.config[:Port]
+    @server.mount(
+      '/hello',
+      WEBrick::HTTPServlet::ProcHandler.new(method(:do_hello).to_proc)
+    )
+    @server.ssl_context.ssl_version = ssl_version
+    @server_thread = start_server_thread(@server)
+    return @server
+  end
+
+  def do_hello(req, res)
+    res['content-type'] = 'text/html'
+    res.body = "hello"
+  end
+
+  def start_server_thread(server)
+    t = Thread.new {
+      Thread.current.abort_on_exception = true
+      server.start
+    }
+    while server.status != :Running
+      sleep 0.1
+      unless t.alive?
+        t.join
+        raise
+      end
+    end
+    t
+  end
+
+  def cert(filename)
+    OpenSSL::X509::Certificate.new(File.read(File.join(DIR, filename)))
+  end
+
+  def key(filename)
+    OpenSSL::PKey::RSA.new(File.read(File.join(DIR, filename)))
+  end
+end

data/spec/td/client/bulk_import_spec.rb

@@ -0,0 +1,199 @@
+# encoding: utf-8
+
+require 'spec_helper'
+require 'td/client/spec_resources'
+require 'tempfile'
+
+describe 'BulkImport API' do
+  include_context 'spec symbols'
+  include_context 'common helper'
+
+  let :api do
+    API.new(nil)
+  end
+
+  let :packed do
+    s = StringIO.new
+    Zlib::GzipWriter.wrap(s) do |f|
+      pk = MessagePack::Packer.new(f)
+      pk.write([1, '2', 3.0])
+      pk.write([4, '5', 6.0])
+      pk.write([7, '8', 9.0])
+      pk.flush
+    end
+    s.string
+  end
+
+  let(:endpoint_domain) { TreasureData::API::DEFAULT_IMPORT_ENDPOINT }
+
+  describe 'create_bulk_import' do
+    it 'should create a new bulk_import' do
+      stub_api_request(:post, "/v3/bulk_import/create/#{e(bi_name)}/#{e(db_name)}/#{e(table_name)}").
+        to_return(:body => {'bulk_import' => bi_name}.to_json)
+
+      expect(api.create_bulk_import(bi_name, db_name, table_name)).to be_nil
+    end
+
+    it 'should return 422 error with invalid name' do
+      name = 'D'
+      err_msg = "Validation failed: Name is too short" # " (minimum is 3 characters)"
+      stub_api_request(:post, "/v3/bulk_import/create/#{e(name)}/#{e(db_name)}/#{e(table_name)}").
+        to_return(:status => 422, :body => {'message' => err_msg}.to_json)
+
+      expect {
+        api.create_bulk_import(name, db_name, table_name)
+      }.to raise_error(TreasureData::APIError, /#{err_msg}/)
+    end
+
+    it 'should return 404 error with non exist database name' do
+      db = 'no_such_db'
+      err_msg = "Couldn't find UserDatabase with name = #{db}"
+      stub_api_request(:post, "/v3/bulk_import/create/#{e(bi_name)}/#{e(db)}/#{e(table_name)}").
+        to_return(:status => 404, :body => {'message' => err_msg}.to_json)
+
+      expect {
+        api.create_bulk_import(bi_name, db, table_name)
+      }.to raise_error(TreasureData::APIError, /#{err_msg}/)
+    end
+
+    it 'should return 404 error with non exist table name' do
+      table = 'no_such_table'
+      err_msg = "Couldn't find UserTableReference with name = #{table}"
+      stub_api_request(:post, "/v3/bulk_import/create/#{e(bi_name)}/#{e(db_name)}/#{e(table)}").
+        to_return(:status => 404, :body => {'message' => err_msg}.to_json)
+
+      expect {
+        api.create_bulk_import(bi_name, db_name, table)
+      }.to raise_error(TreasureData::APIError, /#{err_msg}/)
+    end
+  end
+
+  describe 'delete_bulk_import' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/delete/name').
+        with(:body => 'foo=bar')
+      expect(api.delete_bulk_import('name', 'foo' => 'bar')).to eq(nil)
+    end
+  end
+
+  describe 'show_bulk_import' do
+    it 'runs' do
+      stub_api_request(:get, '/v3/bulk_import/show/name').
+        to_return(:body => {'status' => 'status', 'other' => 'other'}.to_json)
+      expect(api.show_bulk_import('name')['status']).to eq('status')
+    end
+  end
+
+  describe 'list_bulk_imports' do
+    it 'runs' do
+      stub_api_request(:get, '/v3/bulk_import/list').
+        with(:query => 'foo=bar').
+        to_return(:body => {'bulk_imports' => %w(1 2 3)}.to_json)
+      expect(api.list_bulk_imports('foo' => 'bar')).to eq(%w(1 2 3))
+    end
+  end
+
+  describe 'list_bulk_import_parts' do
+    it 'runs' do
+      stub_api_request(:get, '/v3/bulk_import/list_parts/name').
+        with(:query => 'foo=bar').
+        to_return(:body => {'parts' => %w(1 2 3)}.to_json)
+      expect(api.list_bulk_import_parts('name', 'foo' => 'bar')).to eq(%w(1 2 3))
+    end
+  end
+
+  describe 'bulk_import_upload_part' do
+    it 'runs' do
+      t = Tempfile.new('bulk_import_spec')
+      File.open(t.path, 'w') do |f|
+        f << '12345'
+      end
+      stub_request(:put, "https://#{TreasureData::API::DEFAULT_ENDPOINT}/v3/bulk_import/upload_part/name/part").
+        with(:body => '12345')
+      File.open(t.path) do |f|
+        expect(api.bulk_import_upload_part('name', 'part', f, 5)).to eq(nil)
+      end
+    end
+
+    if ''.respond_to?(:encode)
+      it 'encodes part_name in UTF-8' do
+        t = Tempfile.new('bulk_import_spec')
+        File.open(t.path, 'w') do |f|
+          f << '12345'
+        end
+        stub_request(:put, "https://#{TreasureData::API::DEFAULT_ENDPOINT}/v3/bulk_import/upload_part/name/" + CGI.escape('日本語(Japanese)'.encode('UTF-8'))).
+          with(:body => '12345')
+        File.open(t.path) do |f|
+          expect(api.bulk_import_upload_part('name', '日本語(Japanese)'.encode('Windows-31J'), f, 5)).to eq(nil)
+        end
+      end
+    end
+  end
+
+  describe 'bulk_import_delete_part' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/delete_part/name/part')
+      expect(api.bulk_import_delete_part('name', 'part')).to eq(nil)
+    end
+  end
+
+  describe 'freeze_bulk_import' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/freeze/name')
+      expect(api.freeze_bulk_import('name')).to eq(nil)
+    end
+  end
+
+  describe 'unfreeze_bulk_import' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/unfreeze/name')
+      expect(api.unfreeze_bulk_import('name')).to eq(nil)
+    end
+  end
+
+  describe 'perform_bulk_import' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/perform/name').
+        to_return(:body => {'job_id' => 12345}.to_json)
+      expect(api.perform_bulk_import('name')).to eq('12345')
+    end
+  end
+
+  describe 'commit_bulk_import' do
+    it 'runs' do
+      stub_api_request(:post, '/v3/bulk_import/commit/name').
+        to_return(:body => {'job_id' => 12345}.to_json)
+      expect(api.commit_bulk_import('name')).to eq(nil)
+    end
+  end
+
+  describe 'bulk_import_error_records' do
+    it 'returns [] on empty' do
+      stub_api_request(:get, '/v3/bulk_import/error_records/name').
+        to_return(:body => '')
+      expect(api.bulk_import_error_records('name')).to eq([])
+    end
+
+    it 'returns nil on empty if block given' do
+      stub_api_request(:get, '/v3/bulk_import/error_records/name').
+        to_return(:body => '')
+      expect(api.bulk_import_error_records('name'){}).to eq(nil)
+    end
+
+    it 'returns unpacked result' do
+      stub_api_request(:get, '/v3/bulk_import/error_records/name').
+        to_return(:body => packed)
+      expect(api.bulk_import_error_records('name')).to eq([[1, '2', 3.0], [4, '5', 6.0], [7, '8', 9.0]])
+    end
+
+    it 'yields unpacked result if block given' do
+      stub_api_request(:get, '/v3/bulk_import/error_records/name').
+        to_return(:body => packed)
+      result = []
+      api.bulk_import_error_records('name') do |row|
+        result << row
+      end
+      expect(result).to eq([[1, '2', 3.0], [4, '5', 6.0], [7, '8', 9.0]])
+    end
+  end
+end

data/spec/td/client/bulk_load_spec.rb

@@ -0,0 +1,401 @@
+require 'spec_helper'
+require 'td/client/spec_resources'
+require 'tempfile'
+
+describe 'BulkImport API' do
+  include_context 'spec symbols'
+  include_context 'common helper'
+
+  let :api do
+    Client.new(nil, {:max_cumul_retry_delay => -1})
+  end
+
+  let :retry_api do
+    API.new(nil, {:retry_delay => 1, :max_cumul_retry_delay => 1})
+  end
+
+  let :original_config do
+    {
+      :config => {
+        :type => "s3_file",
+        :access_key_id => "key",
+        :secret_access_key => "secret",
+        :endpoint => "s3.amazonaws.com",
+        :bucket => "td-bulk-loader-test-tokyo",
+        :path_prefix => "in/nahi/sample"
+      }
+    }
+  end
+
+  let :guessed_config do
+    {
+      "config" => {
+        "type" => "s3_file",
+        "access_key_id" => "key",
+        "secret_access_key" => "secret",
+        "endpoint" => "s3.amazonaws.com",
+        "bucket" => "td-bulk-loader-test-tokyo",
+        "path_prefix" => "in/nahi/sample",
+        "parser" => {
+          "charset" => "UTF-8",
+          "newline" => "LF",
+          "type" => "csv",
+          "delimiter" => ",",
+          "header_line" => false,
+          "columns" => [
+            {"name" => "time", "type" => "long"},
+            {"name" => "c1", "type" => "long"},
+            {"name" => "c2", "type" => "string"},
+            {"name" => "c3", "type" => "string"},
+          ]
+        },
+        "decoders" => [
+          {"type" => "gzip"}
+        ]
+      }
+    }
+  end
+
+  let :preview_result do
+    {
+      "schema" => [
+        {"index" => 0, "name" => "c0", "type" => "string"},
+        {"index" => 1, "name" => "c1", "type" => "long"},
+        {"index" => 2, "name" => "c2", "type" => "string"},
+        {"index" => 3, "name" => "c3", "type" => "string"}
+      ],
+      "records" => [
+        ["19920116", 32864, "06612", "00195"],
+        ["19910729", 14824, "07706", "00058"],
+        ["19950708", 27559, "03244", "00034"],
+        ["19931010", 11270, "03459", "00159"],
+        ["19981117", 20461, "01409", "00128"],
+        ["19981117", 20461, "00203", "00128"],
+        ["19930108", 44402, "01489", "00001"],
+        ["19960104", 16528, "04848", "00184"],
+        ["19960104", 16528, "01766", "00184"],
+        ["19881022", 26114, "06960", "00175"]
+      ]
+    }
+  end
+
+  let :bulk_load_session do
+    guessed_config.dup.merge(
+      {
+        "name" => "nahi_test_1",
+        "cron" => "@daily",
+        "timezone" => "Asia/Tokyo",
+        "delay" => 3600
+      }
+    )
+  end
+
+  let :bulk_load_job do
+    guessed_config.dup.merge(
+      {
+        "job_id" => 123456,
+        "account_id" => 1,
+        "status" => "success",
+        "records" => 10,
+        "schema" => [["c0", "string", ""], ["c1", "long", ""], ["c2", "string", ""], ["c3", "string", ""]],
+        "database" => {"id" => 189263, "name" => "nahidb"},
+        "table" => {"id" => 176281, "name" => "bulkload_import_test"},
+        "created_at" => 1426738133,
+        "updated_at" => 1426738145,
+        "start_at" => 1426738134,
+        "end_at" => 1426738144
+      }
+    )
+  end
+
+  describe 'guess' do
+    it 'returns guessed json' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:body => guessed_config.to_json)
+      expect(api.bulk_load_guess(
+        original_config
+      )).to eq(guessed_config)
+    end
+
+    it 'raises an error' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:status => 500, :body => guessed_config.to_json)
+      expect {
+        api.bulk_load_guess(
+          original_config
+        )
+      }.to raise_error(TreasureData::APIError)
+    end
+
+    it 'perform redo on 500 error' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:status => 500, :body => guessed_config.to_json)
+      begin
+        expect(retry_api.bulk_load_guess(
+          original_config
+        )).to != nil
+      rescue TreasureData::APIError => e
+        expect(e.message).to match(/^500: BulkLoad configuration guess failed/)
+      end
+    end
+
+    it 'perform retries on connection failure' do
+      api = retry_api
+      allow(api.instance_eval { @api }).to receive(:post).and_raise(SocketError.new('>>'))
+      begin
+        retry_api.bulk_load_guess(
+          original_config
+        )
+      rescue SocketError => e
+        expect(e.message).to eq('>> (Retried 1 times in 1 seconds)')
+      end
+    end
+  end
+
+  describe 'guess with old format' do
+    it 'returns guessed json' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:body => guessed_config.to_json)
+      expect(api.bulk_load_guess(
+        original_config
+      )).to eq(guessed_config)
+    end
+
+    it 'raises an error' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:status => 500, :body => guessed_config.to_json)
+      expect {
+        api.bulk_load_guess(
+          original_config
+        )
+      }.to raise_error(TreasureData::APIError)
+    end
+
+    it 'perform redo on 500 error' do
+      stub_api_request(:post, '/v3/bulk_loads/guess').
+        with(:body => original_config.to_json).
+        to_return(:status => 500, :body => guessed_config.to_json)
+      begin
+        expect(retry_api.bulk_load_guess(
+          original_config
+        )).to != nil
+      rescue TreasureData::APIError => e
+        expect(e.message).to match(/^500: BulkLoad configuration guess failed/)
+      end
+    end
+
+    it 'perform retries on connection failure' do
+      api = retry_api
+      allow(api.instance_eval { @api }).to receive(:post).and_raise(SocketError.new('>>'))
+      begin
+        retry_api.bulk_load_guess(
+          original_config
+        )
+      rescue SocketError => e
+        expect(e.message).to eq('>> (Retried 1 times in 1 seconds)')
+      end
+    end
+  end
+
+  describe 'preview' do
+    it 'returns preview json' do
+      stub_api_request(:post, '/v3/bulk_loads/preview').
+        with(:body => guessed_config.to_json).
+        to_return(:body => preview_result.to_json)
+      expect(api.bulk_load_preview(
+        guessed_config
+      )).to eq(preview_result)
+    end
+
+    it 'raises an error' do
+      stub_api_request(:post, '/v3/bulk_loads/preview').
+        with(:body => guessed_config.to_json).
+        to_return(:status => 500, :body => preview_result.to_json)
+      expect {
+        api.bulk_load_preview(
+          guessed_config
+        )
+      }.to raise_error(TreasureData::APIError)
+    end
+  end
+
+  describe 'issue' do
+    it 'returns job id' do
+      expected_request = guessed_config.dup
+      expected_request['database'] = 'database'
+      expected_request['table'] = 'table'
+      stub_api_request(:post, '/v3/job/issue/bulkload/database').
+        with(:body => expected_request.to_json).
+        to_return(:body => {'job_id' => 12345}.to_json)
+      expect(api.bulk_load_issue(
+        'database',
+        'table',
+        guessed_config
+      )).to eq('12345')
+    end
+  end
+
+  describe 'list' do
+    it 'returns BulkLoadSession' do
+      stub_api_request(:get, '/v3/bulk_loads').
+        to_return(:body => [bulk_load_session, bulk_load_session].to_json)
+      result = api.bulk_load_list
+      expect(result.size).to eq(2)
+      expect(result.first).to eq(bulk_load_session)
+    end
+
+    it 'returns empty' do
+      stub_api_request(:get, '/v3/bulk_loads').
+        to_return(:body => [].to_json)
+      expect(api.bulk_load_list.size).to eq(0)
+    end
+  end
+
+  describe 'create' do
+    it 'returns registered bulk_load_session' do
+      expected_request = guessed_config.dup
+      expected_request['name'] = 'nahi_test_1'
+      expected_request['cron'] = '@daily'
+      expected_request['timezone'] = 'Asia/Tokyo'
+      expected_request['delay'] = 3600
+      expected_request['database'] = 'database'
+      expected_request['table'] = 'table'
+      stub_api_request(:post, '/v3/bulk_loads').
+        with(:body => expected_request.to_json).
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_create(
+        'nahi_test_1',
+        'database',
+        'table',
+        guessed_config,
+        {
+          cron: '@daily',
+          timezone: 'Asia/Tokyo',
+          delay: 3600
+        }
+      )).to eq(bulk_load_session)
+    end
+
+    it 'accepts empty option' do
+      expected_request = guessed_config.dup
+      expected_request['name'] = 'nahi_test_1'
+      expected_request['database'] = 'database'
+      expected_request['table'] = 'table'
+      stub_api_request(:post, '/v3/bulk_loads').
+        with(:body => expected_request.to_json).
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_create(
+        'nahi_test_1',
+        'database',
+        'table',
+        guessed_config
+      )).to eq(bulk_load_session)
+    end
+
+    it 'accepts time_column option' do
+      expected_request = guessed_config.dup
+      expected_request['name'] = 'nahi_test_1'
+      expected_request['time_column'] = 'c0'
+      expected_request['database'] = 'database'
+      expected_request['table'] = 'table'
+      stub_api_request(:post, '/v3/bulk_loads').
+        with(:body => expected_request.to_json).
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_create(
+        'nahi_test_1',
+        'database',
+        'table',
+        guessed_config,
+        {
+          time_column: 'c0'
+        }
+      )).to eq(bulk_load_session)
+    end
+  end
+
+  describe 'show' do
+    it 'returns bulk_load_session' do
+      stub_api_request(:get, '/v3/bulk_loads/nahi_test_1').
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_show('nahi_test_1')).to eq(bulk_load_session)
+    end
+  end
+
+  describe 'update' do
+    it 'returns updated bulk_load_session' do
+      stub_api_request(:put, '/v3/bulk_loads/nahi_test_1').
+        with(:body => bulk_load_session.to_json).
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_update(
+        'nahi_test_1',
+        bulk_load_session
+      )).to eq(bulk_load_session)
+    end
+
+    it 'returns updated bulk_load_session' do
+      updated_bulk_load_session = bulk_load_session.merge({'timezone' => 'America/Los Angeles'})
+      stub_api_request(:put, '/v3/bulk_loads/nahi_test_1').
+        with(:body => updated_bulk_load_session.to_json).
+        to_return(:body => updated_bulk_load_session.to_json)
+      expect(api.bulk_load_update(
+        'nahi_test_1',
+        updated_bulk_load_session
+      )).to eq updated_bulk_load_session
+    end
+
+    it 'can remove the cron schedule ' do
+      updated_bulk_load_session = bulk_load_session.merge({'cron' => ''})
+      # NOTE: currently the API just ignores an empty 'cron' specification update
+      # I am assuming that once fixed, the API will return a nil for cron if unscheduled.
+      expected_bulk_load_session = (bulk_load_session['cron'] = nil)
+      stub_api_request(:put, '/v3/bulk_loads/nahi_test_1').
+        with(:body => updated_bulk_load_session.to_json).
+        to_return(:body => expected_bulk_load_session.to_json)
+      expect(api.bulk_load_update(
+        'nahi_test_1',
+        updated_bulk_load_session
+      )).to eq expected_bulk_load_session
+    end
+  end
+
+  describe 'delete' do
+    it 'returns updated bulk_load_session' do
+      stub_api_request(:delete, '/v3/bulk_loads/nahi_test_1').
+        to_return(:body => bulk_load_session.to_json)
+      expect(api.bulk_load_delete('nahi_test_1')).to eq(bulk_load_session)
+    end
+  end
+
+  describe 'history' do
+    it 'returns list of jobs' do
+      stub_api_request(:get, '/v3/bulk_loads/nahi_test_1/jobs').
+        to_return(:body => [bulk_load_job, bulk_load_job].to_json)
+      result = api.bulk_load_history('nahi_test_1')
+      expect(result.size).to eq(2)
+      expect(result.first).to eq(bulk_load_job)
+    end
+  end
+
+  describe 'run' do
+    it 'returns job_id' do
+      stub_api_request(:post, '/v3/bulk_loads/nahi_test_1/jobs').
+        with(:body => '{}').
+        to_return(:body => {'job_id' => 12345}.to_json)
+      expect(api.bulk_load_run('nahi_test_1')).to eq('12345')
+    end
+
+    it 'accepts scheduled_time' do
+      now = Time.now.to_i
+      stub_api_request(:post, '/v3/bulk_loads/nahi_test_1/jobs').
+        with(:body => {scheduled_time: now.to_s}.to_json).
+        to_return(:body => {'job_id' => 12345}.to_json)
+      expect(api.bulk_load_run('nahi_test_1', now)).to eq('12345')
+    end
+  end
+
+end