def setUp(self):
self.network_name = "test_net_name"
self.driver = DefaultPolicyDriver(self.network_name)
assert_equal(self.driver.profile_name, self.network_name)
# Mock the DatastoreClient
self.client = MagicMock(spec=DatastoreClient)
self.driver._client = self.client
def setUp(self):
self.network_name = "test_net_name"
self.driver = DefaultPolicyDriver(self.network_name)
assert_equal(self.driver.profile_name, self.network_name)
# Mock the DatastoreClient
self.client = MagicMock(spec=DatastoreClient)
self.driver._client = self.client
class DefaultPolicyDriverTest(unittest.TestCase):
"""
Test class for DefaultPolicyDriver class.
"""
def setUp(self):
self.network_name = "test_net_name"
self.driver = DefaultPolicyDriver(self.network_name)
assert_equal(self.driver.profile_name, self.network_name)
# Mock the DatastoreClient
self.client = MagicMock(spec=DatastoreClient)
self.driver._client = self.client
def test_apply_new_profile(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = []
endpoint.endpoint_id = "12345"
self.client.profile_exists.return_value = False
# Mock out generate_tags.
self.driver.generate_tags = MagicMock(spec=self.driver.generate_tags)
self.driver.generate_tags.return_value = ["tag1", "tag2"]
# Call
self.driver.apply_profile(endpoint)
# Assert
self.client.append_profiles_to_endpoint.assert_called_once_with(
profile_names=[self.network_name], endpoint_id="12345")
def test_apply_same_profile(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = [self.network_name]
endpoint.endpoint_id = "12345"
self.client.profile_exists.return_value = False
# Call
self.driver.apply_profile(endpoint)
# Assert
assert_false(self.client.append_profiles_to_endpoint.called)
def test_apply_profile_error(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = []
endpoint.endpoint_id = "12345"
endpoint.name = "cali12345"
self.client.profile_exists.return_value = False
self.client.append_profiles_to_endpoint.side_effect = KeyError
# Call
assert_raises(ApplyProfileError, self.driver.apply_profile, endpoint)
def test_remove_profile(self):
# Should do nothing.
self.driver.remove_profile()
@parameterized.expand([
("invalid=name"),
("^regex$"),
])
def test_invalid_network_name(self, net_name):
assert_raises(ValueError, DefaultPolicyDriver, net_name)
class DefaultPolicyDriverTest(unittest.TestCase):
"""
Test class for DefaultPolicyDriver class.
"""
def setUp(self):
self.network_name = "test_net_name"
self.driver = DefaultPolicyDriver(self.network_name)
assert_equal(self.driver.profile_name, self.network_name)
# Mock the DatastoreClient
self.client = MagicMock(spec=DatastoreClient)
self.driver._client = self.client
def test_apply_new_profile(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = []
endpoint.endpoint_id = "12345"
self.client.profile_exists.return_value = False
# Mock out generate_tags.
self.driver.generate_tags = MagicMock(spec=self.driver.generate_tags)
self.driver.generate_tags.return_value = ["tag1", "tag2"]
# Call
self.driver.apply_profile(endpoint)
# Assert
self.client.append_profiles_to_endpoint.assert_called_once_with(
profile_names=[self.network_name], endpoint_id="12345"
)
def test_apply_same_profile(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = [self.network_name]
endpoint.endpoint_id = "12345"
self.client.profile_exists.return_value = False
# Call
self.driver.apply_profile(endpoint)
# Assert
assert_false(self.client.append_profiles_to_endpoint.called)
def test_apply_profile_error(self):
# Mock.
endpoint = MagicMock(spec=Endpoint)
endpoint.profile_ids = []
endpoint.endpoint_id = "12345"
endpoint.name = "cali12345"
self.client.profile_exists.return_value = False
self.client.append_profiles_to_endpoint.side_effect = KeyError
# Call
assert_raises(ApplyProfileError, self.driver.apply_profile, endpoint)
def test_remove_profile(self):
# Should do nothing.
self.driver.remove_profile()
@parameterized.expand([
("invalid=name"), ("^regex$"),
])
def test_invalid_network_name(self, net_name):
assert_raises(ValueError, DefaultPolicyDriver, net_name)