def test_get_technology_id(session):
"""Verify that the get_technology_id() returns
correct id of the technology in the table provided
technology type.
"""
# insert data
techs = [
Technologies(id=123, description='123 Tech'),
Technologies(id=567, description='567 Tech'),
Technologies(id=890, description='890 Tech')
]
session.bulk_save_objects(techs)
session.commit()
assert Technologies.get_technology_id('123 Tech') == 123
assert Technologies.get_technology_id('567 Tech') == 567
assert Technologies.get_technology_id('890 Tech') == 890
assert Technologies.get_technology_id('hdheukkslkl') is None
def __init__(self, reg_device_id, technology):
"""Constructor."""
self.reg_device_id = reg_device_id
self.technology_id = Technologies.get_technology_id(technology)
