Add test case for race-in-cache exercise

2-race-in-cache/check_test.go

@@ -0,0 +1,22 @@
+//////////////////////////////////////////////////////////////////////
+//
+// DO NOT EDIT THIS PART
+// Your task is to edit `main.go`
+//
+
+package main
+
+import "testing"
+
+func TestMain(t *testing.T) {
+	cache := run()
+
+	cacheLen := len(cache.cache)
+	pagesLen := cache.pages.Len()
+	if cacheLen != CacheSize {
+		t.Errorf("Incorrect cache size %v", cacheLen)
+	}
+	if pagesLen != CacheSize {
+		t.Errorf("Incorrect pages size %v", pagesLen)
+	}
+}

2-race-in-cache/main.go

@@ -69,11 +69,17 @@ func (l *Loader) Load(key string) string {
 	return val
 }
 
-func main() {
+func run() *KeyStoreCache {
 	loader := Loader{
 		DB: GetMockDB(),
 	}
 	cache := New(&loader)
 
 	RunMockServer(cache)
+
+	return cache
+}
+
+func main() {
+	run()
 }