Python Config.get_authentication_headers Examples

Example #1

File: test_config.py Project: psriramula/dj-dna-streams-python

    def test_get_headers_user_key(self):
        # Arrange
        user_key = "just some user key"
        config = Config(user_key)

        headers_expected = {'user-key': user_key}

        fetch_jwt_mock = self.patch_module(config._fetch_jwt, '')

        # Act
        headers_actual = config.get_authentication_headers()

        # Assert
        assert headers_actual == headers_expected
        fetch_jwt_mock.assert_not_called()

Example #2

File: test_config.py Project: psriramula/dj-dna-streams-python

    def test_get_headers_jwt(self):
        # Arrange
        config = Config()

        fileFolder = os.path.dirname(os.path.realpath(__file__))
        config._set_customer_config_path(
            os.path.join(fileFolder, 'test_customer_config.json'))

        jwt = "Bearer of Bad News"
        fetch_jwt_mock = self.patch_module(config._fetch_jwt, jwt)

        headers_expected = {'Authorization': jwt}

        # Act
        headers_actual = config.get_authentication_headers()

        # Assert
        assert headers_actual == headers_expected
        fetch_jwt_mock.assert_called_once()