def test_paste_to_for_cut_different_sheet(self):
self.user = User()
self.user.save()
c2 = Cell()
c2.formula = 'foo'
c2.formatted_value = 'fv'
d2 = Cell()
d2.formula = 'fv to become formula'
d2.formatted_value = d2.formula
c3 = Cell()
d3 = Cell()
c5 = Cell()
c5.formula = 'a safe source cell'
dest_cells = {
(3, 4): c2,
(3, 5): c3,
(4, 4): d2,
(4, 5): d3,
}
source_sheet = Sheet()
source_sheet.owner = self.user
source_sheet.save()
clipboard = Clipboard()
clipboard.to_cells = Mock()
clipboard.to_cells.return_value = dest_cells.iteritems()
clipboard.is_cut = True
clipboard.source_sheet = source_sheet
clipboard.source_left = 3
clipboard.source_top = 2
clipboard.source_right = 4
clipboard.source_bottom = 3
destination_sheet = Sheet()
destination_sheet.owner = self.user
destination_worksheet = Worksheet()
destination_worksheet.C4.formula = 'i am in danger!'
destination_worksheet.A1.formula = 'i am safe :-)'
destination_sheet.jsonify_worksheet(destination_worksheet)
destination_sheet.save()
clipboard.paste_to(destination_sheet, (3, 4), (3, 4))
updated_worksheet = destination_sheet.unjsonify_worksheet()
self.assertEquals(updated_worksheet.A1.formula, 'i am safe :-)')
self.assertEquals(updated_worksheet.C4.formula, c2.formula)
self.assertEquals(updated_worksheet.D4.formula, d2.formula)
self.assertEquals(updated_worksheet.C5.formula, c3.formula)
self.assertEquals(updated_worksheet.d5.formula, d3.formula)
# paste should reset the clipboard so that future pastes act as
# though they came from a copy
self.assertEquals(clipboard.source_left, 3)
self.assertEquals(clipboard.source_top, 4)
self.assertEquals(clipboard.source_right, 4)
self.assertEquals(clipboard.source_bottom, 5)
self.assertEquals(clipboard.is_cut, False)
self.assertEquals(clipboard.source_sheet, None)
def test_cut_calls_copy_then_cuts_and_remembers_some_stuff(self):
clipboard = Clipboard()
sheet = Sheet()
user = User()
user.username = '******'
user.save()
sheet.owner = user
clipboard.copy = Mock(wraps=clipboard.copy)
worksheet = Worksheet()
worksheet.A1.formula = 'outside'
worksheet.A2.formula = 'inside'
sheet.jsonify_worksheet(worksheet)
sheet.save()
clipboard.cut(sheet, (1, 2), (3, 4))
self.assertCalledOnce(clipboard.copy, sheet, (1, 2), (3, 4))
worksheet = sheet.unjsonify_worksheet()
self.assertEquals(worksheet.A1.formula, 'outside')
self.assertEquals(worksheet.A2, Cell())
self.assertEquals(clipboard.source_left, 1)
self.assertEquals(clipboard.source_top, 2)
self.assertEquals(clipboard.source_right, 3)
self.assertEquals(clipboard.source_bottom, 4)
self.assertEquals(clipboard.is_cut, True)
self.assertEquals(clipboard.source_sheet, sheet)
def test_unicode(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='the new sheet name')
sheet.save()
self.assertEquals(unicode(sheet),
'Sheet %d: %s' % (sheet.id, sheet.name))
def test_sheet_name_not_set_on_save_if_name_is_not_default(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
sheet.name = 'new sheet name'
sheet.save()
self.assertEquals(sheet.name, 'new sheet name')
def test_paste_to_for_cut_different_sheet(self):
self.user = User()
self.user.save()
c2 = Cell()
c2.formula = 'foo'
c2.formatted_value = 'fv'
d2 = Cell()
d2.formula = 'fv to become formula'
d2.formatted_value = d2.formula
c3 = Cell()
d3 = Cell()
c5 = Cell()
c5.formula = 'a safe source cell'
dest_cells = {
(3, 4): c2,
(3, 5): c3,
(4, 4): d2,
(4, 5): d3,
}
source_sheet = Sheet()
source_sheet.owner = self.user
source_sheet.save()
clipboard = Clipboard()
clipboard.to_cells = Mock()
clipboard.to_cells.return_value = dest_cells.iteritems()
clipboard.is_cut = True
clipboard.source_sheet = source_sheet
clipboard.source_left = 3
clipboard.source_top = 2
clipboard.source_right = 4
clipboard.source_bottom = 3
destination_sheet = Sheet()
destination_sheet.owner = self.user
destination_worksheet = Worksheet()
destination_worksheet.C4.formula = 'i am in danger!'
destination_worksheet.A1.formula = 'i am safe :-)'
destination_sheet.jsonify_worksheet(destination_worksheet)
destination_sheet.save()
clipboard.paste_to(destination_sheet, (3, 4), (3, 4))
updated_worksheet = destination_sheet.unjsonify_worksheet()
self.assertEquals(updated_worksheet.A1.formula, 'i am safe :-)')
self.assertEquals(updated_worksheet.C4.formula, c2.formula)
self.assertEquals(updated_worksheet.D4.formula, d2.formula)
self.assertEquals(updated_worksheet.C5.formula, c3.formula)
self.assertEquals(updated_worksheet.d5.formula, d3.formula)
# paste should reset the clipboard so that future pastes act as
# though they came from a copy
self.assertEquals(clipboard.source_left, 3)
self.assertEquals(clipboard.source_top, 4)
self.assertEquals(clipboard.source_right, 4)
self.assertEquals(clipboard.source_bottom, 5)
self.assertEquals(clipboard.is_cut, False)
self.assertEquals(clipboard.source_sheet, None)
def test_cut_calls_copy_then_cuts_and_remembers_some_stuff(self):
clipboard = Clipboard()
sheet = Sheet()
user = User()
user.username = '******'
user.save()
sheet.owner = user
clipboard.copy = Mock(wraps=clipboard.copy)
worksheet = Worksheet()
worksheet.A1.formula = 'outside'
worksheet.A2.formula = 'inside'
sheet.jsonify_worksheet(worksheet)
sheet.save()
clipboard.cut(sheet, (1, 2), (3, 4))
self.assertCalledOnce(clipboard.copy, sheet, (1, 2), (3, 4))
worksheet = sheet.unjsonify_worksheet()
self.assertEquals(worksheet.A1.formula, 'outside')
self.assertEquals(worksheet.A2, Cell())
self.assertEquals(clipboard.source_left, 1)
self.assertEquals(clipboard.source_top, 2)
self.assertEquals(clipboard.source_right, 3)
self.assertEquals(clipboard.source_bottom, 4)
self.assertEquals(clipboard.is_cut, True)
self.assertEquals(clipboard.source_sheet, sheet)
def test_sheet_name_not_set_on_save_if_name_is_not_default(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
sheet.name = 'new sheet name'
sheet.save()
self.assertEquals(sheet.name, 'new sheet name')
def import_xls(request, username):
if request.user.username != username:
return HttpResponseForbidden(render_to_string("403.html"))
handle, filename = mkstemp()
try:
os.write(handle, request.FILES['file'].read())
wb = xlrd.open_workbook(filename)
for xl_sheet in wb.sheets():
if xl_sheet.nrows > 0 and xl_sheet.ncols > 0:
name = '%s - %s' % (
splitext(request.FILES['file'].name)[0],
xl_sheet.name
)
sheet = Sheet(owner=request.user, name=name)
sheet.jsonify_worksheet(worksheet_from_excel(xl_sheet))
sheet.save()
try:
calculate(request, sheet.owner.username, sheet.id)
except:
pass
except Exception:
return render_to_response(
'import_xls_error.html',
{},
context_instance=RequestContext(request)
)
finally:
os.close(handle)
os.unlink(filename)
return HttpResponseRedirect('/')
def test_uuid_stays_constant_between_reads(self):
sheet = Sheet()
user = User(username='******')
user.save()
sheet.owner = user
sheet.save()
sheet2 = Sheet.objects.get(pk=sheet.id)
self.assertEquals(sheet.api_key, sheet2.api_key)
def test_uuid_stays_constant_between_reads(self):
sheet = Sheet()
user = User(username='******')
user.save()
sheet.owner = user
sheet.save()
sheet2 = Sheet.objects.get(pk=sheet.id)
self.assertEquals(sheet.api_key, sheet2.api_key)
def new_sheet(request):
sheet = Sheet(owner=request.user)
sheet.save()
# need response redirect in order to reset url
return HttpResponseRedirect(reverse('sheet_page', kwargs={
'username' : request.user.username,
'sheet_id' : sheet.id
}))
def test_unicode(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
more_info_url = 'http://far.away/'
fs = FeaturedSheet(sheet=sheet, description=description, more_info_url=more_info_url)
fs.save()
self.assertEquals(unicode(fs), u'Feature: %s' % (sheet.name,))
def test_can_construct_without_more_info_url(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
fs = FeaturedSheet(sheet=sheet, description=description)
fs.save()
self.assertEquals(fs.sheet, sheet)
self.assertEquals(fs.description, description)
self.assertEquals(fs.more_info_url, '')
def test_can_construct_without_more_info_url(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
fs = FeaturedSheet(sheet=sheet, description=description)
fs.save()
self.assertEquals(fs.sheet, sheet)
self.assertEquals(fs.description, description)
self.assertEquals(fs.more_info_url, '')
def test_can_construct_with_more_info_url(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
more_info_url = 'http://far.away/'
fs = FeaturedSheet(sheet=sheet, description=description, more_info_url=more_info_url)
fs.save()
self.assertEquals(fs.sheet, sheet)
self.assertEquals(fs.description, description)
self.assertEquals(fs.more_info_url, more_info_url)
def test_unicode(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
more_info_url = 'http://far.away/'
fs = FeaturedSheet(sheet=sheet,
description=description,
more_info_url=more_info_url)
fs.save()
self.assertEquals(unicode(fs), u'Feature: %s' % (sheet.name, ))
def test_can_construct_with_more_info_url(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='sheet to feature')
sheet.save()
description = 'twas brillig and the slithy toves'
more_info_url = 'http://far.away/'
fs = FeaturedSheet(sheet=sheet,
description=description,
more_info_url=more_info_url)
fs.save()
self.assertEquals(fs.sheet, sheet)
self.assertEquals(fs.description, description)
self.assertEquals(fs.more_info_url, more_info_url)
def test_roundtrip_column_widths_to_db(self, mock_jsonlib):
COLUMN_WIDTHS = {'1': 11, '2': 22, '3': 33}
mock_jsonlib.loads.return_value = COLUMN_WIDTHS
mock_jsonlib.dumps.return_value = sentinel.json
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
DEFAULT_COLUMN_WIDTHS_JSON = '{}'
self.assertEquals(mock_jsonlib.loads.call_args,
((DEFAULT_COLUMN_WIDTHS_JSON, ), {}))
sheet.column_widths = COLUMN_WIDTHS
sheet.save()
self.assertEqual(mock_jsonlib.dumps.call_args, ((COLUMN_WIDTHS, ), {}))
pk = sheet.id
sheet2 = Sheet.objects.get(pk=pk)
self.assertEquals(sheet2.column_widths, COLUMN_WIDTHS)
def test_roundtrip_column_widths_to_db(self, mock_jsonlib):
COLUMN_WIDTHS = {'1': 11, '2': 22, '3': 33}
mock_jsonlib.loads.return_value = COLUMN_WIDTHS
mock_jsonlib.dumps.return_value = sentinel.json
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
DEFAULT_COLUMN_WIDTHS_JSON = '{}'
self.assertEquals(
mock_jsonlib.loads.call_args,
((DEFAULT_COLUMN_WIDTHS_JSON,), {})
)
sheet.column_widths = COLUMN_WIDTHS
sheet.save()
self.assertEqual(
mock_jsonlib.dumps.call_args,
((COLUMN_WIDTHS,), {})
)
pk = sheet.id
sheet2 = Sheet.objects.get(pk=pk)
self.assertEquals(sheet2.column_widths, COLUMN_WIDTHS)
def test_copy_sheet_allows_other_users_to_copy_public_sheets(self):
user = User(username='******')
user.save()
worksheet = Worksheet()
worksheet.a1.value = 'some-cell-content'
sheet = Sheet()
sheet.owner = user
sheet.is_public = True
sheet.jsonify_worksheet(worksheet)
sheet.save()
original_sheet_id = sheet.id
other_user = User(username='******')
other_user.save()
retval = copy_sheet_to_user(sheet, other_user)
other_user_sheets = Sheet.objects.filter(owner=other_user)
self.assertEquals(len(other_user_sheets), 1)
copied_sheet = other_user_sheets[0]
self.assertFalse(copied_sheet.is_public)
copied_worksheet = copied_sheet.unjsonify_worksheet()
self.assertEquals(copied_worksheet.a1.value, 'some-cell-content')
self.assertEquals(copied_sheet.id, retval.id)
self.assertNotEquals(retval.id, original_sheet_id)
def test_copy_sheet_allows_other_users_to_copy_public_sheets(self):
user = User(username='******')
user.save()
worksheet = Worksheet()
worksheet.a1.value = 'some-cell-content'
sheet = Sheet()
sheet.owner = user
sheet.is_public = True
sheet.jsonify_worksheet(worksheet)
sheet.save()
original_sheet_id = sheet.id
other_user = User(username='******')
other_user.save()
retval = copy_sheet_to_user(sheet, other_user)
other_user_sheets = Sheet.objects.filter(owner=other_user)
self.assertEquals(len(other_user_sheets), 1)
copied_sheet = other_user_sheets[0]
self.assertFalse(copied_sheet.is_public)
copied_worksheet = copied_sheet.unjsonify_worksheet()
self.assertEquals(copied_worksheet.a1.value, 'some-cell-content')
self.assertEquals(copied_sheet.id, retval.id)
self.assertNotEquals(retval.id, original_sheet_id)
def test_unicode(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user, name='the new sheet name')
sheet.save()
self.assertEquals(unicode(sheet), 'Sheet %d: %s' % (sheet.id, sheet.name))
def test_sheet_name_set_on_save_if_name_is_default(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
sheet.save()
self.assertEquals(sheet.name, 'Sheet %d' % (sheet.id,))
def test_sheet_name_set_on_save_if_name_is_default(self):
user = User(username='******')
user.save()
sheet = Sheet(owner=user)
sheet.save()
self.assertEquals(sheet.name, 'Sheet %d' % (sheet.id, ))
