def test_node_communicator_all_options(self, mock_comm): expected_comm = mock.MagicMock() mock_comm.return_value = expected_comm user = "******" groups = ["some", "group"] tokens = {"node": "token"} timeout = 10 env = LibraryEnvironment( self.mock_logger, self.mock_reporter, user_login=user, user_groups=groups, auth_tokens_getter=lambda:tokens, request_timeout=timeout ) comm = env.node_communicator() self.assertEqual(expected_comm, comm) mock_comm.assert_called_once_with( self.mock_logger, self.mock_reporter, tokens, user, groups, timeout )
def test_node_communicator_no_options(self, mock_comm): expected_comm = mock.MagicMock() mock_comm.return_value = expected_comm env = LibraryEnvironment(self.mock_logger, self.mock_reporter) comm = env.node_communicator() self.assertEqual(expected_comm, comm) mock_comm.assert_called_once_with(self.mock_logger, self.mock_reporter, {}, None, [])
def test_node_communicator_no_options(self, mock_comm): expected_comm = mock.MagicMock() mock_comm.return_value = expected_comm env = LibraryEnvironment(self.mock_logger, self.mock_reporter) comm = env.node_communicator() self.assertEqual(expected_comm, comm) mock_comm.assert_called_once_with( self.mock_logger, self.mock_reporter, {}, None, [] )
def test_node_communicator_all_options(self, mock_comm): expected_comm = mock.MagicMock() mock_comm.return_value = expected_comm user = "******" groups = ["some", "group"] tokens = {"node": "token"} env = LibraryEnvironment(self.mock_logger, self.mock_reporter, user_login=user, user_groups=groups, auth_tokens_getter=lambda: tokens) comm = env.node_communicator() self.assertEqual(expected_comm, comm) mock_comm.assert_called_once_with(self.mock_logger, self.mock_reporter, tokens, user, groups)