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)
