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_then_paste_same_sheet(self): sheet = Sheet() worksheet = Worksheet() worksheet.A1.formula = 'safe' worksheet.A2.formula = 'topleft' worksheet.B2.formula = 'topright' worksheet.A3.formula = 'bottomleft' worksheet.B3.formula = 'bottomright' sheet.jsonify_worksheet(worksheet) clipboard = Clipboard() clipboard.cut(sheet, (1, 2), (2, 3)) updated_worksheet = sheet.unjsonify_worksheet() self.assertEquals(updated_worksheet.A1.formula, 'safe') self.assertEquals(updated_worksheet.A2, Cell() ) self.assertEquals(updated_worksheet.B2, Cell() ) self.assertEquals(updated_worksheet.A3, Cell() ) clipboard.paste_to(sheet, (2, 3), (2, 3)) updated_worksheet = sheet.unjsonify_worksheet() self.assertEquals(updated_worksheet.A1.formula, 'safe') self.assertEquals(updated_worksheet.A2, Cell() ) self.assertEquals(updated_worksheet.B2, Cell() ) self.assertEquals(updated_worksheet.A3, Cell() ) self.assertEquals(updated_worksheet.B3.formula, 'topleft') self.assertEquals(updated_worksheet.C3.formula, 'topright') self.assertEquals(updated_worksheet.C4.formula, 'bottomright') self.assertEquals(updated_worksheet.B4.formula, 'bottomleft')
def test_clipboad_copy_retrieves_stuff_from_sheet_and_removes_offset(self): self.maxDiff = None clipboard = Clipboard() sheet = Sheet() worksheet = Worksheet() worksheet.C2.formula = 'foo' worksheet.C2.formatted_value = 'fv' worksheet.D2.formatted_value = 'fv to become formula' sheet.jsonify_worksheet(worksheet) start = (3, 2) end = (4, 3) cut_called_previously = True clipboard.is_cut = cut_called_previously clipboard.copy(sheet, start, end) self.assertEquals( json.loads(clipboard.contents_json), { '0,0':{'formula': 'foo', 'formatted_value': 'fv'}, '1,0':{'formula': 'fv to become formula', 'formatted_value': 'fv to become formula'}, '0,1':{'formula': '', 'formatted_value': ''}, '1,1':{'formula': '', 'formatted_value': ''} } ) self.assertEquals(clipboard.source_left, 3) self.assertEquals(clipboard.source_top, 2) self.assertEquals(clipboard.source_right, 4) self.assertEquals(clipboard.source_bottom, 3) 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_to_ui_json_meta_data_zero_size(self): sheet = Sheet(width=0, height=0) expected = dict(width=sheet.width, height=sheet.height, name='Untitled') self.assertEquals( json.loads(sheet_to_ui_json_meta_data(sheet, Worksheet())), expected)
def test_to_ui_json_meta_data_ten_by_five_empty(self): sheet = Sheet(width=10, height=5) expected = dict(width=sheet.width, height=sheet.height, name='Untitled') self.assertEquals( json.loads(sheet_to_ui_json_meta_data(sheet, Worksheet())), expected)
def test_to_ui_json_meta_data_includes_worksheet_console_text(self): sheet = Sheet(width=10, height=5) worksheet = Worksheet() worksheet._console_text = ['error1', 'error2'] expected = dict(width=sheet.width, height=sheet.height, name='Untitled', console_text=worksheet._console_text) self.assertEquals( json.loads(sheet_to_ui_json_meta_data(sheet, worksheet)), expected)
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_to_ui_json_meta_data_includes_columns_widths(self): sheet = Sheet(width=10, height=5) sheet.column_widths = {u'1': 1, u'2': 22, u'3': 333} expected = dict( width=sheet.width, height=sheet.height, name='Untitled', column_widths=sheet.column_widths, ) self.assertEquals( json.loads(sheet_to_ui_json_meta_data(sheet, Worksheet())), expected)
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_formulae_are_rewritten(self, mock_rewrite): mock_rewrite.return_value = '=C8' sheet = Sheet() worksheet = Worksheet() worksheet.A1.formula = '=B6' sheet.jsonify_worksheet(worksheet) clipboard = Clipboard() clipboard.copy(sheet, (1, 1), (1, 1)) clipboard.paste_to(sheet, (2, 3), (2, 3)) self.assertCalledOnce( mock_rewrite, '=B6', 1, 2, False, (1, 1, 1, 1) )
def test_to_ui_json_meta_data_includes_usercode_errors(self): sheet = Sheet(width=10, height=5) worksheet = Worksheet() worksheet._usercode_error = {'message': 'ABC', 'line': 123} expected_json_contents = { 'width': sheet.width, 'height': sheet.height, 'name': sheet.name, 'usercode_error': { 'message': 'ABC', 'line': '123' } } self.assertEquals( json.loads(sheet_to_ui_json_meta_data(sheet, worksheet)), expected_json_contents)
def test_paste_to_should_tile_clipboard_contents_across_selected_range(self): sheet = Sheet() worksheet = Worksheet() worksheet.A1.formula = '=c1' worksheet.B1.formula = '=d1' worksheet.A2.formula = '=c2' worksheet.B2.formula = '=d2' sheet.jsonify_worksheet(worksheet) clipboard = Clipboard() clipboard.copy(sheet, (1, 1), (2, 2)) clipboard.paste_to(sheet, (3, 3), (5, 7)) worksheet = sheet.unjsonify_worksheet() self.assertEquals(worksheet.C3.formula, '=E3') self.assertEquals(worksheet.D3.formula, '=F3') self.assertEquals(worksheet.E3.formula, '=G3') self.assertEquals(worksheet.F3.formula, None) self.assertEquals(worksheet.C4.formula, '=E4') self.assertEquals(worksheet.D4.formula, '=F4') self.assertEquals(worksheet.E4.formula, '=G4') self.assertEquals(worksheet.F4.formula, None) self.assertEquals(worksheet.C5.formula, '=E5') self.assertEquals(worksheet.D5.formula, '=F5') self.assertEquals(worksheet.E5.formula, '=G5') self.assertEquals(worksheet.F5.formula, None) self.assertEquals(worksheet.C6.formula, '=E6') self.assertEquals(worksheet.D6.formula, '=F6') self.assertEquals(worksheet.E6.formula, '=G6') self.assertEquals(worksheet.F6.formula, None) self.assertEquals(worksheet.C7.formula, '=E7') self.assertEquals(worksheet.D7.formula, '=F7') self.assertEquals(worksheet.E7.formula, '=G7') self.assertEquals(worksheet.F7.formula, None) self.assertEquals(worksheet.C8.formula, None) self.assertEquals(worksheet.D8.formula, None) self.assertEquals(worksheet.E8.formula, None) self.assertEquals(worksheet.F8.formula, None)
def test_run_worksheet_should_return_worksheet_with_calculated_values_only( self, mock_urllib2): self.maxDiff = None original_sheet = Worksheet() original_sheet.A2.formula = '1' original_sheet.A2.value = 1 original_sheet.C3.formula = '5' original_sheet.C3.value = 5 original_sheet.E4.formula = '=A2 + C3' original_sheet.E4.value = 6 expected_sheet = Worksheet() expected_sheet.name = 'Untitled' for (col, row), cell in original_sheet.items(): expected_sheet[col, row].value = cell.value foreign_sheet = Sheet() foreign_sheet.owner = User(username='******', password='******') foreign_sheet.owner.save() foreign_sheet.contents_json = worksheet_to_json(original_sheet) foreign_sheet.calculate() mock_opener = mock_urllib2.build_opener.return_value mock_urlopen_file = mock_opener.open.return_value mock_urlopen_file.read.return_value = _sheet_to_value_only_json( foreign_sheet.name, worksheet_from_json(foreign_sheet.contents_json)) worksheet_url = 'ws_url/' result = run_worksheet(worksheet_url, None, sentinel.private_key) target_url = '%sv%s/json/' % (worksheet_url, CURRENT_API_VERSION) self.assertCalledOnce(mock_opener.open, target_url, data=urlencode({ 'dirigible_l337_private_key': sentinel.private_key })) self.assertEquals(type(result), Worksheet) self.assertEquals(result, expected_sheet)
def test_paste_from_copy_does_not_rewrite_source_sheet_formulae( self, mock_rewrite_source_sheet_formulae): source_worksheet = Worksheet() source_worksheet.A1.formula = '=pre-rewrite' source_sheet = Sheet() source_sheet.jsonify_worksheet(source_worksheet) clipboard = Clipboard() clipboard.copy(source_sheet, (1, 2), (3, 5)) clipboard.to_cells = Mock() clipboard.to_cells.return_value = {}.items() def check_to_cells_not_run(*args): self.assertFalse(clipboard.to_cells.call_args_list) mock_rewrite_source_sheet_formulae.side_effect = check_to_cells_not_run clipboard.paste_to(source_sheet, (2, 3), (2, 3)) self.assertFalse( mock_rewrite_source_sheet_formulae.called )
def test_should_handle_cell_formula_overrides_from_GET(self, mock_get_object): mock_sheet = mock_get_object.return_value = Sheet() mock_sheet.owner = self.user worksheet = Worksheet() worksheet[1, 1].formula = 'initial-formula1' worksheet[2, 1].formula = '222' worksheet[3, 1].formula = '=B1+111' mock_sheet.jsonify_worksheet(worksheet) mock_sheet.name = 'mysheet' mock_sheet.allow_json_api_access = True mock_sheet.api_key = 'key' mock_sheet.save = self.die self.request.GET = { u'1,1':u'overriddenformula', u'D1':u'=B1+222', 'api_key': 'key' } self.request.method = 'GET' response = calculate_and_get_json_for_api(self.request, self.user.username, self.sheet.id) self.assertTrue(isinstance(response, HttpResponse)) expected_json = { 'name': 'mysheet', '1' : { '1': 'overriddenformula', }, '2': { '1': 222, }, '3': { '1': 333, }, '4': { '1': 444, }, } self.assertEquals(json.loads(response.content), expected_json)
def test_paste_to_for_copy(self): 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() cells = { (2, 2): c2, (2, 3): c3, (3, 2): d2, (3, 3): d3, } clipboard = Clipboard() clipboard.source_left = 1 clipboard.source_top = 1 clipboard.source_right = 2 clipboard.source_bottom = 2 clipboard.to_cells = Mock() clipboard.to_cells.return_value = cells.iteritems() sheet = Sheet() worksheet = Worksheet() worksheet.B3.formula = 'i am in danger!' worksheet.A1.formula = 'i am safe :-)' sheet.jsonify_worksheet(worksheet) clipboard.paste_to(sheet, (2, 2), (2, 2)) updated_worksheet = sheet.unjsonify_worksheet() self.assertEquals(updated_worksheet.A1.formula, 'i am safe :-)') self.assertEquals(updated_worksheet.B2.formula, c2.formula) self.assertEquals(updated_worksheet.C2.formula, d2.formula) self.assertEquals(updated_worksheet.B3.formula, c3.formula) self.assertEquals(updated_worksheet.C3.formula, d3.formula)
def test_clipboard_remembers_to_close_stingIO_stream(self, mock_stringio): sheet = Sheet() clipboard = Clipboard() mock_stringio.StringIO.return_value = Mock() clipboard.copy(sheet, (1, 1), (1, 1)) self.assertCalledOnce(mock_stringio.StringIO.return_value.close)