Function TestBasicOperations

plugin/plugin_test/kv_storage_test.go:126–186

Test basic operations including CRUD and edge cases

(t *testing.T)

// Test basic operations including CRUD and edge cases
func TestBasicOperations(t *testing.T) {
	setupTestEnvironment()
	kv := testPlugin.operator
	ctx := context.Background()

	t.Run("BasicCRUD", func(t *testing.T) {
		// Set/Get
		mustSet(t, kv, ctx, "group1", "key1", "value1")
		mustGet(t, kv, ctx, "group1", "key1", "value1")

		// Update
		mustSet(t, kv, ctx, "group1", "key1", "new_value")
		mustGet(t, kv, ctx, "group1", "key1", "new_value")

		// Delete
		mustDel(t, kv, ctx, "group1", "key1")
		assertNotFound(t, kv, ctx, "group1", "key1")

		// Group operation
		mustSet(t, kv, ctx, "group1", "key2", "value2")
		mustSet(t, kv, ctx, "group1", "key3", "value3")
		groupData, err := kv.GetByGroup(ctx, plugin.KVParams{Group: "group1", Page: 1, PageSize: 10})
		if err != nil {
			t.Fatalf("Failed to get group data: %v", err)
		}

		// the groupData should only have key2 and key3 because key1 is deleted
		if len(groupData) != 2 {
			t.Errorf("Expected 2 items, got %d", len(groupData))
		}
		if groupData["key2"] != "value2" || groupData["key3"] != "value3" {
			t.Errorf("Unexpected group data: %v", groupData)
		}
	})

	t.Run("EdgeCases", func(t *testing.T) {
		// Empty key
		err := kv.Set(ctx, plugin.KVParams{Group: "group", Key: "", Value: "value"})
		assertError(t, err, plugin.ErrKVKeyEmpty, "Empty key test")

		// Empty group query
		_, err = kv.GetByGroup(ctx, plugin.KVParams{Group: "", Page: 1, PageSize: 10})
		assertError(t, err, plugin.ErrKVGroupEmpty, "Empty group test")

		// Non-existent key
		assertNotFound(t, kv, ctx, "non_exist_group", "non_exist_key")

		// Cache penetration protection
		key := fmt.Sprintf("non_exist_key_%d", time.Now().UnixNano())
		assertNotFound(t, kv, ctx, "cache_penetration", key)
	})

	t.Run("CacheConsistency", func(t *testing.T) {
		mustSet(t, kv, ctx, "cache_group", "cache_key", "cache_value")
		mustGet(t, kv, ctx, "cache_group", "cache_key", "cache_value")

		// Update and verify immediate consistency
		mustSet(t, kv, ctx, "cache_group", "cache_key", "updated_value")