1 """Tests for mysql_stats.""" 2 3 import common 4 5 import collections 6 import mock 7 import unittest 8 9 import mysql_stats 10 11 12 class MysqlStatsTest(unittest.TestCase): 13 """Unittest for mysql_stats.""" 14 15 def testQueryAndEmit(self): 16 """Test for QueryAndEmit.""" 17 connection = mock.Mock() 18 connection.Fetchall.return_value = [( 19 'Column-name', 0)] 20 21 # This shouldn't raise an exception. 22 mysql_stats.QueryAndEmit(collections.defaultdict(lambda: 0), connection) 23 24 25 if __name__ == '__main__': 26 unittest.main() 27