| 222 | } |
| 223 | |
| 224 | func (s) TestEndpointMap_Get(t *testing.T) { |
| 225 | em := NewEndpointMap[int]() |
| 226 | em.Set(endpoint1, 1) |
| 227 | // The second endpoint endpoint21 should override. |
| 228 | em.Set(endpoint12, 1) |
| 229 | em.Set(endpoint21, 2) |
| 230 | em.Set(endpoint3, 3) |
| 231 | em.Set(endpoint4, 4) |
| 232 | em.Set(endpoint5, 5) |
| 233 | em.Set(endpoint6, 6) |
| 234 | em.Set(endpoint7, 7) |
| 235 | |
| 236 | if got, ok := em.Get(endpoint1); !ok || got != 1 { |
| 237 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 1) |
| 238 | } |
| 239 | if got, ok := em.Get(endpoint12); !ok || got != 2 { |
| 240 | t.Fatalf("em.Get(endpoint12) = %v, %v; want %v, true", got, ok, 2) |
| 241 | } |
| 242 | if got, ok := em.Get(endpoint21); !ok || got != 2 { |
| 243 | t.Fatalf("em.Get(endpoint21) = %v, %v; want %v, true", got, ok, 2) |
| 244 | } |
| 245 | if got, ok := em.Get(endpoint3); !ok || got != 3 { |
| 246 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 3) |
| 247 | } |
| 248 | if got, ok := em.Get(endpoint4); !ok || got != 4 { |
| 249 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 4) |
| 250 | } |
| 251 | if got, ok := em.Get(endpoint5); !ok || got != 5 { |
| 252 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 5) |
| 253 | } |
| 254 | if got, ok := em.Get(endpoint6); !ok || got != 6 { |
| 255 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 6) |
| 256 | } |
| 257 | if got, ok := em.Get(endpoint7); !ok || got != 7 { |
| 258 | t.Fatalf("em.Get(endpoint1) = %v, %v; want %v, true", got, ok, 7) |
| 259 | } |
| 260 | if _, ok := em.Get(endpoint123); ok { |
| 261 | t.Fatalf("em.Get(endpoint123) = _, %v; want _, false", ok) |
| 262 | } |
| 263 | } |
| 264 | |
| 265 | func (s) TestEndpointMap_Delete(t *testing.T) { |
| 266 | em := NewEndpointMap[struct{}]() |