Skip to content

Commit

Permalink
refactor: removing NoError in favor of consistent Expect(err).ToNot(H…
Browse files Browse the repository at this point in the history
…aveOccurred()) (#681)
  • Loading branch information
Bryce-Soghigian authored Feb 8, 2025
1 parent ea76305 commit ffd73ff
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 15 deletions.
16 changes: 8 additions & 8 deletions pkg/controllers/nodeclaim/garbagecollection/suite_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@ import (
"github.com/Azure/karpenter-provider-azure/pkg/controllers/nodeclaim/garbagecollection"
"github.com/Azure/karpenter-provider-azure/pkg/operator/options"
"github.com/Azure/karpenter-provider-azure/pkg/providers/instance"
. "github.com/Azure/karpenter-provider-azure/pkg/test/expectations"
"github.com/Azure/karpenter-provider-azure/pkg/utils"

. "github.com/onsi/ginkgo/v2"
Expand Down Expand Up @@ -86,6 +85,7 @@ var _ = BeforeSuite(func() {
fakeClock = &clock.FakeClock{}
cluster = state.NewCluster(fakeClock, env.Client)
prov = provisioning.NewProvisioner(env.Client, events.NewRecorder(&record.FakeRecorder{}), cloudProvider, cluster, fakeClock)

})

var _ = AfterSuite(func() {
Expand Down Expand Up @@ -331,15 +331,15 @@ var _ = Describe("NetworkInterface Garbage Collection", func() {
azureEnv.NetworkInterfacesAPI.NetworkInterfaces.Store(lo.FromPtr(nic.ID), *nic)

nicsBeforeGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsBeforeGC)).To(Equal(1))

// Run garbage collection
ExpectSingletonReconciled(ctx, networkInterfaceGCController)

// Verify NIC still exists after GC
nicsAfterGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsAfterGC)).To(Equal(1))
})
It("should delete a NIC if there is no associated VM", func() {
Expand All @@ -352,13 +352,13 @@ var _ = Describe("NetworkInterface Garbage Collection", func() {
azureEnv.NetworkInterfacesAPI.NetworkInterfaces.Store(lo.FromPtr(nic.ID), *nic)
azureEnv.NetworkInterfacesAPI.NetworkInterfaces.Store(lo.FromPtr(nic2.ID), *nic2)
nicsBeforeGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsBeforeGC)).To(Equal(2))
// add a nic to azure env, and call reconcile. It should show up in the list before reconcile
// then it should not showup after
ExpectSingletonReconciled(ctx, networkInterfaceGCController)
nicsAfterGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsAfterGC)).To(Equal(0))
})
It("should not delete a NIC if there is an associated VM", func() {
Expand All @@ -371,7 +371,7 @@ var _ = Describe("NetworkInterface Garbage Collection", func() {
ExpectSingletonReconciled(ctx, networkInterfaceGCController)
// We should still have a network interface here
nicsAfterGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsAfterGC)).To(Equal(1))

})
Expand All @@ -391,12 +391,12 @@ var _ = Describe("NetworkInterface Garbage Collection", func() {
ExpectSingletonReconciled(ctx, networkInterfaceGCController)
// We should still have a network interface here
nicsAfterGC, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsAfterGC)).To(Equal(1))

ExpectSingletonReconciled(ctx, virtualMachineGCController)
nicsAfterVMReconciliation, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(nicsAfterVMReconciliation)).To(Equal(0))

})
Expand Down
3 changes: 1 addition & 2 deletions pkg/providers/instance/suite_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,6 @@ import (
karpv1 "sigs.k8s.io/karpenter/pkg/apis/v1"
corecloudprovider "sigs.k8s.io/karpenter/pkg/cloudprovider"

. "github.com/Azure/karpenter-provider-azure/pkg/test/expectations"
. "knative.dev/pkg/logging/testing"
. "sigs.k8s.io/karpenter/pkg/test/expectations"
"sigs.k8s.io/karpenter/pkg/test/v1alpha1"
Expand Down Expand Up @@ -232,7 +231,7 @@ var _ = Describe("InstanceProvider", func() {
azureEnv.NetworkInterfacesAPI.NetworkInterfaces.Store(lo.FromPtr(managedNic.ID), *managedNic)
azureEnv.NetworkInterfacesAPI.NetworkInterfaces.Store(lo.FromPtr(unmanagedNic.ID), *unmanagedNic)
interfaces, err := azureEnv.InstanceProvider.ListNics(ctx)
ExpectNoError(err)
Expect(err).ToNot(HaveOccurred())
Expect(len(interfaces)).To(Equal(1))
Expect(interfaces[0].Name).To(Equal(managedNic.Name))
})
Expand Down
5 changes: 0 additions & 5 deletions pkg/test/expectations/expectations.go
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,3 @@ func ExpectDecodedCustomData(env *test.Environment) string {

return decodedString
}

func ExpectNoError(err error) {
GinkgoHelper()
Expect(err).To(BeNil())
}

0 comments on commit ffd73ff

Please sign in to comment.