fix: fix bandit test

getlantern · Nov 22, 2024 · d1c7f73 · d1c7f73
1 parent 3c40f7d
commit d1c7f73
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/dialer/bandit_test.go b/dialer/bandit_test.go
@@ -124,10 +124,12 @@ func TestNewBandit(t *testing.T) {
 				assert.NotNil(t, got)
 				assert.NoError(t, err)
 				assert.IsType(t, &BanditDialer{}, got)
-				assert.Equal(t, 0.7, got.(*BanditDialer).bandit.Rewards[0])
-				assert.Equal(t, 10, got.(*BanditDialer).bandit.Counts[0])
-				assert.Equal(t, float64(0), got.(*BanditDialer).bandit.Rewards[1])
-				assert.Equal(t, 0, got.(*BanditDialer).bandit.Counts[1])
+				rewards := got.(*BanditDialer).bandit.GetRewards()
+				counts := got.(*BanditDialer).bandit.GetCounts()
+				assert.Equal(t, 0.7, rewards[0])
+				assert.Equal(t, 10, counts[0])
+				assert.Equal(t, float64(0), rewards[1])
+				assert.Equal(t, 0, counts[1])
 			},
 		},
 	}