Function TestShardingKeys

ch04/sharding_test.go:73–107

TestShardingKeys tests the Keys method by adding 5 values to the map and checking that each one exists in the keys list exactly once.

(t *testing.T)

// TestShardingKeys tests the Keys method by adding 5 values to the map and checking
// that each one exists in the keys list exactly once.
func TestShardingKeys(t *testing.T) {
	const BUCKETS = 17

	sMap := NewShardedMap[string, int](BUCKETS)

	truthMap := map[string]int{
		"alpha":   1,
		"beta":    2,
		"gamma":   3,
		"delta":   4,
		"epsilon": 5,
	}

	for k, v := range truthMap {
		sMap.Set(k, v)
	}

	keys := sMap.Keys()

	if len(truthMap) != len(keys) {
		t.Error("Map/keys mismatch")
	}

	for _, key := range sMap.Keys() {
		if _, ok := truthMap[key]; !ok {
			t.Error("Key", key, "not in truthMap")
		}

		delete(truthMap, key)
	}

	if len(truthMap) != 0 {
		t.Error("Key mismatch")
	}
}

// TestShardingDelete tests the Delete method by adding and then removing five values.
func TestShardingDelete(t *testing.T) {