def test_get_all(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
states = State.get_all_states(cur)
assert states is not None # Ensure something was returned
assert len(states) > 1 # Ensure multiple types were returned
conn.commit()
pass
def test_get(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
state = State.get_state(cur, 0)
assert state is None # Ensure none was returned for bad ID
state = State.get_state(cur, 1)
assert state is not None # Ensure something was returned for valid ID
assert state.id == 1 # Ensure correct type was returned
# Ensure we can commit without violating constraints
conn.commit()
pass
def test_crud(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
# Get nonexistent
user = User.get_user(cur, 0)
assert user is None # Get where user doesn't exist returns None
# Create user
user_id = User.create_user(
cur, User(user_id=None, username='******', pin='1234', user_type_id=1,
created_at=None, updated_at=None, deleted=None))
assert user_id is not None # ID was present
assert user_id != 0 # ID was something sensible
# Get user
user = User.get_user(cur, user_id)
assert user.id == user_id # ID was set
assert user.pin == '1234' # Pin name was set
# Update user
user.pin = '5678'
User.update_user(cur, user)
user = User.get_user(cur, user_id)
assert user.pin == '5678' # First name was updated
# Delete user
User.delete_user(cur, user_id)
user = User.get_user(cur, user_id)
assert user is None # None is returned when deleted is true
# Double-check that row still exists with manual query
cur.execute('SELECT * FROM users WHERE id=%s', (user_id,))
assert cur.rowcount == 1 # Ensure we got a row
user = User.user_from_tuple(cur.fetchone())
assert user.deleted # Make sure row still exists, and deleted is true
# Ensure we can commit without violating constraints
conn.commit()
pass
def test_crud(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
# Get nonexistent
address = Address.get_address(cur, 0)
assert address is None # Get where address doesn't exist returns None
# Create address
address_id = Address.create_address(
cur, Address(address_id=None, street='1st street', city='city', states_id=1, postal_code='55555',
country_abbreviation='US', created_at=None, updated_at=None, deleted=None))
assert address_id is not None # ID was present
assert address_id != 0 # ID was something sensible
# Get address
address = Address.get_address(cur, address_id)
assert address.id == address_id # ID was set
assert address.street == '1st street' # Street was set
# Update address
address.street = '2nd street'
Address.update_address(cur, address)
address = Address.get_address(cur, address_id)
assert address.street == '2nd street' # Street was updated
# Delete address
Address.delete_address(cur, address_id)
address = Address.get_address(cur, address_id)
assert address is None # None is returned when deleted is true
# Double-check that row still exists with manual query
cur.execute('SELECT * FROM addresses WHERE id=%s', (address_id,))
assert cur.rowcount == 1 # Ensure we got a row
address = Address.address_from_tuple(cur.fetchone())
assert address.deleted # Make sure row still exists, and deleted is true
# Ensure we can commit without violating constraints
conn.commit()
pass
def test_attempt_login(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
# Create children
user_id = User.create_user(
cur, User(user_id=None, username='******', pin='1234', user_type_id=1,
created_at=None, updated_at=None, deleted=None))
user = User.attempt_login(cur, 'bad_username', '1234') # Test wrong name
assert user is None
user = User.attempt_login(cur, 'username', '0000') # Test wrong pin
assert user is None
user = User.attempt_login(cur, 'username', '1234') # Test correct creds
assert user is not None
conn.commit()
pass
def test_crud(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
# Create children
address_id = Address.create_address(
cur,
Address(address_id=None,
street='1st street',
city='city',
states_id=1,
postal_code='55555',
country_abbreviation='US',
created_at=None,
updated_at=None,
deleted=None))
# Get nonexistent
contact_information = ContactInformation.get_contact_information(
cur, 0)
assert contact_information is None # Get where contact information doesn't exist returns None
# Create contact information
contact_information_id = ContactInformation.create_contact_information(
cur,
ContactInformation(contact_information_id=None,
email='*****@*****.**',
phone_number='+15555555555',
address_id=address_id,
created_at=None,
updated_at=None,
deleted=None))
assert contact_information_id is not None # ID was present
assert contact_information_id != 0 # ID was something sensible
# Get contact information
contact_information = ContactInformation.get_contact_information(
cur, contact_information_id)
assert contact_information.id == contact_information_id # ID was set
assert contact_information.phone_number == '+15555555555' # Phone number was set
# Update contact information
contact_information.phone_number = '+16085555555'
ContactInformation.update_contact_information(cur, contact_information)
contact_information = ContactInformation.get_contact_information(
cur, contact_information_id)
assert contact_information.phone_number == '+16085555555' # Phone number was updated
# Delete contact information
ContactInformation.delete_contact_information(cur,
contact_information_id)
contact_information = ContactInformation.get_contact_information(
cur, contact_information_id)
assert contact_information is None # None is returned when deleted is true
# Double-check that row still exists with manual query
cur.execute('SELECT * FROM contact_information WHERE id=%s',
(contact_information_id, ))
assert cur.rowcount == 1 # Ensure we got a row
contact_information = ContactInformation.contact_information_from_tuple(
cur.fetchone())
assert contact_information.deleted # Make sure row still exists, and deleted is true
# Ensure we can commit without violating constraints
conn.commit()
pass
def test_crud(self):
# Setup
conn = get_test_connection()
reset_database(conn)
cur = conn.cursor()
# Create children
contact_information_address_id = Address.create_address(
cur,
Address(address_id=None,
street='1st street',
city='city',
states_id=1,
postal_code='55555',
country_abbreviation='US',
created_at=None,
updated_at=None,
deleted=None))
contact_information_id = ContactInformation.create_contact_information(
cur,
ContactInformation(contact_information_id=None,
email='*****@*****.**',
phone_number='+15555555555',
address_id=contact_information_address_id,
created_at=None,
updated_at=None,
deleted=None))
emergency_contact_address_id = Address.create_address(
cur,
Address(address_id=None,
street='1st street',
city='city',
states_id=1,
postal_code='55555',
country_abbreviation='US',
created_at=None,
updated_at=None,
deleted=None))
emergency_contact_id = EmergencyContact.create_emergency_contact(
cur,
EmergencyContact(emergency_contact_id=None,
first_name='First',
last_name='Last',
contact_relationship_id=1,
email='*****@*****.**',
phone_number='+15555555555',
address_id=emergency_contact_address_id,
created_at=None,
updated_at=None,
deleted=None))
# Get nonexistent
user_data = UserData.get_user_data(cur, 0)
assert user_data is None # Get where user data doesn't exist returns None
# Create user data
user_data_id = UserData.create_user_data(
cur,
UserData(user_data_id=None,
first_name='first',
last_name='last',
date_of_birth=Date(1998, 1, 10),
contact_information_id=contact_information_id,
emergency_contact_id=emergency_contact_id,
created_at=None,
updated_at=None,
deleted=None))
assert user_data_id is not None # ID was present
assert user_data_id != 0 # ID was something sensible
# Get user data
user_data = UserData.get_user_data(cur, user_data_id)
assert user_data.id == user_data_id # ID was set
assert user_data.first_name == 'first' # First name was set
# Update user data
user_data.first_name = 'second'
UserData.update_user_data(cur, user_data)
user_data = UserData.get_user_data(cur, user_data_id)
assert user_data.first_name == 'second' # First name was updated
# Delete user data
UserData.delete_user_data(cur, user_data_id)
user_data = UserData.get_user_data(cur, user_data_id)
assert user_data is None # None is returned when deleted is true
# Double-check that row still exists with manual query
cur.execute('SELECT * FROM user_data WHERE id=%s', (user_data_id, ))
assert cur.rowcount == 1 # Ensure we got a row
user_data = UserData.user_data_from_tuple(cur.fetchone())
assert user_data.deleted # Make sure row still exists, and deleted is true
# Ensure we can commit without violating constraints
conn.commit()
pass
