func()

in azkustoingest/mock.go [46:93]

• 45 lines of code
• 5 McCabe index (conditional complexity)

func (m mockClient) Mgmt(ctx context.Context, db string, query azkustodata.Statement, options ...azkustodata.QueryOption) (v1.Dataset, error) {
	if m.onMgmt != nil {
		rows, err := m.onMgmt(ctx, db, query, options...)
		if err != nil || rows != nil {
			return rows, err
		}
	}

	if query.String() == ".get kusto identity token" {
		return v1.NewDataset(ctx, errors.OpMgmt, v1.V1{
			Tables: []v1.RawTable{
				{
					TableName: "Table",
					Columns: []v1.RawColumn{
						{
							ColumnName: "AuthorizationContext",
							ColumnType: string(types.String),
						},
					},
					Rows: []v1.RawRow{
						{
							Row:    []interface{}{"mock"},
							Errors: nil,
						},
					},
				},
			}})
	}

	return v1.NewDataset(ctx, errors.OpMgmt, v1.V1{
		Tables: []v1.RawTable{
			{
				TableName: "Table",
				Columns: []v1.RawColumn{
					{
						ColumnName: "ResourceTypeName",
						ColumnType: string(types.String),
					},
					{
						ColumnName: "StorageRoot",
						ColumnType: string(types.String),
					},
				},
				Rows: []v1.RawRow{},
			},
		}})

}