app_dart/test/src/bigquery/fake_tabledata_resource.dart

// Copyright 2019 The Flutter Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

import 'package:googleapis/bigquery/v2.dart';

/// A fake bigquery tabledataresourceApi implementation.
///
/// This tabledataResourceApi considers only a simple case
/// where we focus on the number of rows inserted. This can be
/// easily extended for other test cases.
class FakeTabledataResource implements TabledataResource {
  List<TableDataInsertAllRequestRows>? rows;
  @override
  Future<TableDataInsertAllResponse> insertAll(
      TableDataInsertAllRequest request, String projectId, String datasetId, String tableId,
      {String? $fields}) async {
    rows = request.rows;
    return TableDataInsertAllResponse.fromJson(<String, String>{});
  }

  @override
  Future<TableDataList> list(String projectId, String datasetId, String tableId,
      {int? maxResults, String? selectedFields, String? startIndex, String? pageToken, String? $fields}) async {
    final List<Map<String, Object>> tableRowList = <Map<String, Object>>[];
    for (TableDataInsertAllRequestRows tableDataInsertAllRequestRows in rows!) {
      final dynamic value = tableDataInsertAllRequestRows.json;
      final List<Map<String, Object?>> tableCellList = <Map<String, Object?>>[];
      tableCellList.add(<String, Object?>{'v': value});
      tableRowList.add(<String, Object>{'f': tableCellList});
    }

    return TableDataList.fromJson(<String, Object>{'totalRows': rows!.length.toString(), 'rows': tableRowList});
  }
}