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 FakeTabledataResourceApi implements TabledataResourceApi {
  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 Map<String, Object> 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
    });
  }
}