def test_mysql(self, pymysql_mock, get_config_mock): connection = mock.MagicMock() curs = mock.MagicMock() side_effect = [(k, 0) for k in mysql.COLLECT_COUNTERS] side_effect.append(None) # Instead of StopIteration pymsql uses None curs.fetchone.side_effect = side_effect connection.cursor.return_value = curs pymysql_mock.return_value = connection result = mysql.collect() self.assertEqual(sorted(mysql.COLLECT_COUNTERS), sorted(result.keys())) self.assertTrue(all([val == 0 for val in result.values()]))
def get_mysql(): global mysql_data mysql_data = mysql.collect()