|
1 | 1 | package chainfee |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "maps" |
4 | 5 | "math/big" |
5 | 6 | "math/rand" |
| 7 | + "slices" |
6 | 8 | "sort" |
7 | 9 | "testing" |
8 | 10 | "time" |
9 | 11 |
|
10 | 12 | "github.com/stretchr/testify/mock" |
11 | | - "golang.org/x/exp/maps" |
12 | 13 |
|
13 | 14 | mapset "github.com/deckarep/golang-set/v2" |
14 | 15 | "github.com/stretchr/testify/require" |
@@ -187,11 +188,11 @@ func Test_processor_Observation(t *testing.T) { |
187 | 188 | require.GreaterOrEqual(t, obs.TimestampNow.UnixNano(), tStart.UnixNano()) |
188 | 189 | require.LessOrEqual(t, obs.TimestampNow.UnixNano(), tEnd.UnixNano()) |
189 | 190 | require.Equal(t, tc.chainFeeComponents, obs.FeeComponents) |
190 | | - require.ElementsMatch(t, slicesWithoutDst, maps.Keys(obs.FeeComponents)) |
| 191 | + require.ElementsMatch(t, slicesWithoutDst, slices.Collect(maps.Keys(obs.FeeComponents))) |
191 | 192 | require.Equal(t, tc.nativeTokenPrices, obs.NativeTokenPrices) |
192 | | - require.ElementsMatch(t, slicesWithoutDst, maps.Keys(obs.NativeTokenPrices)) |
| 193 | + require.ElementsMatch(t, slicesWithoutDst, slices.Collect(maps.Keys(obs.NativeTokenPrices))) |
193 | 194 | require.Equal(t, tc.expectedChainFeePriceUpdates, obs.ChainFeeUpdates) |
194 | | - require.ElementsMatch(t, slicesWithoutDst, maps.Keys(obs.ChainFeeUpdates)) |
| 195 | + require.ElementsMatch(t, slicesWithoutDst, slices.Collect(maps.Keys(obs.ChainFeeUpdates))) |
195 | 196 | require.Equal(t, tc.fChain, obs.FChain) |
196 | 197 | }) |
197 | 198 | } |
@@ -385,9 +386,12 @@ func Test_unique_chain_filter_in_Observation(t *testing.T) { |
385 | 386 | return |
386 | 387 | } |
387 | 388 |
|
388 | | - require.True(t, tc.expUniqueChains == len(maps.Keys(obs.FeeComponents))) |
389 | | - require.True(t, tc.expUniqueChains == len(maps.Keys(obs.NativeTokenPrices))) |
390 | | - require.ElementsMatch(t, maps.Keys(obs.FeeComponents), maps.Keys(obs.NativeTokenPrices)) |
| 389 | + require.True(t, tc.expUniqueChains == len(obs.FeeComponents)) |
| 390 | + require.True(t, tc.expUniqueChains == len(obs.NativeTokenPrices)) |
| 391 | + require.ElementsMatch(t, |
| 392 | + slices.Collect(maps.Keys(obs.FeeComponents)), |
| 393 | + slices.Collect(maps.Keys(obs.NativeTokenPrices)), |
| 394 | + ) |
391 | 395 | }) |
392 | 396 | } |
393 | 397 | } |
0 commit comments