def test_db_script(self, mock_cursor):
db = ConnectDb()
db.db_script()
mock_cursor().cursor().executescript.assert_called_once()
mock_cursor().commit.assert_called_once()
mock_cursor().cursor().close.assert_called_once()
mock_cursor().close.assert_called_once()
def test_db_script_fail(self, mock_cursor, mock_sys, mock_connect):
db = ConnectDb()
db.db_script()
mock_sys.exit.assert_called_once()
mock_cursor().cursor().close.assert_called_once()
mock_cursor().rollback.assert_called_once()
mock_cursor().close.assert_called_once()
member = Member()
member.member_id = result[0]
member.member_features()
if result is None or flag:
print(Color.F_Red + 'wrong credentials' + Base.END)
input()
main()
except Exception as e:
print(e)
sys.exit()
finally:
conn.close_conn()
# only executes when login file run as main script or run directly from terminal
if __name__ == '__main__':
"""
creating db schema and calling main function
"""
con = ConnectDb()
con.db_script()
main()
