test_odeo.py

import unittest

import transaction
from agratek.api.merchant.views.vendor import build_request
from opensipkd.base.models import Partner
from opensipkd.pasar.models import PartnerProduk, Produk
from pyramid.paster import get_appsettings
from pyramid_rpc.jsonrpc import JsonRpcError
from pyramid import testing


class TestOdeo(unittest.TestCase):
    def setUp(self):
        settings = get_appsettings('test.ini', name='main')
        self.config = testing.setUp(settings=settings)
        # self.config.include('..models')
        # self.config.include('..routes')

        #session_factory = self.config.registry['dbsession_factory']
        #self.session = get_tm_session(session_factory, transaction.manager)
        #
        # self.config = testing.setUp()
        # from pyramid.paster import get_appsettings
        from opensipkd.base import main
        app = main({}, **settings)

    def test_vendor(self):
        """
        Test that it can sum a list of integers
        """
        produk_kd = 'TS-5'
        bill_no = '081311045668'
        partner_kd = 'ODEO'
        vendor = Partner.query_kode(partner_kd).first()
        produk = Produk.query_kode(produk_kd).first()
        if not vendor or not produk:
            raise JsonRpcError(message="Vendor Not Found")
        values = dict(vendor_id = vendor.id,
                      produk_id = produk.id,
                      destination=bill_no)

        result = build_request('payment', values)
        for r in result:
            print(r,":", result[r])

if __name__ == '__main__':
    unittest.main()