github.com/vmware/govmomi@v0.51.0/simulator/host_firewall_system_test.go (about)

     1  // © Broadcom. All Rights Reserved.
     2  // The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries.
     3  // SPDX-License-Identifier: Apache-2.0
     4  
     5  package simulator
     6  
     7  import (
     8  	"context"
     9  	"testing"
    10  
    11  	"github.com/vmware/govmomi/object"
    12  	"github.com/vmware/govmomi/simulator/esx"
    13  )
    14  
    15  func TestHostFirewallSystem(t *testing.T) {
    16  	ctx := context.Background()
    17  
    18  	m := ESX()
    19  	m.Datastore = 0
    20  	m.Machine = 0
    21  
    22  	err := m.Create()
    23  	if err != nil {
    24  		t.Fatal(err)
    25  	}
    26  
    27  	c := m.Service.client()
    28  
    29  	host := object.NewHostSystem(c, esx.HostSystem.Reference())
    30  
    31  	hfs, _ := host.ConfigManager().FirewallSystem(ctx)
    32  
    33  	err = hfs.DisableRuleset(ctx, "enoent")
    34  	if err == nil {
    35  		t.Error("expected error")
    36  	}
    37  
    38  	err = hfs.EnableRuleset(ctx, "enoent")
    39  	if err == nil {
    40  		t.Error("expected error")
    41  	}
    42  
    43  	err = hfs.DisableRuleset(ctx, "sshServer")
    44  	if err != nil {
    45  		t.Error(err)
    46  	}
    47  
    48  	err = hfs.EnableRuleset(ctx, "sshServer")
    49  	if err != nil {
    50  		t.Error(err)
    51  	}
    52  
    53  	_, err = hfs.Info(ctx)
    54  	if err != nil {
    55  		t.Fatal(err)
    56  	}
    57  }