Improve compatibility with advanced DataStore features

README.md

@@ -21,7 +21,14 @@ client.Put(ctx, key, &task)
 client.Get(ctx, key, &task)
 ```
 
-**Supported:** Get, Put, Delete, GetMulti, PutMulti, DeleteMulti, RunInTransaction, AllKeys, NameKey, IDKey, parent keys.
+**Supported:**
+- **CRUD**: Get, Put, Delete, GetMulti, PutMulti, DeleteMulti
+- **Transactions**: RunInTransaction, NewTransaction, Commit, Rollback
+- **Queries**: Filter, Order, Limit, Offset, Ancestor, Project, Distinct, DistinctOn, Namespace, Run (iterator), Count
+- **Cursors**: Start, End, DecodeCursor
+- **Keys**: NameKey, IDKey, IncompleteKey, AllocateIDs, parent keys
+- **Mutations**: NewInsert, NewUpdate, NewUpsert, NewDelete, Mutate
+- **Types**: string, int, int64, int32, bool, float64, time.Time, slices ([]string, []int64, []int, []float64, []bool)
 
 ## Migrating from Official Client
 
@@ -35,6 +42,6 @@ Use `ds9mock` package for in-memory testing. See [TESTING.md](TESTING.md) for in
 
 ## Limitations
 
-Not supported: property filters, ordering, cursors, ancestor queries, slices/arrays, embedded structs, key allocation.
+Not supported: embedded structs, nested slices, map types, some advanced query features (streaming aggregations, OR filters).
 
 See [example/](example/) for usage. Apache 2.0 licensed.

auth/auth_test.go

@@ -41,13 +41,13 @@ func TestSetMetadataURL(t *testing.T) {
 
 func TestAccessTokenFromMetadata(t *testing.T) {
 	tests := []struct {
-		name           string
-		statusCode     int
 		response       any
-		wantErr        bool
+		name           string
 		wantToken      string
 		errContains    string
 		metadataFlavor string
+		statusCode     int
+		wantErr        bool
 	}{
 		{
 			name:       "success",
@@ -214,12 +214,12 @@ func TestAccessTokenFromADC(t *testing.T) {
 	defer tokenServer.Close()
 
 	tests := []struct {
+		setupEnv    func()
 		name        string
 		credsData   string
-		setupEnv    func()
-		wantErr     bool
 		errContains string
 		wantToken   string
+		wantErr     bool
 	}{
 		{
 			name: "success with valid credentials",
@@ -310,11 +310,11 @@ func TestAccessTokenFromADC(t *testing.T) {
 func TestProjectID(t *testing.T) {
 	tests := []struct {
 		name        string
-		statusCode  int
 		response    string
-		wantErr     bool
 		wantProject string
 		errContains string
+		statusCode  int
+		wantErr     bool
 	}{
 		{
 			name:        "success",
@@ -414,10 +414,10 @@ func TestProjectIDMetadataServerDown(t *testing.T) {
 func TestExchangeRefreshTokenErrors(t *testing.T) {
 	tests := []struct {
 		name        string
-		statusCode  int
 		response    string
-		wantErr     bool
 		errContains string
+		statusCode  int
+		wantErr     bool
 	}{
 		{
 			name:        "unauthorized",