fix: Respect Multiplexer No Clients (#313)

* Update execution.go * Respond to comments * simplify * Update provider.go * Update execution.go * Update execution.go * Update execution.go * Update execution.go * Add Low Level Copy support * Create mock_storage.go * Update storage_test.go * Update mocks_test.go * Fixed code gen * Update storage.go * break out * Update storage.go * Update mock_storage.go * comments * Update storage.go * add copier to storage * Update storage.go * Update execution.go * Update execution_test.go * Update execution.go * Update execution_test.go * Update execution.go * Update execution_test.go
cloudquery · Jun 7, 2022 · c873426 · c873426
1 parent 67d776f
commit c873426
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/provider/execution/execution.go b/provider/execution/execution.go
@@ -79,12 +79,11 @@ func NewTableExecutor(resourceName string, db Storage, logger hclog.Logger, tabl
 func (e TableExecutor) Resolve(ctx context.Context, meta schema.ClientMeta) (uint64, diag.Diagnostics) {
 	var clients []schema.ClientMeta
 
+	clients = append(clients, meta)
+
 	if e.Table.Multiplex != nil {
 		clients = e.Table.Multiplex(meta)
 	}
-	if len(clients) == 0 {
-		clients = append(clients, meta)
-	}
 
 	return e.doMultiplexResolve(ctx, clients)
 }

diff --git a/provider/execution/execution_test.go b/provider/execution/execution_test.go
@@ -194,7 +194,7 @@ func TestTableExecutor_Resolve(t *testing.T) {
 				Resolver:  returnValueResolver,
 				Columns:   commonColumns,
 			},
-			ExpectedResourceCount: 1,
+			ExpectedResourceCount: 0,
 		},
 		{
 			// if tables don't define a resolver, an execution error by execution