brotherton-erpnext/erpnext/regional/united_states/test_united_states.py

# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
# License: GNU General Public License v3. See license.txt

import unittest

import frappe

from erpnext.regional.report.irs_1099.irs_1099 import execute as execute_1099_report


class TestUnitedStates(unittest.TestCase):
    def test_irs_1099_custom_field(self):

        if not frappe.db.exists("Supplier", "_US 1099 Test Supplier"):
            doc = frappe.new_doc("Supplier")
            doc.supplier_name = "_US 1099 Test Supplier"
            doc.supplier_group = "Services"
            doc.supplier_type = "Company"
            doc.country = "United States"
            doc.tax_id = "04-1234567"
            doc.irs_1099 = 1
            doc.save()
            frappe.db.commit()
            supplier = frappe.get_doc('Supplier', "_US 1099 Test Supplier")
            self.assertEqual(supplier.irs_1099, 1)

    def test_irs_1099_report(self):
        make_payment_entry_to_irs_1099_supplier()
        filters = frappe._dict({"fiscal_year": "_Test Fiscal Year 2016", "company": "_Test Company 1"})
        columns, data = execute_1099_report(filters)
        expected_row = {'supplier': '_US 1099 Test Supplier',
                        'supplier_group': 'Services',
                        'payments': 100.0,
                        'tax_id': '04-1234567'}
        self.assertEqual(data[0], expected_row)


def make_payment_entry_to_irs_1099_supplier():

    frappe.db.sql("delete from `tabGL Entry` where party='_US 1099 Test Supplier'")
    frappe.db.sql("delete from `tabGL Entry` where against='_US 1099 Test Supplier'")
    frappe.db.sql("delete from `tabPayment Entry` where party='_US 1099 Test Supplier'")

    pe = frappe.new_doc("Payment Entry")
    pe.payment_type = "Pay"
    pe.company = "_Test Company 1"
    pe.posting_date = "2016-01-10"
    pe.paid_from = "_Test Bank USD - _TC1"
    pe.paid_to = "_Test Payable USD - _TC1"
    pe.paid_amount = 100
    pe.received_amount = 100
    pe.reference_no = "For IRS 1099 testing"
    pe.reference_date = "2016-01-10"
    pe.party_type = "Supplier"
    pe.party = "_US 1099 Test Supplier"
    pe.insert()
    pe.submit()