import unittest
import mock
from ...authentication.enterprise import Enterprise


class TestEnterprise(unittest.TestCase):

    @mock.patch('auth0.v2.authentication.enterprise.Enterprise.get')
    def test_saml_metadata(self, mock_get):

        e = Enterprise('my.domain.com')

        e.saml_metadata('cid')

        mock_get.assert_called_with(
            url='https://my.domain.com/samlp/metadata/cid'
        )

    @mock.patch('auth0.v2.authentication.enterprise.Enterprise.get')
    def test_wsfed_metadata(self, mock_get):

        e = Enterprise('my.domain.com')

        e.wsfed_metadata()

        mock_get.assert_called_with(
            url='https://my.domain.com/wsfed/FederationMetadata' \
                '/2007-06/FederationMetadata.xml'
        )

Add tests for Enterprise