def setUp(self): super(BareMinimumMainTestCase, self).setUp() self.conn = mock.MagicMock() with open('sushy/tests/unit/json_samples/' 'bare_minimum_root.json') as f: self.conn.get.return_value.json.return_value = json.load(f) self.root = main.Sushy('http://foo.bar:1234', verify=True, auth=mock.MagicMock(), connector=self.conn)
def test_custom_connector(self, mock_Sushy_Connector): connector_mock = mock.MagicMock() with open('sushy/tests/unit/json_samples/root.json') as f: connector_mock.get.return_value.json.return_value = ( json.load(f)) main.Sushy('http://foo.bar:1234', 'foo', 'bar', connector=connector_mock) self.assertTrue(connector_mock.post.called) self.assertTrue(connector_mock.get.called) self.assertFalse(mock_Sushy_Connector.called)
def setUp(self, mock_session_service, mock_connector, mock_auth): super(RootTestCase, self).setUp() self.conn = mock.Mock() self.sess_serv = mock.Mock() self.sess_serv.create_session.return_value = (None, None) mock_session_service.return_value = self.sess_serv mock_connector.return_value = self.conn with open('sushy_oem_idrac/tests/unit/json_samples/root.json') as f: self.conn.get.return_value.json.return_value = json.load(f) self.root = main.Sushy('http://foo.bar:1234', verify=True, auth=mock_auth)
def setUp(self, mock_connector): super(MainTestCase, self).setUp() self.conn = mock.Mock() mock_connector.return_value = self.conn with open('sushy/tests/unit/json_samples/root.json', 'r') as f: self.conn.get.return_value.json.return_value = json.loads(f.read()) self.root = main.Sushy('http://foo.bar:1234', username='******', password='******', verify=True) mock_connector.assert_called_once_with('http://foo.bar:1234', 'foo', 'bar', True)
def setUp(self, mock_session_service, mock_connector, mock_auth): super(MainTestCase, self).setUp() self.conn = mock.Mock() self.sess_serv = mock.Mock() self.sess_serv.create_session.return_value = (None, None) mock_session_service.return_value = self.sess_serv mock_connector.return_value = self.conn with open('sushy/tests/unit/json_samples/root.json', 'r') as f: self.conn.get.return_value.json.return_value = json.loads(f.read()) self.root = main.Sushy('http://foo.bar:1234', verify=True, auth=mock_auth) mock_connector.assert_called_once_with( 'http://foo.bar:1234', verify=True)