public void GetPumpsState_PumpList_ReturnsPumpStateForEachPump() { _mockIO.Setup(c => c.SendAndReceive(CompanyCommands.GetPumpsState)).Returns("(SAL)"); var forecourt = new Company(_mockIO.Object); var pump1 = new Pump(1); var pump2 = new Pump(2); var pumpList = new List<Pump>() { pump1, pump2 }; var dic = forecourt.GetPumpsState(pumpList); forecourt.Dispose(); dic[pump1].Should().Be(PumpState.Delivering); dic[pump2].Should().Be(PumpState.Free); }