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_connect_to_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'
         }]
     }
     fake_client._vdc_gateway.get_public_ips = mock.MagicMock(
         return_value=['10.18.1.1'])
     with mock.patch('vcloud_plugin_common.VcloudAirClient.get',
                     mock.MagicMock(return_value=fake_client)):
         public_nat.server_connect_to_nat(ctx=fake_ctx)
     fake_client._vdc_gateway.add_nat_rule.assert_called_with(
         'DNAT', '10.18.1.1', 'any', '1.1.1.1', 'any', 'any')
 def test_server_connect_to_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'
         }]
     }
     fake_client._vdc_gateway.get_public_ips = mock.MagicMock(return_value=[
         '10.18.1.1'
     ])
     with mock.patch(
         'vcloud_plugin_common.VcloudAirClient.get',
         mock.MagicMock(return_value=fake_client)
     ):
         public_nat.server_connect_to_nat(ctx=fake_ctx)
     fake_client._vdc_gateway.add_nat_rule.assert_called_with(
         'DNAT', '10.18.1.1', 'any', '1.1.1.1', 'any', 'any'
     )