使用Mox测试函数链中的多个API调用

Question

我正在尝试测试我是否正确地转换了从第三方api返回的数据。我在Mox上遇到了一些麻烦，因为在数据转换过程中我需要达到两个独立的端点。让我通过张贴代码更清楚地解释：

测试：

test "players/0 return all active players" do
Statcasters.SportRadarNbaApi.ClientMock
|> expect(:league_hierarchy, fn ->
{:ok, league_hierarchy_map()}
end)
Statcasters.SportRadarNbaApi.ClientMock
|> expect(:team_profile, fn _ ->
{:ok, team_profile_map()}
end)

assert Statcasters.Sports.Nba.get_players() == ["Kevon Looney", "Patrick McCaw"]
end

代码：

def get_players do
with {:ok, hierarchy} <- @sport_radar_nba_api.league_hierarchy,
team_ids <- get_team_ids(hierarchy),
players <- get_team_players(team_ids)
do
IO.inspect players
end
end
defp get_team_players(team_ids) do
for team_id <- team_ids do
{:ok, team} = @sport_radar_nba_api.team_profile(team_id)
end
end

忽略这样一个事实，即编写的代码实际上不会通过测试。这是我试图弄清楚的测试失败。

问题：

第二个api调用team_profile在测试中被调用两次，因为我迭代两个team_ids，对于每个team_id，我调用api。这是意料之中的事，但测试没有为此做好准备，因为我遇到了这个错误。

错误：

** (Mox.UnexpectedCallError) expected Statcasters.SportRadarNbaApi.ClientMock.team_profile/1 to be called once but it has been called 2 times in process #PID<0.410.0>

这是正确的。我确实调用了两次，但我如何设置测试以期望这个API端点将被调用两次？

Answer 1

expect的第三个可选参数是应调用模拟函数的次数。在这种情况下，只需将其设置为2:

Statcasters.SportRadarNbaApi.ClientMock
|> expect(:team_profile, 2, fn _ ->
{:ok, team_profile_map()}
end)