def test_flatten_query_date_3(self): helper = helpers.Helper('test') source = dict( category=1, date=[504, 511, 512], price=[], quantity=[], status='', user_id=int(USER_ID), username=str(USERNAME), ) result = helper.flatten(source) expect = dict( category='原價轉讓', date='5.4(Fri), 5.11(Fri), 5.12(Sat)', price='', quantity='', status='', user_id=int(USER_ID), username=str(USERNAME), ) self.assertEqual( json.dumps(result, ensure_ascii=False, sort_keys=True), json.dumps(expect, ensure_ascii=False, sort_keys=True))
def test_init_query(self): query = Query(user_id=USER_ID, username=USERNAME).to_dict() helper = helpers.Helper('test') result = helper.flatten(query) expect = dict( category='', date='', price='', quantity='', status='', user_id=int(USER_ID), username=str(USERNAME), ) self.assertDictEqual(result, expect)
def test_generate_tickets_traits_gt_trait_len(self): helper = helpers.Helper('test') tickets = [] ticket = { "category": 2, "date": 511, "id": 20, "price": 2, "quantity": 2, "remarks": None, "row": "", "section": "", "status": 1, "update_at": "2018-03-28 11:47:16", "user_id": USER_ID, "username": USERNAME, "wish_date": "511,512,513", "wish_price": "4,2", "wish_quantity": "1,2" } for i in range(0, 6): tickets.append(ticket) expect_ticket = { 'category': '換飛', 'date': '5.11(Fri)', 'id': 20, 'price': '$880座位', 'quantity': '2', 'row': '', 'section': '', "remarks": '', 'status': '待交易', 'update_at': '2018-03-28 11:47:16', 'username': USERNAME, "wish_date": "5.11(Fri), 5.12(Sat), 5.13(Sun)", "wish_price": "$880座位, $680企位", "wish_quantity": "1, 2" } expect = [[ expect_ticket, expect_ticket, expect_ticket, expect_ticket, expect_ticket ], [expect_ticket]] result = helper.generate_tickets_traits(tickets) # self.assertCountEqual(expect, result) self.assertEqual(expect, result)
def test_init_ticket(self): ticket = Ticket(user_id=USER_ID, username=USERNAME).to_dict() helper = helpers.Helper('test') result = helper.flatten(ticket) expect = dict(category='', date='', price='', quantity='', section='', row='', seat='', status='待交易', remarks='', user_id=int(USER_ID), username=str(USERNAME), wish_date='', wish_price='', wish_quantity='') self.assertDictEqual(result, expect)
def test_generate_tickets_traits_ls_trait_len(self): helper = helpers.Helper('test') tickets = [ dict(category=2, date=511, id=20, price=2, quantity=2, remarks=None, row="", section="", status=1, update_at="2018-03-28 11:47:16", user_id=USER_ID, username=USERNAME, wish_date="511,512,513", wish_price="4,2", wish_quantity="1,2") ] expect = [[{ 'category': '換飛', 'date': '5.11(Fri)', 'id': 20, 'price': '$880座位', 'quantity': '2', 'row': '', 'section': '', "remarks": '', 'status': '待交易', 'update_at': '2018-03-28 11:47:16', 'username': USERNAME, "wish_date": "5.11(Fri), 5.12(Sat), 5.13(Sun)", "wish_price": "$880座位, $680企位", "wish_quantity": "1, 2" }]] result = helper.generate_tickets_traits(tickets) self.assertEqual(expect, result)
def test_error_case(self): ticket = { "category": 2, "date": 504, "price": 4, "quantity": 2, "section": "F2", "row": "p", "seat": "", "status": 1, "remarks": "", "wish_date": [506], "wish_price": [3], "wish_quantity": [], "user_id": USER_ID, "username": USERNAME } expected = { 'category': '換飛', 'date': '5.4(Fri)', 'price': '$680企位', 'quantity': '2', 'section': 'F2', 'row': 'p', 'seat': '', 'status': '待交易', 'remarks': '', 'wish_date': '5.6(Sun)', 'wish_price': '$680座位', 'wish_quantity': '', 'user_id': USER_ID, 'username': USERNAME } helper = helpers.Helper('test') result = helper.flatten(ticket) self.assertDictEqual(expected, result)