8a96d29a3a
Timestamp will be used to identify when the stats are recorded. Change-Id: I1d73f360055447b0e8cde91aadd622250881ab00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import datetime
|
|
import unittest
|
|
from libra.common.lbstats import LBStatistics
|
|
|
|
|
|
class TestLBStatistics(unittest.TestCase):
|
|
def setUp(self):
|
|
self.stats = LBStatistics()
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def testInitValues(self):
|
|
now = datetime.datetime.utcnow()
|
|
self.assertEquals(self.stats.bytes_out, 0)
|
|
self.assertEquals(self.stats.bytes_in, 0)
|
|
ts = self.stats.utc_timestamp
|
|
self.assertEquals(ts.year, now.year)
|
|
self.assertEquals(ts.month, now.month)
|
|
self.assertEquals(ts.day, now.day)
|
|
self.assertEquals(ts.hour, now.hour)
|
|
|
|
def testSetBytesIn(self):
|
|
self.stats.bytes_in = 99
|
|
self.assertEquals(self.stats.bytes_in, 99)
|
|
with self.assertRaises(TypeError):
|
|
self.stats.bytes_in = "NaN"
|
|
|
|
def testSetBytesOut(self):
|
|
self.stats.bytes_out = 100
|
|
self.assertEquals(self.stats.bytes_out, 100)
|
|
with self.assertRaises(TypeError):
|
|
self.stats.bytes_out = "NaN"
|
|
|
|
def testSetUTCTimestamp(self):
|
|
ts = datetime.datetime.utcnow()
|
|
self.stats.utc_timestamp = ts
|
|
self.assertEquals(self.stats.utc_timestamp, ts)
|
|
with self.assertRaises(TypeError):
|
|
self.stats.utc_timestamp = "2012-01-01 12:00:00" |