fboss/fboss/agent/hw/sai/store/tests/FdbStoreTest.cpp at b128dc4510b8a8e48580ecd08900af66acb98da6 · facebook/fboss · 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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/*
 *  Copyright (c) 2004-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the BSD-style license found in the
 *  LICENSE file in the root directory of this source tree. An additional grant
 *  of patent rights can be found in the PATENTS file in the same directory.
 *
 */

#include "fboss/agent/hw/sai/api/FdbApi.h"
#include "fboss/agent/hw/sai/fake/FakeSai.h"
#include "fboss/agent/hw/sai/store/SaiObject.h"
#include "fboss/agent/hw/sai/store/SaiStore.h"

#include "fboss/agent/hw/sai/store/tests/SaiStoreTest.h"

using namespace facebook::fboss;

TEST_F(SaiStoreTest, loadFdb) {
  auto& fdbApi = saiApiTable->fdbApi();
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  fdbApi.create<SaiFdbTraits>(f, {SAI_FDB_ENTRY_TYPE_STATIC, 42, 24});

  SaiStore s(0);
  s.reload();
  auto& store = s.get<SaiFdbTraits>();

  auto got = store.get(f);
  EXPECT_EQ(got->adapterKey(), f);
  EXPECT_EQ(
      std::get<SaiFdbTraits::Attributes::BridgePortId>(got->attributes())
          .value(),
      42);
  EXPECT_EQ(GET_OPT_ATTR(Fdb, Metadata, got->attributes()), 24);
}

TEST_F(SaiStoreTest, fdbLoadCtor) {
  auto& fdbApi = saiApiTable->fdbApi();
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  fdbApi.create<SaiFdbTraits>(f, {SAI_FDB_ENTRY_TYPE_STATIC, 42, 24});

  auto obj = createObj<SaiFdbTraits>(f);
  EXPECT_EQ(obj.adapterKey(), f);
  EXPECT_EQ(
      std::get<SaiFdbTraits::Attributes::BridgePortId>(obj.attributes())
          .value(),
      42);
  EXPECT_EQ(GET_ATTR(Fdb, BridgePortId, obj.attributes()), 42);
  EXPECT_EQ(GET_OPT_ATTR(Fdb, Metadata, obj.attributes()), 24);
}
TEST_F(SaiStoreTest, fdbCreateCtor) {
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  auto obj = createObj<SaiFdbTraits>(
      f, SaiFdbTraits::CreateAttributes{SAI_FDB_ENTRY_TYPE_STATIC, 42, 24}, 0);
  EXPECT_EQ(GET_ATTR(Fdb, BridgePortId, obj.attributes()), 42);
  EXPECT_EQ(GET_OPT_ATTR(Fdb, Metadata, obj.attributes()), 24);
}

TEST_F(SaiStoreTest, fdbSetBridgePort) {
  auto& fdbApi = saiApiTable->fdbApi();
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  fdbApi.create<SaiFdbTraits>(f, {SAI_FDB_ENTRY_TYPE_STATIC, 42, std::nullopt});

  SaiStore s(0);
  s.reload();
  auto& store = s.get<SaiFdbTraits>();

  auto obj = store.get(f);
  EXPECT_EQ(GET_ATTR(Fdb, BridgePortId, obj->attributes()), 42);
  store.setObject(
      f,
      {SAI_FDB_ENTRY_TYPE_STATIC, 43, 23},
      PublisherKey<SaiFdbTraits>::type{});
  EXPECT_EQ(GET_ATTR(Fdb, BridgePortId, obj->attributes()), 43);
  EXPECT_EQ(GET_OPT_ATTR(Fdb, Metadata, obj->attributes()), 23);
}

// Hardware ages dynamic FDB entries out on its own, so an entry saved in warm
// boot state can be gone by the time the store reloads it. Reload must skip it
// rather than let ITEM_NOT_FOUND abort warm boot.
TEST_F(SaiStoreTest, fdbAgedOutBeforeReload) {
  auto& fdbApi = saiApiTable->fdbApi();
  SaiFdbTraits::FdbEntry present(0, 10, folly::MacAddress{"42:42:42:42:42:42"});
  SaiFdbTraits::FdbEntry aged(0, 10, folly::MacAddress{"42:42:42:42:42:43"});
  fdbApi.create<SaiFdbTraits>(present, {SAI_FDB_ENTRY_TYPE_STATIC, 42, 24});
  fdbApi.create<SaiFdbTraits>(aged, {SAI_FDB_ENTRY_TYPE_DYNAMIC, 42, 24});

  // Warm boot state is written with both entries in it
  folly::dynamic adapterKeys;
  {
    SaiStore preWarmBoot(0);
    preWarmBoot.reload();
    adapterKeys = preWarmBoot.adapterKeysFollyDynamic();
  }
  // ... and HW ages the dynamic one out while the agent is down
  fdbApi.remove(aged);

  SaiStore postWarmBoot(0);
  postWarmBoot.reload(&adapterKeys);
  auto& store = postWarmBoot.get<SaiFdbTraits>();
  EXPECT_NE(store.get(present), nullptr);
  EXPECT_EQ(store.get(aged), nullptr);
}

TEST_F(SaiStoreTest, fdbSerDeser) {
  auto& fdbApi = saiApiTable->fdbApi();
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  fdbApi.create<SaiFdbTraits>(f, {SAI_FDB_ENTRY_TYPE_STATIC, 42, 24});
  verifyAdapterKeySerDeser<SaiFdbTraits>({f});
}

TEST_F(SaiStoreTest, fdbToStr) {
  auto& fdbApi = saiApiTable->fdbApi();
  folly::MacAddress mac{"42:42:42:42:42:42"};
  SaiFdbTraits::FdbEntry f(0, 10, mac);
  fdbApi.create<SaiFdbTraits>(f, {SAI_FDB_ENTRY_TYPE_STATIC, 42, 24});
  verifyToStr<SaiFdbTraits>();
}