Subversion Repositories SmartDukaan

from shop2020.thriftpy.model.v1.catalog.ttypes import Item, status, Warehouse,\

    VendorItemPricing

import datetime

import os

from time import strftime

from shop2020.clients.CatalogClient import CatalogClient

from shop2020.utils.Utils import to_java_date, to_py_date

from shop2020.model.v1.catalog.impl.DataAcessors import validate_item_prices, validate_vendor_prices

inventory_client = CatalogClient()

client = inventory_client.get_client()

"""

#deals = client.getBestDeals()

#print deals

warehouse_id, items_in_inventory = client.getItemAvailabilityAtLocation(0, 132)

print warehouse_id

print items_in_inventory

"""

def test_validate():

    # Item prices should follow the following validations.

    # mrp > sellingPrice

    # mrp > mop

    # xferPrice < mop

    item = Item()

    item.id = 1

    item.productGroup = "Handsets"

    item.modelNumber = "MI-310"

    item.brand = "Spice"

    item.color = "Red"

    vip = VendorItemPricing()

    vip.vendorId = 1

    set_prices1(item, vip)

    set_prices2(item, vip)

    return

def set_prices1(item, vip):

    #Test Case 1 (mrp < sellingPrice)

    item.mrp = 220.50

    item.sellingPrice = 225.50

    vip.mop = 111

    vip.dealerPrice = 111

    vip.transferPrice = 111

    validate(item,vip)

    return

def set_prices2(item, vip):

    #Test Case 2 (mrp < mop)

    item.mrp = 230.50

    item.sellingPrice = 225.50

    vip.mop = 235.0

    vip.dealerPrice = 210.0

    vip.transferPrice = 200.5

    validate(item,vip)

    return

def validate(item, vip):

#    try:

#        #validate_prices(item, vip)

#    except Exception as ex:

#        print ex

#        return

#    print 'prices validated'

    return

def test_similar_item():

    print (client.checkSimilarItem('handset', 'spice', 'bldk', None))

    print (client.checkSimilarItem('handset', 'spice', 'bldk', ''))

    print (client.checkSimilarItem('handset', 'spice', 'bldk', 'Black'))

def test_change_item_status():

    #changeItemStatus method testing

    client.changeItemStatus(1, time, status.PAUSED)

    item = client.getItem(1)

    print 'item status = ' + str(item.itemStatus) + ", desc = " + item.status_description

    client.changeItemStatus(1, time, status.PHASED_OUT)

    item = client.getItem(1)

    print 'item status = ' + str(item.itemStatus) + ", desc = " + item.status_description

def test_check_risky_item():

    #mark_risky_item method testing. It is called from the following 3 methods.

    availability = {"battery|nokia|bl 4b|na":1}

    client.updateInventory(2, time, availability)

    #client.reserveItemInWarehouse(167, 1, 1)

    #client.reduceReservationCount(167, 2, 1)

def test_update_similarity_index():

    items=client.getSimilarItemsCatalogIds(3, 2, 1441)

    print items

def test_logging(item_id, risky):

    logfile = open(os.getenv("HOME")+'/'+'riskyNew.log', 'a')

    logfile.write(str(strftime("%Y-%m-%d %H:%M:%S")) + ', ItemId=' + str(item_id) + ', ' + str(risky) + '\n')

def test_get_best_deal_catalog_ids():

    print client.getBestDealsCatalogIds(0, 20, 'Nokia', -1);

def test_get_all_items():

    items = client.getAllItemsByStatus(status.ACTIVE)

    for item in items:

        print item.itemInventory

def test_is_alive():

    print client.isAlive()

def test_get_availability_at_location():

    print client.getItemAvailabilityAtLocation(0, 1)

def test_get_pending_orders_inventory():

    print client.getPendingOrdersInventory()

def testGetVendorsForWarehouse(warehouseId):

    vendors = client.getVendorsForWarehouse(warehouseId)

    for vendor in vendors:

        print vendor

def testGetWarehousesForVendor(vendorId):

    warehouses =  client.getWarehousesForVendor(vendorId)

    for warehouse in warehouses:

        print warehouse

def testVendorItemPricing(itemId, vendorId):

    print client.getItemPricing(itemId, vendorId)

def testArgs(string, floatvar):

    print string

    print floatvar

if __name__ == '__main__':

    time = to_py_date(datetime.datetime.now())

    #test_validate()

    #test_similar_item()

    #test_change_item_status()

    #test_check_risky_item()

    #test_logging(5, True)

    #test_update_similarity_index()

    #test_get_best_deal_catalog_ids()

    #test_get_all_items()

    #test_is_alive()

    #test_get_availability_at_location()

    #test_get_pending_orders_inventory()

    print client.getItemAvailabilityAtLocation(0, 9999)

    print "hello"

    #testGetWarehousesForVendor(1)

    #testGetVendorsForWarehouse(7)

    #testVendorItemPricing(5,-1)

    #string = "hello"

    #floatvar = 5.5

    #testArgs(string, floatvar)

    #testArgs(string, None)