Angular单元测试函数根据不同的参数returnValue不同的值

编写Angular单元测试时，有一个场景是某个函数被多次调用，在mock这个函数的返回值的时候希望根据不同的参数返回不同的值。

当需要针对同一个函数根据不同的参数返回不同的值时，你可以使用 Jasmine 的 calls.fake() 来模拟这种行为。这可以让你为不同的参数设置不同的返回值。

以下是一个示例，演示了如何在单元测试中根据不同的参数模拟返回不同的值：

假设你有一个名为 getData 的服务方法，根据不同的参数返回不同的数据。以下是如何编写单元测试：

import { TestBed } from '@angular/core/testing';
import { DataService } from './data.service';

describe('DataService', () => {
  let service: DataService;

  beforeEach(() => {
    TestBed.configureTestingModule({
      providers: [DataService]
    });
    service = TestBed.inject(DataService);
  });

  it('should return value 1 for param A', () => {
    spyOn(service, 'getData').and.callFake((param: string) => {
      if (param === 'A') {
        return 'Value for A';
      }
      // Add other conditions for different params if needed
    });

    const resultA = service.getData('A');
    expect(resultA).toBe('Value for A');
  });

  it('should return value 2 for param B', () => {
    spyOn(service, 'getData').and.callFake((param: string) => {
      if (param === 'B') {
        return 'Value for B';
      }
      // Add other conditions for different params if needed
    });

    const resultB = service.getData('B');
    expect(resultB).toBe('Value for B');
  });

  // Add other test cases for different parameters and their respective return values
});

在这个示例中，spyOn 函数被用来模拟 DataService 中的 getData 方法，并使用 and.callFake() 来根据传入的参数返回不同的值。针对不同的参数进行了不同的模拟行为，并对返回值进行了验证。

这样的测试用例可以确保函数在不同参数下返回了预期的值，并对函数的行为进行了全面的验证。