def get_and_import_my_sheet(access_token, user_email, file_id):
url = graph_endpoint.format(
'/me/drive/items/{}/workbook/worksheets/Sheet1/UsedRange'.format(
file_id))
r = make_api_call('GET', url, access_token, user_email)
if (r.status_code == requests.codes.ok):
from_data_array(r.json()['formulas'])
return r.json()['formulas']
else:
return "{0}: {1}".format(r.status_code, r.text)
def test_importer_size_money_milestone(self):
data = [
[u'id', u'__history_date', u'Name', u'Budget'],
['', '', 'TEXT', 'NUM'],
[u'1', '2012-03-16T17:41:28+00:00', 'foo', '4'],
[u'm;1/6/2015', '2013-03-16T17:41:28+00:00', u'', u'5'],
[u'm;1/7/2016', '2014-03-16T17:41:28+00:00', u'', u'9']
]
result = from_data_array(data)
milestones = Project.objects.get(id=1).milestones.all()
milestone_0 = milestones[0].history.all()[0]
milestone_1 = milestones[1].history.all()[0]
m_dimensions_0 = milestone_0.dimensions.all()
m_dimensions_1 = milestone_1.dimensions.all()
self.assertResultTrue(result['result'])
self.assertEqual(2, milestones.count())
self.assertEqual(datetime(2015, 6, 1, tzinfo=pytz.utc), milestone_0.due_date)
self.assertEqual(datetime(2016, 7, 1, tzinfo=pytz.utc), milestone_1.due_date)
self.assertEqual(1, milestones[0].history.all().count())
self.assertEqual(parse('2013-03-16T17:41:28+00:00'), milestone_0.history_date)
self.assertEqual(5, m_dimensions_0[0].dimension_milestone_object.value)
self.assertEqual(9, m_dimensions_1[0].dimension_milestone_object.value)
self.assertEquals(
m_dimensions_1[0].project_dimension,
Project.objects.get(id=1).dimensions.all()[1]
)
def test_import_dependencies(self):
data = [
[u'id', u'__history_date', u'Name', u'dependencies'],
['', '', 'TEXT', 'APROJ'],
[u'1', '2013-03-16T17:41:28+00:00', 'Project1'],
[u'2', '2013-03-18T17:41:28+00:00', 'Project2', '1'],
[u'3', '2013-03-19T17:41:28+00:00', 'Project3', '1, 2'],
]
result = from_data_array(data)
dimensions_1 = Project.objects.get(id=1).dimensions.all()
dimensions_2 = Project.objects.get(id=2).dimensions.all()
dimensions_3 = Project.objects.get(id=3).dimensions.all()
dependencies_2 = dimensions_2[1].dimension_object.value.all()
dependencies_3 = dimensions_3[1].dimension_object.value.all()
self.assertResultTrue(result['result'])
self.assertEqual(3, Project.objects.all().count())
self.assertEquals(1, dimensions_1.count())
self.assertEquals(1, dependencies_2.count())
self.assertEquals(Project.objects.get(pk=1), dependencies_2[0])
self.assertEquals(2, dependencies_3.count())
self.assertEquals(Project.objects.get(pk=1), dependencies_3[0])
self.assertEquals(Project.objects.get(pk=2), dependencies_3[1])
def test_keep_unaffected_projects(self):
project = Project()
project.id = 1
project.save()
d1 = NumberDimension()
d1.name = 'Budget'
d1.value = 5
d1.save()
pd1 = ProjectDimension()
pd1.project = project
pd1.dimension_object = d1
pd1.save()
data = [
[u'id', u'__history_date', u'Name'],
['', '', 'TEXT'],
[u'2', '2013-03-16T17:41:28+00:00', 'foo']
]
result = from_data_array(data)
dimensions = Project.objects.get(id=1).dimensions.all()
self.assertResultTrue(result['result'])
self.assertEqual(2, Project.objects.all().count())
self.assertEqual(1, dimensions.count())
self.assertTrue(isinstance(dimensions[0].dimension_object, NumberDimension))
def test_import_dimension_name(self):
data = [
[u'id', u'__history_date', u'Name '],
['', '', 'TEXT'],
[u'1', '2013-03-16T17:41:28+00:00', 'foo'],
]
result = from_data_array(data)
self.assertResultTrue(result['result'])
self.assertEqual(
'Name',
Project.objects.get(id=1).dimensions.all()[0].dimension_object.name
)
def test_importer_unknown_column(self):
data = [
[u'id', u'__history_date', u'UnknownDimensionType'],
['', '', 'FOO'],
[u'2', '2013-03-16T17:41:28+00:00', 'foo']
]
result = from_data_array(data)
# current behaviour in code is that importing returns fail
# if there are unknown type names in the type row.
self.assertFalse(
result['result'],
'Failed to stop at faulty dimensiontype'
)
def test_import_dmy_startdate_dmy_history_date(self):
data = [
[u'id', u'__history_date', u'StartDate'],
['', '', 'DATE'],
[u'1', '03/18/2013', '5/6/2017'],
[u'1', '03/16/2013', '3/4/2015']
]
result = from_data_array(data)
dimensions = Project.objects.get(id=1).dimensions.all()
history = dimensions[0].dimension_object.history.all()
self.assertResultTrue(result['result'])
self.assertEqual(2, history.count())
self.assertEqual(make_aware(datetime(2017, 6, 5)), history[0].value)
self.assertEqual(make_aware(datetime(2015, 4, 3)), history[1].value)
def test_json_unicode_encoding(self):
# Tests that non-ascii characters are importer successfully
data = [[u'id', u'__history_date', u'Name',
u'ProjectManagerDimension',
u'dependenciesDimension',
u'Members'],
['', '', 'TEXT', 'APER', 'APROJ', 'APERS'],
[u'1', '2012-03-16T17:41:28+00:00', u'Biz', u'Urho', u'', u'Matti, Pekka'],
[u'2', '2012-04-16T17:41:28+00:00', u'Böz', u'Yrjö', u'1', u'Päivö, Väinämö']]
result = from_data_array(data)
self.assertTrue(result['result'], 'Failed to validate field types in imported data')
json_response = views.json(None)
self.assertIsInstance(json_response, JsonResponse)
def test_import_members(self):
data = [
[u'id', u'__history_date', u'Members'],
['', '', 'APERS'],
[u'1', '2013-03-16T17:41:28+00:00', 'Pekka ,Matti'],
[u'1', '2013-03-18T17:41:28+00:00', 'Matti'],
[u'1', '2013-03-19T17:41:28+00:00', 'Taneli ,Pekka '],
]
result = from_data_array(data)
dimensions = Project.objects.get(id=1).dimensions.all()
members = dimensions[0].dimension_object.value.all()
self.assertResultTrue(result['result'])
self.assertEqual(1, Project.objects.all().count())
self.assertEqual(3, Person.objects.all().count())
self.assertEqual(Person.objects.get(first_name='Taneli'), members[1])
self.assertEqual(Person.objects.get(first_name='Pekka'), members[0])
def test_import_size_money(self):
data = [
[u'id', u'__history_date', u'Name', u'Budget'],
['', '', 'TEXT', 'NUM'],
[u'2', '7/12/2015', 'foo', '100'],
[u'2', '14/12/2015', u'', '30'],
[u'2', '6/1/2016', u'','4'],
[u'2', '8/2/2016', u'','1251'],
[u'2', '14/3/2015', u'','325']
]
result = from_data_array(data)
dimensions = Project.objects.get(id=2).dimensions.all()
self.assertResultTrue(result['result'])
self.assertEqual(
5,
dimensions[1].dimension_object.history.all().count()
)
def test_delete_organization(self):
data = [[
u'id', u'__history_date', u'Name', u'Budget', u'OwningOrganization'
], ['', '', 'TEXT', 'NUM', 'AORG'],
[u'1', '2012-03-16T17:41:28+00:00', 'foo', u'4', 'org1'],
[u'm;28/6/2015', '2013-03-16T17:41:28+00:00', u'', u'5'],
[u'm;29/6/2016', '2014-03-16T17:41:28+00:00', u'', u'9']]
result = from_data_array(data)
self.assertTrue(result['result'],
'Failed to validate field types in imported data')
Organization.objects.get(pk=1).delete()
self.assertFalse(Organization.objects.all())
self.assertFalse(Project.objects.all())
self.assertFalse(ProjectDimension.objects.all())
self.assertFalse(Milestone.objects.all())
self.assertFalse(DimensionMilestone.objects.all())
self.assertFalse(NumberMilestone.objects.all())
def test_import_projectmanager(self):
data = [
[u'id', u'__history_date', u'ProjectManager'],
['', '', 'APER'],
[u'1', '2013-03-16T17:41:28+00:00', 'Pekka '],
[u'1', '2013-03-18T17:41:28+00:00', 'Matti'],
[u'1', '2013-03-19T17:41:28+00:00', 'Pekka'],
]
result = from_data_array(data)
dimensions = Project.objects.get(id=1).dimensions.all()
history = dimensions[0].dimension_object.history.all()
self.assertResultTrue(result['result'])
self.assertEqual(1, Project.objects.all().count())
self.assertEqual(3, history.count())
self.assertEqual(2, Person.objects.all().count())
self.assertEqual(Person.objects.get(first_name='Pekka'), history[0].value)
self.assertEqual(Person.objects.get(first_name='Matti'), history[1].value)
self.assertEqual(Person.objects.get(first_name='Pekka'), history[2].value)
def test_import_startdate(self):
data = [
[u'id', u'__history_date', u'StartDate'],
['', '', 'DATE'],
[u'1', '2013-03-16T17:41:28+00:00', '2013-05-16T17:41:28+00:00'],
[u'1', '2013-03-18T17:41:28+00:00', '2013-07-16T17:41:28+00:00'],
[u'1', '2013-03-15T17:41:28+00:00', '1/8/2012'],
[u'1', '2013-03-14T17:41:28+00:00', '2015-01-08T00:00:00+00:00'],
]
result = from_data_array(data)
dimensions = Project.objects.get(id=1).dimensions.all()
history = dimensions[0].dimension_object.history.all()
self.assertResultTrue(result['result'])
self.assertEqual(1, Project.objects.all().count())
self.assertEqual(4, history.count())
self.assertEqual(parse('2013-07-16T17:41:28+00:00'), history[0].value)
self.assertEqual(parse('2013-05-16T17:41:28+00:00'), history[1].value)
self.assertEqual(make_aware(datetime(2012, 8, 1)), history[2].value)
self.assertEqual(datetime(2015, 8, 1, tzinfo=pytz.utc), history[3].value)
def test_dayfirst_in_history_date(self):
data = [
[u'id', u'__history_date', u'Name '],
['', '', 'TEXT'],
[u'1', '1/8/2015', 'foo'],
[u'1', '2014-03-09T00:00:00+00:00', 'bar'],
]
result = from_data_array(data)
name = Project.objects.get(pk=1).dimensions.all()[0]
actual_history_date_0 = name.dimension_object.history.all()[0].history_date
actual_history_date_1 = name.dimension_object.history.all()[1].history_date
self.assertResultTrue(result['result'])
self.assertEqual(
datetime(2015, 8, 1, tzinfo=pytz.utc),
actual_history_date_0
)
self.assertEqual(
datetime(2014, 9, 3, tzinfo=pytz.utc),
actual_history_date_1
)
def test_milestones_serializer(self):
data = [[u'id', u'__history_date', u'Name', u'Budget'],
['', '', 'TEXT', 'NUM'],
[u'1', '2012-03-16T17:41:28+00:00', 'foo', u'4'],
[u'm;28/6/2015', '2013-03-16T17:41:28+00:00', u'', u'5'],
[u'm;29/6/2016', '2014-03-16T17:41:28+00:00', u'', u'9']]
result = from_data_array(data)
self.assertTrue(result['result'],
'Failed to validate field types in imported data')
milestones1_serializer = MilestoneHistorySerializer(
Project.objects.get(id=1).milestones.all()[0].history.all()[0])
milestones2_serializer = MilestoneHistorySerializer(
Project.objects.get(id=1).milestones.all()[1].history.all()[0])
expected_serialization = [{
'dimensions': [{
'dimension_milestone_object': {
'value': '5.00'
},
'project_dimension': 2
}],
'due_date':
'2015-06-28T00:00:00Z'
}, {
'dimensions': [{
'dimension_milestone_object': {
'value': '9.00'
},
'project_dimension': 2
}],
'due_date':
'2016-06-29T00:00:00Z'
}]
self.assertEquals(
expected_serialization,
[milestones1_serializer.data, milestones2_serializer.data])
def test_import_name(self):
data = [
[u'id', u'__history_date', u'Name'],
['', '', 'TEXT'],
[u'1', '2013-03-16T17:41:28+00:00', 'foo'],
[u'1', '2013-03-17T17:41:28+00:00'],
[u'1', '2013-03-18T17:41:28+00:00', 'baz'],
[u'2', '2013-03-20T17:41:28+00:00', 'dir '],
[u'2', '2013-03-19T17:41:28+00:00', 'biz']
]
result = from_data_array(data)
dimensions_1 = Project.objects.get(id=1).dimensions.all()
dimensions_2 = Project.objects.get(id=2).dimensions.all()
history_1 = dimensions_1[0].dimension_object.history.all()
history_2 = dimensions_2[0].dimension_object.history.all()
self.assertResultTrue(result['result'])
self.assertEqual(2, Project.objects.all().count())
self.assertEqual(2, history_1.count())
self.assertEqual(2, history_2.count())
self.assertEqual('dir', history_2[0].value)
self.assertEqual('biz', history_2[1].value)
self.assertEqual('baz', Project.objects.get(id=1).name)
def test_import_owningorganization(self):
data = [
[u'id', u'__history_date' , u'OwningOrganization', u'Name', u'Budget'],
['' , '' , 'AORG' , 'TEXT' , 'NUM' ],
[u'1' , '2013-03-16T17:41:28+00:00', 'Org1' , 'boo' , '4' ],
[u'1' , '2013-03-18T17:41:28+00:00', 'Org2' , 'biz' ],
[u'1' , '2013-03-19T17:41:28+00:00', 'Org1' ],
]
result = from_data_array(data)
org1_templates = Organization.objects.get(name='Org1').templates.all()
org2_templates = Organization.objects.get(name='Org2').templates.all()
org1_dimensions = org1_templates[0].dimensions.all()
org2_dimensions = org2_templates[0].dimensions.all()
self.assertResultTrue(result['result'])
self.assertEqual(1, org1_templates.count())
self.assertEqual(1, org2_templates.count())
for i in range(5):
print("i={}; name={}".format(i, org1_dimensions[i].name))
self.assertEqual('default', org1_templates[0].name)
# Budget, EndDate, ProjectManager, OwningOrganization, Name
self.assertEqual(5, org1_dimensions.count())
self.assertEqual(5, org2_dimensions.count())
self.assertEqual('Budget', org1_dimensions[0].name)
self.assertEqual(
ContentType.objects.get(model='numberdimension'),
org1_dimensions[0].content_type
)
self.assertEqual('EndDate', org1_dimensions[1].name)
self.assertEqual(
ContentType.objects.get(model='datedimension'),
org1_dimensions[1].content_type
)
self.assertEqual('ProjectManager', org1_dimensions[2].name)
self.assertEqual(
ContentType.objects.get(model='associatedpersondimension'),
org1_dimensions[2].content_type
)
self.assertEqual('OwningOrganization', org1_dimensions[3].name)
self.assertEqual(
ContentType.objects.get(model='associatedorganizationdimension'),
org1_dimensions[3].content_type
)
self.assertEqual('Name', org1_dimensions[4].name)
self.assertEqual(
ContentType.objects.get(model='textdimension'),
org1_dimensions[4].content_type
)
self.assertEqual('Budget', org2_dimensions[0].name)
self.assertEqual(
ContentType.objects.get(model='numberdimension'),
org2_dimensions[0].content_type
)
self.assertEqual('EndDate', org2_dimensions[1].name)
self.assertEqual(
ContentType.objects.get(model='datedimension'),
org2_dimensions[1].content_type
)
self.assertEqual('ProjectManager', org2_dimensions[2].name)
self.assertEqual(
ContentType.objects.get(model='associatedpersondimension'),
org2_dimensions[2].content_type
)
self.assertEqual('OwningOrganization', org2_dimensions[3].name)
self.assertEqual(
ContentType.objects.get(model='associatedorganizationdimension'),
org2_dimensions[3].content_type
)
self.assertEqual('Name', org2_dimensions[4].name)
self.assertEqual(
ContentType.objects.get(model='textdimension'),
org2_dimensions[4].content_type
)
def test_project_serializer(self):
data = [[u'id', u'__history_date', u'Name', u'Budget'],
['', '', 'TEXT', 'NUM'],
[u'1', '2012-03-16T17:41:28+00:00', 'foo', u'4'],
[u'm;28/6/2015', '2013-03-16T17:41:28+00:00', u'', u'5'],
[u'm;29/6/2016', '2014-03-16T17:41:28+00:00', u'', u'9'],
[u'2', '2017-16-03T17:41:28+00:00', 'biz', u'4']]
result = from_data_array(data)
self.assertTrue(result['result'],
'Failed to validate field types in imported data')
projects_serializer = ProjectSerializer(Project.objects.all(),
many=True)
expected_serialization = [{
'id':
1,
'name':
'foo',
'dimensions': [{
'id': 1,
'dimension_object': {
'history': [{
'value': 'foo',
'id': 1,
'history_date': '2012-03-16T17:41:28Z',
'string': 'foo'
}],
'name':
'Name'
},
'dimension_type': 'TextDimension'
}, {
'id': 2,
'dimension_object': {
'history': [{
'value': 4.0,
'id': 1,
'history_date': '2012-03-16T17:41:28Z',
'string': '4.00'
}],
'name':
'Budget'
},
'dimension_type': 'NumberDimension'
}],
'milestones': [{
'due_date':
'2015-06-28T00:00:00Z',
'dimensions': [{
'dimension_milestone_object': {
'value': '5.00'
},
'project_dimension': 2
}]
}, {
'due_date':
'2016-06-29T00:00:00Z',
'dimensions': [{
'dimension_milestone_object': {
'value': '9.00'
},
'project_dimension': 2
}]
}]
}, {
'id':
2,
'name':
'biz',
'dimensions': [{
'id': 3,
'dimension_object': {
'history': [{
'value': 'biz',
'id': 2,
'history_date': '2017-03-16T17:41:28Z',
'string': 'biz'
}],
'name':
'Name'
},
'dimension_type': 'TextDimension'
}, {
'id': 4,
'dimension_object': {
'history': [{
'value': 4.0,
'id': 2,
'history_date': '2017-03-16T17:41:28Z',
'string': '4.00'
}],
'name':
'Budget'
},
'dimension_type': 'NumberDimension'
}],
'milestones': []
}]
self.assertEquals(expected_serialization, projects_serializer.data)