/*
 * © 2021 Thoughtworks, Inc.
 */

import { EstimationResult, GroupBy } from '@cloud-carbon-footprint/common'
import EmissionsByDayAndServiceTable from '../EmissionsByDayAndServiceTable'
import moment = require('moment')

describe('EmissionsTable', () => {
  const timestampOne = moment.utc('2020-07-10').toDate()
  const timestampTwo = moment.utc('2020-07-09').toDate()
  const testAccountId = 'test account id'
  const testAccountName = 'test account'
  const region = 'us-east-1'

  const input: EstimationResult[] = [
    {
      timestamp: timestampOne,
      periodStartDate: timestampOne,
      periodEndDate: moment(timestampOne).add(1, 'd').toDate(),
      groupBy: GroupBy.day,
      serviceEstimates: [
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'EBS',
          kilowattHours: 1,
          co2e: 1,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'S3',
          kilowattHours: 2,
          co2e: 2,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'EC2',
          kilowattHours: 3,
          co2e: 3,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'ElastiCache',
          kilowattHours: 4,
          co2e: 4,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'RDS',
          kilowattHours: 4,
          co2e: 4,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'Lambda',
          kilowattHours: 1,
          co2e: 1,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'gcp',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'ComputeEngine',
          kilowattHours: 1,
          co2e: 1,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
      ],
    },
    {
      timestamp: timestampTwo,
      periodStartDate: timestampTwo,
      periodEndDate: moment(timestampTwo).add(1, 'd').toDate(),
      groupBy: GroupBy.day,
      serviceEstimates: [
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'EBS',
          kilowattHours: 7,
          co2e: 8,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'S3',
          kilowattHours: 55,
          co2e: 1,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'EC2',
          kilowattHours: 90,
          co2e: 77,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'ElastiCache',
          kilowattHours: 747,
          co2e: 787,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'RDS',
          kilowattHours: 747,
          co2e: 787,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
        {
          cloudProvider: 'aws',
          accountId: testAccountId,
          accountName: testAccountName,
          serviceName: 'Lambda',
          kilowattHours: 300,
          co2e: 300,
          cost: 7,
          region: region,
          usesAverageCPUConstant: false,
        },
      ],
    },
  ]

  let result: { table: string[][]; colWidths: number[] }

  beforeEach(() => {
    result = EmissionsByDayAndServiceTable(input)
  })

  it('prints out the given estimation results grouped by date', () => {
    expect(result.table).toEqual([
      [
        'Date (UTC)',
        'EBS kilowatt hours',
        'EBS metric tons CO2e Emissions',
        'S3 kilowatt hours',
        'S3 metric tons CO2e Emissions',
        'EC2 kilowatt hours',
        'EC2 metric tons CO2e Emissions',
        'ElastiCache kilowatt hours',
        'ElastiCache metric tons CO2e Emissions',
        'RDS kilowatt hours',
        'RDS metric tons CO2e Emissions',
        'Lambda kilowatt hours',
        'Lambda metric tons CO2e Emissions',
        'ComputeEngine kilowatt hours',
        'ComputeEngine metric tons CO2e Emissions',
        'SUM kilowatt hours',
        'SUM metric tons CO2e Emissions',
      ],
      [
        '2020-07-09',
        '7.00',
        '8.000000',
        '55.00',
        '1.000000',
        '90.00',
        '77.000000',
        '747.00',
        '787.000000',
        '747.00',
        '787.000000',
        '300.00',
        '300.000000',
        '0.00',
        '0.000000',
        '1946.00',
        '1960.000000',
      ],
      [
        '2020-07-10',
        '1.00',
        '1.000000',
        '2.00',
        '2.000000',
        '3.00',
        '3.000000',
        '4.00',
        '4.000000',
        '4.00',
        '4.000000',
        '1.00',
        '1.000000',
        '1.00',
        '1.000000',
        '16.00',
        '16.000000',
      ],
      [
        'Total',
        '8.00',
        '9.000000',
        '57.00',
        '3.000000',
        '93.00',
        '80.000000',
        '751.00',
        '791.000000',
        '751.00',
        '791.000000',
        '301.00',
        '301.000000',
        '1.00',
        '1.000000',
        '1962.00',
        '1976.000000',
      ],
    ])
  })

  it('does the right columns', () => {
    expect(result.colWidths).toEqual([
      15, 20, 25, 20, 25, 20, 25, 20, 25, 20, 25, 20, 25, 20, 25, 20, 25,
    ])
  })
})