[xcvrd] add unit test infrastructure and unit tests for xcvrd (#133)

vdahiya12 · web-flow · commit b0be7ca52868 · 2020-12-15T04:10:34.000-08:00
Summary:
This PR provides the necessary infrastructure to add pytest support and integration in sonic-xcvrd submodule.
This PR also adds unit tests for xcvrd daemon.
What is the motivation for this PR?
To add the pytest unittest support in sonic-platform-daemon, sonix-xcvrd daemon as well as add some unit tests
Signed-off-by: vaibhav-dahiya &lt;vdahiya@microsoft.com&gt;
diff --git a/sonic-xcvrd/pytest.ini b/sonic-xcvrd/pytest.ini
@@ -0,0 +1,2 @@
+[pytest]
+addopts = --cov=xcvrd --cov-report html
diff --git a/sonic-xcvrd/setup.cfg b/sonic-xcvrd/setup.cfg
@@ -0,0 +1,2 @@
+[aliases]
+test=pytest
diff --git a/sonic-xcvrd/setup.py b/sonic-xcvrd/setup.py
@@ -24,6 +24,10 @@
     setup_requires=[
         'wheel'
     ],
+    tests_require=[
+        'pytest',
+        'pytest-cov',
+    ],
     classifiers=[
         'Development Status :: 4 - Beta',
         'Environment :: No Input/Output (Daemon)',
diff --git a/sonic-xcvrd/tests/__init__.py b/sonic-xcvrd/tests/__init__.py
diff --git a/sonic-xcvrd/tests/mock_platform.py b/sonic-xcvrd/tests/mock_platform.py
@@ -0,0 +1,262 @@
+class MockDevice:
+    def __init__(self):
+        self.name = None
+        self.presence = True
+        self.model = 'FAN Model'
+        self.serial = 'Fan Serial'
+
+    def get_name(self):
+        return self.name
+
+    def get_presence(self):
+        return self.presence
+
+    def get_model(self):
+        return self.model
+
+    def get_serial(self):
+        return self.serial
+
+    def get_position_in_parent(self):
+        return 1
+
+    def is_replaceable(self):
+        return True
+
+    def get_status(self):
+        return True
+
+
+class MockFan(MockDevice):
+    STATUS_LED_COLOR_RED = 'red'
+    STATUS_LED_COLOR_GREEN = 'green'
+
+    def __init__(self):
+        MockDevice.__init__(self)
+        self.speed = 20
+        self.speed_tolerance = 20
+        self.target_speed = 20
+        self.status = True
+        self.direction = 'intake'
+        self.led_status = 'red'
+
+    def get_speed(self):
+        return self.speed
+
+    def get_speed_tolerance(self):
+        return self.speed_tolerance
+
+    def get_target_speed(self):
+        return self.target_speed
+
+    def get_status(self):
+        return self.status
+
+    def get_direction(self):
+        return self.direction
+
+    def get_status_led(self):
+        return self.led_status
+
+    def set_status_led(self, value):
+        self.led_status = value
+
+    def make_under_speed(self):
+        self.speed = 1
+        self.target_speed = 2
+        self.speed_tolerance = 0
+
+    def make_over_speed(self):
+        self.speed = 2
+        self.target_speed = 1
+        self.speed_tolerance = 0
+
+    def make_normal_speed(self):
+        self.speed = 1
+        self.target_speed = 1
+        self.speed_tolerance = 0
+
+
+class MockErrorFan(MockFan):
+    def get_speed(self):
+        raise Exception('Fail to get speed')
+
+
+class MockPsu(MockDevice):
+    def __init__(self):
+        MockDevice.__init__(self)
+        self.fan_list = []
+
+    def get_all_fans(self):
+        return self.fan_list
+
+
+class MockFanDrawer(MockDevice):
+    def __init__(self, index):
+        MockDevice.__init__(self)
+        self.name = 'FanDrawer {}'.format(index)
+        self.fan_list = []
+        self.led_status = 'red'
+
+    def get_name(self):
+        return self.name
+
+    def get_all_fans(self):
+        return self.fan_list
+
+    def get_status_led(self):
+        return self.led_status
+
+    def set_status_led(self, value):
+        self.led_status = value
+
+
+class MockThermal(MockDevice):
+    def __init__(self, index=None):
+        MockDevice.__init__(self)
+        self.name = None
+        self.name = 'Thermal {}'.format(index) if index != None else None
+        self.temperature = 2
+        self.minimum_temperature = 1
+        self.maximum_temperature = 5
+        self.high_threshold = 3
+        self.low_threshold = 1
+        self.high_critical_threshold = 4
+        self.low_critical_threshold = 0
+
+    def get_name(self):
+        return self.name
+
+    def get_temperature(self):
+        return self.temperature
+
+    def get_minimum_recorded(self):
+        return self.minimum_temperature
+
+    def get_maximum_recorded(self):
+        return self.maximum_temperature
+
+    def get_high_threshold(self):
+        return self.high_threshold
+
+    def get_low_threshold(self):
+        return self.low_threshold
+
+    def get_high_critical_threshold(self):
+        return self.high_critical_threshold
+
+    def get_low_critical_threshold(self):
+        return self.low_critical_threshold
+
+    def make_over_temper(self):
+        self.high_threshold = 2
+        self.temperature = 3
+        self.low_threshold = 1
+
+    def make_under_temper(self):
+        self.high_threshold = 3
+        self.temperature = 1
+        self.low_threshold = 2
+
+    def make_normal_temper(self):
+        self.high_threshold = 3
+        self.temperature = 2
+        self.low_threshold = 1
+
+
+class MockErrorThermal(MockThermal):
+    def get_temperature(self):
+        raise Exception('Fail to get temperature')
+
+
+class MockChassis:
+    def __init__(self):
+        self.fan_list = []
+        self.psu_list = []
+        self.thermal_list = []
+        self.fan_drawer_list = []
+        self.sfp_list = []
+        self.is_chassis_system = False
+
+    def get_all_fans(self):
+        return self.fan_list
+
+    def get_all_psus(self):
+        return self.psu_list
+
+    def get_all_thermals(self):
+        return self.thermal_list
+
+    def get_all_fan_drawers(self):
+        return self.fan_drawer_list
+
+    def get_all_sfps(self):
+        return self.sfp_list
+
+    def get_num_thermals(self):
+        return len(self.thermal_list)
+
+    def make_absence_fan(self):
+        fan = MockFan()
+        fan.presence = False
+        fan_drawer = MockFanDrawer(len(self.fan_drawer_list))
+        fan_drawer.fan_list.append(fan)
+        self.fan_list.append(fan)
+        self.fan_drawer_list.append(fan_drawer)
+
+    def make_fault_fan(self):
+        fan = MockFan()
+        fan.status = False
+        fan_drawer = MockFanDrawer(len(self.fan_drawer_list))
+        fan_drawer.fan_list.append(fan)
+        self.fan_list.append(fan)
+        self.fan_drawer_list.append(fan_drawer)
+
+    def make_under_speed_fan(self):
+        fan = MockFan()
+        fan.make_under_speed()
+        fan_drawer = MockFanDrawer(len(self.fan_drawer_list))
+        fan_drawer.fan_list.append(fan)
+        self.fan_list.append(fan)
+        self.fan_drawer_list.append(fan_drawer)
+
+    def make_over_speed_fan(self):
+        fan = MockFan()
+        fan.make_over_speed()
+        fan_drawer = MockFanDrawer(len(self.fan_drawer_list))
+        fan_drawer.fan_list.append(fan)
+        self.fan_list.append(fan)
+        self.fan_drawer_list.append(fan_drawer)
+
+    def make_error_fan(self):
+        fan = MockErrorFan()
+        fan_drawer = MockFanDrawer(len(self.fan_drawer_list))
+        fan_drawer.fan_list.append(fan)
+        self.fan_list.append(fan)
+        self.fan_drawer_list.append(fan_drawer)
+
+    def make_over_temper_thermal(self):
+        thermal = MockThermal()
+        thermal.make_over_temper()
+        self.thermal_list.append(thermal)
+
+    def make_under_temper_thermal(self):
+        thermal = MockThermal()
+        thermal.make_under_temper()
+        self.thermal_list.append(thermal)
+
+    def make_error_thermal(self):
+        thermal = MockErrorThermal()
+        self.thermal_list.append(thermal)
+
+    def is_modular_chassis(self):
+        return self.is_chassis_system
+
+    def set_modular_chassis(self, is_true):
+        self.is_chassis_system = is_true
+
+    def set_my_slot(self, my_slot):
+        self.my_slot = my_slot
+
+    def get_my_slot(self):
+        return self.my_slot
diff --git a/sonic-xcvrd/tests/mock_swsscommon.py b/sonic-xcvrd/tests/mock_swsscommon.py
@@ -0,0 +1,31 @@
+STATE_DB = ''
+CHASSIS_STATE_DB = ''
+
+
+class Table:
+    def __init__(self, db, table_name):
+        self.table_name = table_name
+        self.mock_dict = {}
+
+    def _del(self, key):
+        if key in self.mock_dict:
+            del self.mock_dict[key]
+        pass
+
+    def set(self, key, fvs):
+        self.mock_dict[key] = fvs
+        pass
+
+    def get(self, key):
+        if key in self.mock_dict:
+            return self.mock_dict[key]
+        return None
+
+    def get_size(self):
+        return (len(self.mock_dict))
+
+
+class FieldValuePairs:
+    def __init__(self, fvs):
+        self.fv_dict = dict(fvs)
+        pass
diff --git a/sonic-xcvrd/tests/t0-sample-port-config.ini b/sonic-xcvrd/tests/t0-sample-port-config.ini
@@ -0,0 +1,14 @@
+# name          lanes             alias
+Ethernet0       29,30,31,32       fortyGigE0/0
+Ethernet4       25,26,27,28       fortyGigE0/4
+Ethernet8       37,38,39,40       fortyGigE0/8
+Ethernet12      33,34,35,36       fortyGigE0/12
+Ethernet16      41,42,43,44       fortyGigE0/16
+Ethernet20      45,46,47,48       fortyGigE0/20
+Ethernet24      5,6,7,8           fortyGigE0/24
+Ethernet28      1,2,3,4           fortyGigE0/28
+Ethernet32      9,10,11,12        fortyGigE0/32
+Ethernet36      13,14,15,16       fortyGigE0/36
+Ethernet40      21,22,23,24       fortyGigE0/40
+Ethernet44      17,18,19,20       fortyGigE0/44
+Ethernet48      49,50,51,52       fortyGigE0/48
diff --git a/sonic-xcvrd/tests/test_xcvrd.py b/sonic-xcvrd/tests/test_xcvrd.py

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+[pytest]`
	`2`	`+addopts = --cov=xcvrd --cov-report html`