Groceries R Us

#!/usr/bin/env python # # Copyright (C) 2009 Google Inc. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # This module is used for version 2 of the Google Data APIs. __author__ = 'j.s@google.com (Jeff Scudder)' import unittest import gdata.spreadsheets.data import gdata.test_config as conf import atom.core SPREADSHEET = """ http://spreadsheets.google.com/feeds/spreadsheets/private/full/key 2006-11-17T18:24:18.231Z Groceries R Us Groceries R Us Fitzwilliam Darcy fitz@gmail.com """ WORKSHEETS_FEED = """ http://spreadsheets.google.com/feeds/worksheets/key/private/full 2006-11-17T18:23:45.173Z Groceries R Us Fitzwilliam Darcy fitz@gmail.com 1 1 1 http://spreadsheets.google.com/feeds/worksheets/0/private/full/1 2006-11-17T18:23:45.173Z Sheet1 Sheet1 100 20 """ NEW_WORKSHEET = """ Expenses 50 10 """ EDIT_WORKSHEET = """ http://spreadsheets.google.com/feeds/worksheets/k/private/full/w 2007-07-30T18:51:30.666Z Income Expenses 45 15 """ NEW_TABLE = """ Table 1

This is a list of all who have registered to vote and whether or not they qualify to vote.

""" TABLES_FEED = """ http://spreadsheets.google.com/feeds/key/tables 2009-04-28T02:38:53.134Z Sample table and record feed Liz liz@gmail.com 2 1 http://spreadsheets.google.com/feeds/key/tables/0 2009-04-28T01:20:32.707Z 2009-04-28T01:20:32.707Z Table 1

This is a list of all who have registered to vote and whether or not they qualify to vote.

http://spreadsheets.google.com/feeds/key/tables/1 2009-04-28T01:20:38.313Z 2009-04-28T01:20:38.313Z Table 2

List of detailed information about each voter.

""" NEW_RECORD = """ Darcy 2/10/1785 28 Darcy No """ RECORDS_FEED = """ http://spreadsheets.google.com/feeds/key/records/0 2009-04-28T02:38:53.134Z Table 1 Liz liz@gmail.com 2 1 http://spreadsheets.google.com/feeds/key/records/0/cn6ca 2009-04-28T02:38:53.134Z 2009-04-28T02:38:53.134Z Darcy Birthday: 2/10/1785, Age: 28, Name: Darcy, CanVote: No 2/10/1785 28 Darcy No http://spreadsheets.google.com/feeds/key/records/0/cokwr 2009-04-28T02:38:53.134Z 2009-04-28T02:38:53.134Z Jane Birthday: 1/6/1791, Age: 22, Name: Jane, CanVote: Yes 1/6/1791 22 Jane Yes """ LIST_FEED = """ http://spreadsheets.google.com/feeds/list/key/worksheetId/private/full 2006-11-17T18:23:45.173Z Sheet1 Fitzwilliam Darcy fitz@gmail.com 8 1 8 http://spreadsheets.google.com/feeds/list/k/w/private/full/r 2006-11-17T18:23:45.173Z Bingley Hours: 10, Items: 2, IPM: 0.0033 Bingley 10 2 0.0033 http://spreadsheets.google.com/feeds/list/k/w/private/full/rowId 2006-11-17T18:23:45.173Z Charlotte Hours: 60, Items: 18000, IPM: 5 Charlotte 60 18000 5 """ NEW_ROW = """ 1 1 60 Elizabeth Bennet """ UPDATED_ROW = """ http://spreadsheets.google.com/feeds/list/k/w/private/full/rowId 2006-11-17T18:23:45.173Z Bingley Hours: 10, Items: 2, IPM: 0.0033 Bingley 20 4 0.0033 """ CELLS_FEED = """ http://spreadsheets.google.com/feeds/cells/key/worksheetId/private/full 2006-11-17T18:27:32.543Z Sheet1 Fitzwilliam Darcy fitz@gmail.com 1 36 100 20 http://spreadsheets.google.com/feeds/cells/k/w/private/full/R1C1 2006-11-17T18:27:32.543Z A1 Name Name http://spreadsheets.google.com/feeds/cells/k/w/private/full/R1C2 2006-11-17T18:27:32.543Z B1 Hours Hours http://spreadsheets.google.com/feeds/cells/k/w/private/full/R9C4 2006-11-17T18:27:32.543Z D9 5 5 """ BATCH_CELLS = """ http://spreadsheets.google.com/feeds/cells/key/worksheetId/private/full A1 http://spreadsheets.google.com/feeds/cells/k/w/private/full/cellId A2 A2 http://spreadsheets.google.com/feeds/cells/k/w/private/full/cellId """ class SpreadsheetEntryTest(unittest.TestCase): def setUp(self): self.spreadsheet = atom.core.parse( SPREADSHEET, gdata.spreadsheets.data.Spreadsheet) def test_check_parsing(self): self.assertEqual(self.spreadsheet.etag, '"BxAUSQUJRCp7ImBq"') self.assertEqual(self.spreadsheet.id.text, 'http://spreadsheets.google.com/feeds/spreadsheets' '/private/full/key') self.assertEqual(self.spreadsheet.updated.text, '2006-11-17T18:24:18.231Z') self.assertEqual(self.spreadsheet.find_worksheets_feed(), 'http://spreadsheets.google.com/feeds/worksheets' '/key/private/full') self.assertEqual(self.spreadsheet.find_self_link(), 'http://spreadsheets.google.com/feeds/spreadsheets' '/private/full/key') def test_get_spreadsheet_key(self): self.assertEqual(self.spreadsheet.get_spreadsheet_key(), 'key') # Change the value of the self link. self.spreadsheet.id.text = '42' self.assertEqual(self.spreadsheet.GetSpreadsheetKey(), '42') class WorksheetEntryTest(unittest.TestCase): def setUp(self): self.worksheets = atom.core.parse( WORKSHEETS_FEED, gdata.spreadsheets.data.WorksheetsFeed) def test_check_parsing(self): self.assertEqual(len(self.worksheets.entry), 1) self.assertEqual(self.worksheets.entry[0].get_id(), 'http://spreadsheets.google.com/feeds/worksheets/0/private/full/1') def test_get_worksheet_id(self): self.assertEqual(self.worksheets.entry[0].get_worksheet_id(), '1') self.worksheets.entry[0].id.text = '////spam' self.assertEqual(self.worksheets.entry[0].GetWorksheetId(), 'spam') class ListEntryTest(unittest.TestCase): def test_get_and_set_column_value(self): row = atom.core.parse(NEW_ROW, gdata.spreadsheets.data.ListEntry) row.set_value('hours', '3') row.set_value('name', 'Lizzy') self.assertEqual(row.get_value('hours'), '3') self.assertEqual(row.get_value('ipm'), '1') self.assertEqual(row.get_value('items'), '60') self.assertEqual(row.get_value('name'), 'Lizzy') self.assertEqual(row.get_value('x'), None) row.set_value('x', 'Test') self.assertEqual(row.get_value('x'), 'Test') row_xml = str(row) self.assert_(row_xml.find(':x') > -1) self.assert_(row_xml.find('>Test -1) self.assert_(row_xml.find(':hours') > -1) self.assert_(row_xml.find('>3 -1) self.assert_(row_xml.find(':ipm') > -1) self.assert_(row_xml.find('>1 -1) self.assert_(row_xml.find(':items') > -1) self.assert_(row_xml.find('>60 -1) self.assert_(row_xml.find(':name') > -1) self.assert_(row_xml.find('>Lizzy -1) self.assertEqual(row_xml.find(':zzz'), -1) self.assertEqual(row_xml.find('>foo