"""Vendor Seed Data. Seed data is in vendor_seed_data. Call get_max_vendor_id() to get the max(vendor_id) in the generated data. """ from tests.testutils import constants from tests.testutils.generators import VendorGenerator VENDORS_LIST = [ { 'owner': 'odd', 'name': 'Generic Test Label with status \'{status}\'', 'company': '', 'status': constants.TEST_VENDOR_STATUS_SIGNED, 'country': 1, 'is_owned': constants.TEST_VENDOR_NO, 'is_distributor': constants.TEST_VENDOR_NO_SHORT, 'email': 'test1@example.com' }, { 'owner': 'odd', 'name': 'Generic Test Label with status \'{status}\'', 'company': '', 'status': constants.TEST_VENDOR_STATUS_DELETED, 'country': 1, 'is_owned': constants.TEST_VENDOR_NO, 'is_distributor': constants.TEST_VENDOR_NO_SHORT, 'email': 'test2@example.com' }, { 'owner': 'odd', 'name': 'Generic Test Label with status \'{status}\'', 'company': '', 'status': constants.TEST_VENDOR_STATUS_INACTIVE, 'country': 1, 'is_owned': constants.TEST_VENDOR_YES, 'is_distributor': constants.TEST_VENDOR_YES_SHORT, 'email': 'test3@example.com' } ] def get_max_vendor_id(): """Get max vendor_id.""" return _generator.vendor_id _generator = VendorGenerator() _generator.add_vendors(VENDORS_LIST) vendor_seed_data = _generator.items