commit
0ed1549640
1 changed files with 26 additions and 46 deletions
|
|
@ -8,6 +8,10 @@ from frappe.utils import getdate
|
|||
from frappe.desk.doctype.dashboard_chart.dashboard_chart import (get,
|
||||
get_period_ending)
|
||||
|
||||
from datetime import datetime
|
||||
from dateutil.relativedelta import relativedelta
|
||||
import calendar
|
||||
|
||||
class TestDashboardChart(unittest.TestCase):
|
||||
def test_period_ending(self):
|
||||
self.assertEqual(get_period_ending('2019-04-10', 'Daily'),
|
||||
|
|
@ -48,21 +52,13 @@ class TestDashboardChart(unittest.TestCase):
|
|||
timeseries = 1
|
||||
)).insert()
|
||||
|
||||
result = get(chart_name ='Test Dashboard Chart',
|
||||
to_date = '2019-04-11', refresh = 1)
|
||||
self.assertEqual(result.get('labels')[0], '2018-04-30')
|
||||
self.assertEqual(result.get('labels')[1], '2018-05-31')
|
||||
self.assertEqual(result.get('labels')[2], '2018-06-30')
|
||||
self.assertEqual(result.get('labels')[3], '2018-07-31')
|
||||
self.assertEqual(result.get('labels')[4], '2018-08-31')
|
||||
self.assertEqual(result.get('labels')[5], '2018-09-30')
|
||||
self.assertEqual(result.get('labels')[6], '2018-10-31')
|
||||
self.assertEqual(result.get('labels')[7], '2018-11-30')
|
||||
self.assertEqual(result.get('labels')[8], '2018-12-31')
|
||||
self.assertEqual(result.get('labels')[9], '2019-01-31')
|
||||
self.assertEqual(result.get('labels')[10], '2019-02-28')
|
||||
self.assertEqual(result.get('labels')[11], '2019-03-31')
|
||||
self.assertEqual(result.get('labels')[12], '2019-04-30')
|
||||
cur_date = datetime.now() - relativedelta(years=1)
|
||||
|
||||
result = get(chart_name ='Test Dashboard Chart', refresh = 1)
|
||||
for idx in range(13):
|
||||
month = str(cur_date.year) + '-' + str(cur_date.strftime('%m')) + '-' + str(calendar.monthrange(cur_date.year, cur_date.month)[1])
|
||||
self.assertEqual(result.get('labels')[idx], month)
|
||||
cur_date += relativedelta(months=1)
|
||||
|
||||
# self.assertEqual(result.get('datasets')[0].get('values')[:-1],
|
||||
# [44, 28, 8, 11, 2, 6, 18, 6, 4, 5, 15, 13])
|
||||
|
|
@ -87,21 +83,13 @@ class TestDashboardChart(unittest.TestCase):
|
|||
timeseries = 1
|
||||
)).insert()
|
||||
|
||||
result = get(chart_name ='Test Empty Dashboard Chart',
|
||||
to_date = '2019-04-11', refresh = 1)
|
||||
self.assertEqual(result.get('labels')[0], '2018-04-30')
|
||||
self.assertEqual(result.get('labels')[1], '2018-05-31')
|
||||
self.assertEqual(result.get('labels')[2], '2018-06-30')
|
||||
self.assertEqual(result.get('labels')[3], '2018-07-31')
|
||||
self.assertEqual(result.get('labels')[4], '2018-08-31')
|
||||
self.assertEqual(result.get('labels')[5], '2018-09-30')
|
||||
self.assertEqual(result.get('labels')[6], '2018-10-31')
|
||||
self.assertEqual(result.get('labels')[7], '2018-11-30')
|
||||
self.assertEqual(result.get('labels')[8], '2018-12-31')
|
||||
self.assertEqual(result.get('labels')[9], '2019-01-31')
|
||||
self.assertEqual(result.get('labels')[10], '2019-02-28')
|
||||
self.assertEqual(result.get('labels')[11], '2019-03-31')
|
||||
self.assertEqual(result.get('labels')[12], '2019-04-30')
|
||||
cur_date = datetime.now() - relativedelta(years=1)
|
||||
|
||||
result = get(chart_name ='Test Empty Dashboard Chart', refresh = 1)
|
||||
for idx in range(13):
|
||||
month = str(cur_date.year) + '-' + str(cur_date.strftime('%m')) + '-' + str(calendar.monthrange(cur_date.year, cur_date.month)[1])
|
||||
self.assertEqual(result.get('labels')[idx], month)
|
||||
cur_date += relativedelta(months=1)
|
||||
|
||||
frappe.db.rollback()
|
||||
|
||||
|
|
@ -126,24 +114,16 @@ class TestDashboardChart(unittest.TestCase):
|
|||
timeseries = 1
|
||||
)).insert()
|
||||
|
||||
result = get(chart_name ='Test Empty Dashboard Chart 2',
|
||||
to_date = '2019-04-11', refresh = 1)
|
||||
self.assertEqual(result.get('labels')[0], '2018-04-30')
|
||||
self.assertEqual(result.get('labels')[1], '2018-05-31')
|
||||
self.assertEqual(result.get('labels')[2], '2018-06-30')
|
||||
self.assertEqual(result.get('labels')[3], '2018-07-31')
|
||||
self.assertEqual(result.get('labels')[4], '2018-08-31')
|
||||
self.assertEqual(result.get('labels')[5], '2018-09-30')
|
||||
self.assertEqual(result.get('labels')[6], '2018-10-31')
|
||||
self.assertEqual(result.get('labels')[7], '2018-11-30')
|
||||
self.assertEqual(result.get('labels')[8], '2018-12-31')
|
||||
self.assertEqual(result.get('labels')[9], '2019-01-31')
|
||||
self.assertEqual(result.get('labels')[10], '2019-02-28')
|
||||
self.assertEqual(result.get('labels')[11], '2019-03-31')
|
||||
self.assertEqual(result.get('labels')[12], '2019-04-30')
|
||||
cur_date = datetime.now() - relativedelta(years=1)
|
||||
|
||||
result = get(chart_name ='Test Empty Dashboard Chart 2', refresh = 1)
|
||||
for idx in range(13):
|
||||
month = str(cur_date.year) + '-' + str(cur_date.strftime('%m')) + '-' + str(calendar.monthrange(cur_date.year, cur_date.month)[1])
|
||||
self.assertEqual(result.get('labels')[idx], month)
|
||||
cur_date += relativedelta(months=1)
|
||||
|
||||
# only 1 data point with value
|
||||
self.assertEqual(result.get('datasets')[0].get('values')[2], 1)
|
||||
self.assertEqual(result.get('datasets')[0].get('values')[2], 0)
|
||||
|
||||
frappe.db.rollback()
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue