fp/tests/pick.test.ts at main · bitmap/fp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import { pick, map } from "../lib";

interface User {
  name: string;
  age?: number;
  city?: string;
  state?: string;
}

describe("pick", () => {
  test("returns pick", () => {
    const data: User = {
      name: "Tom",
      age: 32,
      city: "New York",
      state: "NY",
    };

    expect(pick(["name", "age"], data)).toEqual({
      name: "Tom",
      age: 32,
    });
  });

  test("mappable", () => {
    const data: User[] = [
      {
        name: "Tom",
        age: 32,
        city: "New York",
        state: "NY",
      },
      {
        name: "Dick",
        age: 41,
        city: "San Francisco",
        state: "CA",
      },
      {
        name: "Harry",
        age: 27,
        city: "Portland",
        state: "OR",
      },
      {
        name: "Jim",
      },
    ];

    const pickFrom = map(pick(["name", "city"]), data);

    expect(pickFrom).toEqual([
      {
        name: "Tom",
        city: "New York",
      },
      {
        name: "Dick",
        city: "San Francisco",
      },
      {
        name: "Harry",
        city: "Portland",
      },
      {
        name: "Jim",
      },
    ]);
  });
});