add tests

test/cache/driver.go

@@ -36,6 +36,25 @@ const cachePlanRaw = `queries:
         operator: eq
         placeholder:
           index: 0
+  - query: UPDATE ` + "`" + `users` + "`" + ` SET ` + "`" + `name` + "`" + ` = ? WHERE ` + "`" + `id` + "`" + ` = ?;
+    type: update
+    table: users
+    targets:
+      - column: name
+        placeholder:
+          index: 0
+    conditions:
+      - column: id
+        operator: eq
+        placeholder:
+          index: 1
+  - query: INSERT INTO ` + "`" + `users` + "`" + ` (` + "`" + `name` + "`" + `, ` + "`" + `age` + "`" + `, ` + "`" + `created_at` + "`" + `) VALUES (?, ?, ?);
+    type: insert
+    table: users
+    columns:
+      - name
+      - age
+      - created_at
 `
 
 const schemaRaw = `CREATE TABLE ` + "`" + `users` + "`" + ` (

test/cache/reset.go

@@ -0,0 +1,15 @@
+package cache
+
+import (
+	"time"
+
+	"github.com/motoki317/sc"
+)
+
+func ResetCache() {
+	for key := range caches {
+		v := caches[key]
+		*v.cache = *sc.NewMust(replaceFn, 10*time.Minute, 10*time.Minute)
+		caches[key] = v
+	}
+}

test/cache_test.go

@@ -2,14 +2,15 @@ package test
 
 import (
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/traP-jp/isuc/normalizer"
 	"github.com/traP-jp/isuc/test/cache"
 )
 
 func TestSimpleQuery(t *testing.T) {
-	cache.PurgeAllCaches()
+	cache.ResetCache()
 	db := NewDB(t)
 
 	var user User
@@ -24,3 +25,95 @@ func TestSimpleQuery(t *testing.T) {
 	assert.Equal(t, 0, stats.Hits)
 	assert.Equal(t, 1, stats.Misses)
 }
+
+func TestSimpleQueryCache(t *testing.T) {
+	cache.ResetCache()
+	db := NewDB(t)
+
+	var user User
+	err := db.Get(&user, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, InitialData[0], user)
+
+	var user2 User
+	err = db.Get(&user2, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, InitialData[0], user2)
+
+	stats := cache.ExportCacheStats()[normalizer.NormalizeQuery("SELECT * FROM `users` WHERE `id` = ?")]
+	assert.Equal(t, 1, stats.Hits)
+	assert.Equal(t, 1, stats.Misses)
+}
+
+func TestSelectAfterUpdate(t *testing.T) {
+	cache.ResetCache()
+	db := NewDB(t)
+
+	var user User
+	err := db.Get(&user, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, InitialData[0], user)
+
+	_, err = db.Exec("UPDATE `users` SET `name` = ? WHERE `id` = ?", "updated", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+	user.Name = "updated"
+
+	// no cache hit because users with id=1 is updated
+	var user2 User
+	err = db.Get(&user2, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, user, user2)
+
+	stats := cache.ExportCacheStats()[normalizer.NormalizeQuery("SELECT * FROM `users` WHERE `id` = ?")]
+	assert.Equal(t, 0, stats.Hits)
+	assert.Equal(t, 2, stats.Misses)
+}
+
+func TestSelectAfterInsert(t *testing.T) {
+	cache.ResetCache()
+	db := NewDB(t)
+
+	var user User
+	err := db.Get(&user, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, InitialData[0], user)
+
+	newUser := User{
+		Name:      "new",
+		Age:       10,
+		CreatedAt: time.Date(2021, 1, 1, 0, 0, 0, 0, time.UTC),
+	}
+	_, err = db.Exec("INSERT INTO `users` (`name`, `age`, `created_at`) VALUES (?, ?, ?)", newUser.Name, newUser.Age, newUser.CreatedAt)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// cache hit because users with id=1 is not updated
+	err = db.Get(&user, "SELECT * FROM `users` WHERE `id` = ?", 1)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	AssertUser(t, InitialData[0], user)
+
+	stats := cache.ExportCacheStats()[normalizer.NormalizeQuery("SELECT * FROM `users` WHERE `id` = ?")]
+	assert.Equal(t, 1, stats.Hits)
+	assert.Equal(t, 1, stats.Misses)
+}

test/sample.yaml

@@ -14,3 +14,22 @@ queries:
         operator: eq
         placeholder:
           index: 0
+  - query: UPDATE `users` SET `name` = ? WHERE `id` = ?;
+    type: update
+    table: users
+    targets:
+      - column: name
+        placeholder:
+          index: 0
+    conditions:
+      - column: id
+        operator: eq
+        placeholder:
+          index: 1
+  - query: INSERT INTO `users` (`name`, `age`, `created_at`) VALUES (?, ?, ?);
+    type: insert
+    table: users
+    columns:
+      - name
+      - age
+      - created_at

test/user.go

@@ -17,6 +17,7 @@ type User struct {
 }
 
 func AssertUser(t *testing.T, expected, actual User) {
+	t.Helper()
 	assert.Equal(t, expected.ID, actual.ID, "ID")
 	assert.Equal(t, expected.Name, actual.Name, "Name")
 	assert.Equal(t, expected.Age, actual.Age, "Age")