"""Tests for mysql_stats.""" | |
import common | |
import collections | |
import mock | |
import unittest | |
import mysql_stats | |
class MysqlStatsTest(unittest.TestCase): | |
"""Unittest for mysql_stats.""" | |
def testQueryAndEmit(self): | |
"""Test for QueryAndEmit.""" | |
cursor = mock.Mock() | |
cursor.fetchone.return_value = ('Column-name', 0) | |
# This shouldn't raise an exception. | |
mysql_stats.QueryAndEmit(collections.defaultdict(lambda: 0), cursor) | |
if __name__ == '__main__': | |
unittest.main() |