shopify_api/test/draft_order_test.rb

require 'test_helper'

class DraftOrderTest < Test::Unit::TestCase
  def setup
    super

    fake 'draft_orders/517119332', body: load_fixture('draft_order')

    @draft_order = ShopifyAPI::DraftOrder.find(517119332)
  end

  def test_get_draft_order
    fake 'draft_orders/517119332', method: :get, status: 200, body: load_fixture('draft_order')

    draft_order = ShopifyAPI::DraftOrder.find(517119332)

    assert_equal 517119332, draft_order.id
  end

  def test_get_all_draft_orders
    fake 'draft_orders', method: :get, status: 200, body: load_fixture('draft_orders')

    draft_orders = ShopifyAPI::DraftOrder.all

    assert_equal 1, draft_orders.length
    assert_equal 517119332, draft_orders.first.id
  end

  def test_get_count_draft_orders
    fake 'draft_orders/count', method: :get, status: 200, body: '{"count": 16}'

    draft_orders_count = ShopifyAPI::DraftOrder.count

    assert_equal 16, draft_orders_count
  end

  def test_create_draft_order
    fake 'draft_orders', method: :post, status: 201, body: load_fixture('draft_order')

    draft_order = ShopifyAPI::DraftOrder.create(line_items: [{ quantity: 1, variant_id: 39072856 }])

    assert_equal '{"draft_order":{"line_items":[{"quantity":1,"variant_id":39072856}]}}', WebMock.last_request.body
    assert_equal 39072856, draft_order.line_items.first.variant_id
  end

  def test_create_draft_order_202
    fake 'draft_orders', method: :post, status: 202, body: load_fixture('draft_order')

    draft_order = ShopifyAPI::DraftOrder.create(line_items: [{ quantity: 1, variant_id: 39072856 }])

    assert_equal 39072856, draft_order.line_items.first.variant_id
  end

  def test_update_draft_order
    draft_order_response = ActiveSupport::JSON.decode(load_fixture('draft_order'))
    draft_order_response['draft_order']['note'] = 'Test new note'
    @draft_order.note = 'Test new note'
    fake 'draft_orders/517119332', method: :put, status: 200, body: ActiveSupport::JSON.encode(draft_order_response)

    @draft_order.save

    assert_equal draft_order_response['draft_order']['note'], @draft_order.note
  end

  def test_send_invoice_with_no_params
    draft_order_invoice_fixture = load_fixture('draft_order_invoice')
    draft_order_invoice = ActiveSupport::JSON.decode(draft_order_invoice_fixture)
    fake 'draft_orders/517119332/send_invoice', method: :post, body: draft_order_invoice_fixture

    draft_order_invoice_response = @draft_order.send_invoice

    assert_equal '{"draft_order_invoice":{}}', WebMock.last_request.body
    assert_kind_of ShopifyAPI::DraftOrderInvoice, draft_order_invoice_response
    assert_equal draft_order_invoice['draft_order_invoice']['to'], draft_order_invoice_response.to
  end

  def test_send_invoice_with_params
    draft_order_invoice_fixture = load_fixture('draft_order_invoice')
    draft_order_invoice = ActiveSupport::JSON.decode(draft_order_invoice_fixture)
    fake 'draft_orders/517119332/send_invoice', method: :post, body: draft_order_invoice_fixture

    draft_order_invoice_response = @draft_order.send_invoice(ShopifyAPI::DraftOrderInvoice.new(draft_order_invoice['draft_order_invoice']))

    assert_equal draft_order_invoice, ActiveSupport::JSON.decode(WebMock.last_request.body)
    assert_kind_of ShopifyAPI::DraftOrderInvoice, draft_order_invoice_response
    assert_equal draft_order_invoice['draft_order_invoice']['to'], draft_order_invoice_response.to
  end

  def test_delete_draft_order
    fake 'draft_orders/517119332', method: :delete, body: 'destroyed'
    assert @draft_order.destroy
  end

  def test_add_metafields_to_draft_order
    fake 'draft_orders/517119332/metafields', method: :post, status: 201, body: load_fixture('metafield')

    field = @draft_order.add_metafield(ShopifyAPI::Metafield.new(namespace: 'contact', key: 'email', value: '123@example.com', value_type: 'string'))

    assert_equal ActiveSupport::JSON.decode('{"metafield":{"namespace":"contact","key":"email","value":"123@example.com","value_type":"string"}}'), ActiveSupport::JSON.decode(WebMock.last_request.body)
    assert !field.new_record?
    assert_equal 'contact', field.namespace
    assert_equal 'email', field.key
    assert_equal '123@example.com', field.value
  end

  def test_get_all_metafields_for_draft_order
    fake 'draft_orders/517119332/metafields', body: load_fixture('metafields')

    metafields = @draft_order.metafields

    assert_equal 3, metafields.length
    assert metafields.all? { |m| m.is_a?(ShopifyAPI::Metafield) }
  end

  def test_get_2_metafields_for_draft_order
    body = ActiveSupport::JSON.decode load_fixture 'metafields'
    body['metafields'].slice! 2, 1

    fake 'draft_orders/517119332/metafields.json?limit=2', body: body.to_json, extension: false

    metafields = @draft_order.metafields limit: 2

    assert_equal 2, metafields.length
    assert metafields.all?{ |m| m.is_a? ShopifyAPI::Metafield }
  end

  def test_complete_draft_order_with_no_params
    completed_fixture = load_fixture('draft_order_completed')
    completed_draft = ActiveSupport::JSON.decode(completed_fixture)['draft_order']
    fake 'draft_orders/517119332/complete', method: :put, status: 200, body: completed_fixture

    @draft_order.complete

    assert_equal completed_draft['status'], @draft_order.status
    assert_equal completed_draft['order_id'], @draft_order.order_id
    refute_nil @draft_order.completed_at
  end

  def test_complete_draft_order_with_params
    completed_fixture = load_fixture('draft_order_completed')
    completed_draft = ActiveSupport::JSON.decode(completed_fixture)['draft_order']
    complete_params = { payment_pending: true }
    fake 'draft_orders/517119332/complete.json?payment_pending=true', extension: false, method: :put, status: 200, body: completed_fixture

    @draft_order.complete(complete_params)

    assert_equal completed_draft['status'], @draft_order.status
    assert_equal completed_draft['order_id'], @draft_order.order_id
    refute_nil @draft_order.completed_at
  end
end