def test_public_server_connected_to_nat(self): self.ctx.target.node.properties['rules'] = \ self.test_config['public_nat']['rules_port'] rules_count = self.get_rules_count() public_nat.server_connect_to_nat() self.assertEqual(rules_count + 3, self.get_rules_count()) public_nat.server_disconnect_from_nat() self.assertEqual(rules_count, self.get_rules_count())
def test_public_server_connected_to_nat(self): self.ctx.target.node.properties['rules'] = \ self.test_config['public_nat']['rules_port'] rules_count = self.get_rules_count() public_nat.server_connect_to_nat() self.assertEqual(rules_count + 3, self.get_rules_count()) public_nat.server_disconnect_from_nat() self.assertEqual(rules_count, self.get_rules_count())
def test_server_disconnect_from_nat(self): fake_client, fake_ctx = self.generate_client_and_context_server() fake_ctx._target.instance.runtime_properties = { network_plugin.PUBLIC_IP: '192.168.1.1' } fake_ctx._target.node.properties = { 'nat': { 'edge_gateway': 'gateway' }, 'rules': [{ 'type': 'DNAT' }] } with mock.patch('vcloud_plugin_common.VcloudAirClient.get', mock.MagicMock(return_value=fake_client)): public_nat.server_disconnect_from_nat(ctx=fake_ctx) fake_client._vdc_gateway.del_nat_rule.assert_called_with( 'DNAT', '192.168.1.1', 'any', '1.1.1.1', 'any', 'any')
def test_server_disconnect_from_nat(self): fake_client, fake_ctx = self.generate_client_and_context_server() fake_ctx._target.instance.runtime_properties = { network_plugin.PUBLIC_IP: '192.168.1.1' } fake_ctx._target.node.properties = { 'nat': { 'edge_gateway': 'gateway' }, 'rules': [{ 'type': 'DNAT' }] } with mock.patch( 'vcloud_plugin_common.VcloudAirClient.get', mock.MagicMock(return_value=fake_client) ): public_nat.server_disconnect_from_nat(ctx=fake_ctx) fake_client._vdc_gateway.del_nat_rule.assert_called_with( 'DNAT', '192.168.1.1', 'any', '1.1.1.1', 'any', 'any' )