class RolesTest(TestCase): def setUp(self): self.host = '127.0.0.1' self.connection = connection(self.host) self._client = Roles(self.connection) @mock.patch.object(ResourceClient, 'get_all') def test_get_all_called_once(self, mock_get_all): filter = 'name=TestName' sort = 'name:ascending' self._client.get_all(2, 500, filter=filter, sort=sort) mock_get_all.assert_called_once_with(count=500, filter='name=TestName', sort='name:ascending', start=2) @mock.patch.object(ResourceClient, 'get_all') def test_get_all_called_once_with_default(self, mock_get_all): self._client.get_all() mock_get_all.assert_called_once_with(count=-1, filter=u'', sort=u'', start=0) @mock.patch.object(ResourceClient, 'get') def test_get_by_id_called_once(self, mock_get): id = "Infrastructure administrator" self._client.get(id) mock_get.assert_called_once_with('Infrastructure%20administrator') @mock.patch.object(ResourceClient, 'get') def test_get_by_uri_called_once(self, mock_get): uri = "/rest/roles/Infrastructure administrator" self._client.get(uri) mock_get.assert_called_once_with('/rest/roles/Infrastructure%20administrator')
def roles(self): """ Gets the Roles API client. Returns: Roles: """ if not self.__roles: self.__roles = Roles(self.__connection) return self.__roles
def setUp(self): self.host = '127.0.0.1' self.connection = connection(self.host) self._client = Roles(self.connection)