Skip to content

Commit 0a14b61

Browse files
fix(irq-tuning): fix UT
fix UT Signed-off-by: 张浩宇 <zhanghaoyu.zhy@bytedance.com>
1 parent a120f57 commit 0a14b61

1 file changed

Lines changed: 16 additions & 0 deletions

File tree

pkg/util/machine/network_linux_test.go

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,7 @@ func TestSetIrqAffinity(t *testing.T) {
736736
capturedCpuset = cpuset
737737
return nil
738738
}).Build()
739+
Mock(GetIrqAffinityCPUs).Return([]int64{4}, nil).Build()
739740

740741
err := SetIrqAffinity(validIrq, validCpu)
741742

@@ -744,6 +745,20 @@ func TestSetIrqAffinity(t *testing.T) {
744745
So(capturedIrq, ShouldEqual, validIrq)
745746
So(capturedCpuset, ShouldEqual, expectedCpusetStr)
746747
})
748+
749+
PatchConvey("Scenario: failed set irq affinity", func() {
750+
irq := 10
751+
cpu := int64(4)
752+
affinityCPUList := []int64{2}
753+
754+
Mock(procm.ApplyProcInterrupts).Return(nil).Build()
755+
Mock(GetIrqAffinityCPUs).Return(affinityCPUList, nil).Build()
756+
757+
err := SetIrqAffinity(irq, cpu)
758+
759+
So(err, ShouldNotBeNil)
760+
So(err.Error(), ShouldEqual, fmt.Sprintf("set irq %d affinity cpu %d, but actually affinity cpus %+v", irq, cpu, affinityCPUList))
761+
})
747762
})
748763
}
749764

@@ -2345,6 +2360,7 @@ func TestListNetNS(t *testing.T) {
23452360
Mock(os.ReadDir).Return([]os.DirEntry{
23462361
&mockDirEntry{entryName: "ns1", isDir: false},
23472362
}, nil).Build()
2363+
Mock(os.Stat).Return(nil, nil).Build()
23482364
Mock(general.GetFileInode).Return(uint64(0), mockErr).Build()
23492365

23502366
nsList, err := ListNetNS(testNetNSDir)

0 commit comments

Comments
 (0)