def test_get_smt_status_notsupported(self, mock_smt_suport): """ Unittest to fetch smt status when smt not supported. """ mock_smt_suport.return_value = False with self.assertRaisesRegexp(OperationFailed, 'GINSMT0006E'): smtmodel = SmtModel() smtmodel.get_smt_status_s390x() mock_smt_suport.assert_called_with()
def test_get_smt_status_supported(self, mock_current, mock_persistent, mock_smt_suport): """ Unittest to fetch smt status """ current_info = {"status": "enabled", "smt": "2"} perisistent_info = {"status": "enabled", "smt": "2"} mock_smt_suport.return_value = True mock_current.return_value = current_info mock_persistent.return_value = perisistent_info smtmodel = SmtModel() out = smtmodel.get_smt_status_s390x() mock_smt_suport.assert_called_with() self.assertEqual(len(out), 2)