def test_single_value_sql5(self): m_execute = Mock() m_fetchall = Mock(return_value=((17, 22),)) m_cursor = MagicMock() m_cursor.execute = m_execute m_cursor.fetchall = m_fetchall conn = MagicMock() conn.cursor.return_value.__enter__.return_value = m_cursor dbutil.single_row_sql(conn, "select 17, 22", (1, 2, 3)) assert conn.cursor.call_count == 1 assert m_cursor.execute.call_count == 1 m_cursor.execute.assert_called_once_with("select 17, 22", (1, 2, 3))
def test_single_value_sql5(self): m_execute = Mock() m_fetchall = Mock(return_value=((17, 22),)) m_cursor = MagicMock() m_cursor.execute = m_execute m_cursor.fetchall = m_fetchall conn = MagicMock() conn.cursor.return_value.__enter__.return_value = m_cursor dbutil.single_row_sql(conn, "select 17, 22", (1, 2, 3)) assert conn.cursor.call_count == 1 assert m_cursor.execute.call_count == 1 m_cursor.execute.assert_called_once_with("select 17, 22", (1, 2, 3))
def test_single_value_sql4(self): m_execute = Mock() m_fetchall = Mock(return_value=None) m_cursor = MagicMock() m_cursor.execute = m_execute m_cursor.fetchall = m_fetchall conn = MagicMock() conn.cursor.return_value.__enter__.return_value = m_cursor with pytest.raises(dbutil.SQLDidNotReturnSingleRow): dbutil.single_row_sql(conn, 'select 17, 22', (1, 2, 3)) assert conn.cursor.call_count == 1 assert m_cursor.execute.call_count == 1 m_cursor.execute.assert_called_once_with("select 17, 22", (1, 2, 3))
def test_single_value_sql4(self): m_execute = Mock() m_fetchall = Mock(return_value=None) m_cursor = MagicMock() m_cursor.execute = m_execute m_cursor.fetchall = m_fetchall conn = MagicMock() conn.cursor.return_value.__enter__.return_value = m_cursor with pytest.raises(dbutil.SQLDidNotReturnSingleRow): dbutil.single_row_sql(conn, 'select 17, 22', (1, 2, 3)) assert conn.cursor.call_count == 1 assert m_cursor.execute.call_count == 1 m_cursor.execute.assert_called_once_with("select 17, 22", (1, 2, 3))
def __getitem__(self, key): """return the job info or raise a KeyError""" sql = """ SELECT next_run, first_run, last_run, last_success, depends_on, error_count, last_error, ongoing FROM cron_job WHERE app_name = %s """ columns = ('next_run', 'first_run', 'last_run', 'last_success', 'depends_on', 'error_count', 'last_error', 'ongoing') try: with transaction_context(self.database_class) as conn: record = single_row_sql(conn, sql, (key, )) except SQLDidNotReturnSingleRow: raise KeyError(key) row = dict(zip(columns, record)) # Unserialize last_error into a Python dict if row['last_error']: row['last_error'] = json.loads(row['last_error']) return row
def update_crashstats_signature(self, signature, report_date, report_build): with transaction_context(self.database) as connection: # Pull the data from the db. If it's there, then do an update. If it's # not there, then do an insert. try: sql = """ SELECT signature, first_build, first_date FROM crashstats_signature WHERE signature=%s """ sig = single_row_sql(connection, sql, (signature, )) sql = """ UPDATE crashstats_signature SET first_build=%s, first_date=%s WHERE signature=%s """ params = (min(sig[1], int(report_build)), min(sig[2], string_to_datetime(report_date)), sig[0]) except SQLDidNotReturnSingleRow: sql = """ INSERT INTO crashstats_signature (signature, first_build, first_date) VALUES (%s, %s, %s) """ params = (signature, report_build, report_date) execute_no_results(connection, sql, params)
def __getitem__(self, key): """return the job info or raise a KeyError""" sql = """ SELECT next_run, first_run, last_run, last_success, depends_on, error_count, last_error, ongoing FROM cron_job WHERE app_name = %s """ columns = ( 'next_run', 'first_run', 'last_run', 'last_success', 'depends_on', 'error_count', 'last_error', 'ongoing' ) try: with transaction_context(self.database_class) as conn: record = single_row_sql(conn, sql, (key,)) except SQLDidNotReturnSingleRow: raise KeyError(key) row = dict(zip(columns, record)) # Unserialize last_error into a Python dict if row['last_error']: row['last_error'] = json.loads(row['last_error']) return row
def update_crashstats_signature(self, signature, report_date, report_build): with transaction_context(self.database) as connection: # Pull the data from the db. If it's there, then do an update. If it's # not there, then do an insert. try: sql = """ SELECT signature, first_build, first_date FROM crashstats_signature WHERE signature=%s """ sig = single_row_sql(connection, sql, (signature,)) sql = """ UPDATE crashstats_signature SET first_build=%s, first_date=%s WHERE signature=%s """ params = ( min(sig[1], int(report_build)), min(sig[2], string_to_datetime(report_date)), sig[0] ) except SQLDidNotReturnSingleRow: sql = """ INSERT INTO crashstats_signature (signature, first_build, first_date) VALUES (%s, %s, %s) """ params = (signature, report_build, report_date) execute_no_results(connection, sql, params)