"""Test delivery_history logic.""" from deliveryhistory.logic import delivery_history def test_get_record(): """Test get_record method.""" item = { 'upc': '345234523340', 'encoder_id': 23, 'date_delivered': '2020-11-23', 'store_id': 1, 'package_size': 345, } expected_record = { 'delivery_id': None, 'upc': '345234523340', 'encoder_id': 23, 'date_delivered': '2020-11-23', 'customer_master_master_id': 1, 'package_size': 345, } record = delivery_history.get_record(item) assert record.to_dict() == expected_record