def test_get_capabilities1(self): """Test with capabilities""" capabilities = {'capabilities': { 'algorithms': ['algo1', 'algo2'], 'protocols': ['udp', 'tcp']}} device_fake1['extra'] = capabilities base_driver = BaseDriver(self.conf, device_fake1) self.assertDictEqual(base_driver.get_capabilities(), capabilities['capabilities'])
def test_get_capabilities(self): """Test without capabilities""" base_driver = BaseDriver(self.conf, device_fake1) self.assertEqual(base_driver.get_capabilities(), None)