/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */ import { QueryObject , SqlaFormData } from '@superset-ui/core' ; import { rollingWindowOperator } from '../../../src' ; const formData = { metrics : [ 'count(*)' , { label : 'sum(val)' , expressionType : 'SQL' , sqlExpression : 'sum(val)' } , ] , time_range : '2015 : 2016' , granularity : 'month' , datasource : 'foo' , viz_type : 'table' , } ; const queryObject = { metrics : [ 'count(*)' , { label : 'sum(val)' , expressionType : 'SQL' , sqlExpression : 'sum(val)' } , ] , time_range : '2015 : 2016' , granularity : 'month' , post_processing : [ { operation : 'pivot' , options : { index : [ '__timestamp' ] , columns : [ 'nation' ] , aggregates : { 'count(*)' : { operator : 'sum' , } , } , } , } , ] , } ; test ( 'skip transformation' , ( ) => { expect ( rollingWindowOperator ( formData , queryObject ) ) . toEqual ( undefined ) ; expect ( rollingWindowOperator ( { ... formData , rolling_type : 'None' } , queryObject ) ) . toEqual ( undefined ) ; expect ( rollingWindowOperator ( { ... formData , rolling_type : 'foobar' } , queryObject ) ) . toEqual ( undefined ) ; const formDataWithoutMetrics = { ... formData } ; delete formDataWithoutMetrics . metrics ; expect ( rollingWindowOperator ( formDataWithoutMetrics , queryObject ) ) . toEqual ( undefined ) ; } ) ; test ( 'rolling_type: cumsum' , ( ) => { expect ( rollingWindowOperator ( { ... formData , rolling_type : 'cumsum' } , queryObject ) ) . toEqual ( { operation : 'cum' , options : { operator : 'sum' , columns : { 'count(*)' : 'count(*)' , 'sum(val)' : 'sum(val)' , } , is_pivot_df : true , } , } ) ; } ) ; test ( 'rolling_type: sum/mean/std' , ( ) => { const rollingTypes = [ 'sum' , 'mean' , 'std' ] ; rollingTypes . forEach ( rollingType => { expect ( rollingWindowOperator ( { ... formData , rolling_type : rollingType } , queryObject ) ) . toEqual ( { operation : 'rolling' , options : { rolling_type : rollingType , window : 1 , min_periods : 0 , columns : { 'count(*)' : 'count(*)' , 'sum(val)' : 'sum(val)' , } , is_pivot_df : true , } , } ) ; } ) ; } ) ; test ( 'rolling window and "actual values" in the time compare' , ( ) => { expect ( rollingWindowOperator ( { ... formData , rolling_type : 'cumsum' , comparison_type : 'values' , time_compare : [ '1 year ago' , '1 year later' ] , } , queryObject ) ) . toEqual ( { operation : 'cum' , options : { operator : 'sum' , columns : { 'count(*)' : 'count(*)' , 'count(*)__1 year ago' : 'count(*)__1 year ago' , 'count(*)__1 year later' : 'count(*)__1 year later' , 'sum(val)' : 'sum(val)' , 'sum(val)__1 year ago' : 'sum(val)__1 year ago' , 'sum(val)__1 year later' : 'sum(val)__1 year later' , } , is_pivot_df : true , } , } ) ; } ) ; test ( 'rolling window and "difference / percentage / ratio" in the time compare' , ( ) => { const comparisionTypes = [ 'difference' , 'percentage' , 'ratio' ] ; comparisionTypes . forEach ( cType => { expect ( rollingWindowOperator ( { ... formData , rolling_type : 'cumsum' , comparison_type : cType , time_compare : [ '1 year ago' , '1 year later' ] , } , queryObject ) ) . toEqual ( { operation : 'cum' , options : { operator : 'sum' , columns : { [ `${ cType }__count(*)__count(*)__1 year ago` ] : `${ cType }__count(*)__count(*)__1 year ago` , [ `${ cType }__count(*)__count(*)__1 year later` ] : `${ cType }__count(*)__count(*)__1 year later` , [ `${ cType }__sum(val)__sum(val)__1 year ago` ] : `${ cType }__sum(val)__sum(val)__1 year ago` , [ `${ cType }__sum(val)__sum(val)__1 year later` ] : `${ cType }__sum(val)__sum(val)__1 year later` , } , is_pivot_df : true , } , } ) ; } ) ; } ) ;