mocks.SpreadsheetMock

Here are the examples of the python api mocks.SpreadsheetMock taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.

7 Examples 7

3 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_id(self):
        """Check whether id attribute is working fine."""
        ss_mock = SpreadsheetMock(CONFIG, "test_id")
        self.assertEqual(ss_mock._id, ss_mock.id)
        with self.assertRaises(AttributeError):
            ss_mock.id = "new_id"

    def test_update_structure_no_update(self):

3 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_update_all_sheets(self):
        """Update sheets one by one."""
        ss_mock = SpreadsheetMock(CONFIG)
        sheet1 = SheetMock("sheet1", SPREADSHEET_ID)
        sheet2 = SheetMock("sheet2", SPREADSHEET_ID)

        ss_mock.sheets = {"sheet1": sheet1, "sheet2": sheet2}
        with mock.patch("sheet.Sheet.update") as update_sheet:
            ss_mock.update_all_sheets()

            # no spreadsheet resource used in
            # this test, so args will be None
            update_sheet.assert_has_calls((mock.call(None, {}), mock.call(None, {})))

    def test_reload_config(self):

3 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_create(self):
        ss_mock = SpreadsheetMock(CONFIG)

        execute_mock = mock.Mock(get=mock.Mock(return_value=SPREADSHEET_ID))
        create_mock = mock.Mock(return_value=mock.Mock(execute=execute_mock))
        ss_mock._ss_resource = mock.Mock(create=create_mock)

        ss_mock._create()
        create_mock.assert_called_once_with(
            body={
                "properties": {"title": "MockTitle"},
                "sheets": [
                    {"properties": {"title": "sheet1"}},
                    {"properties": {"title": "sheet2"}},
                ],
            }
        )

0 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def setUpClass(self):
        """Init mock for spreadsheet."""
        self._ss_mock = SpreadsheetMock(CONFIG)

    def _prepare_batch_mock(self):

0 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_update_structure(self):
        """
        Test spreadsheet structure updating.
        New sheet creation and old sheet deleting included.
        """
        SHEET3_ID = 5623
        RENAME_REQUEST = {
            "updateSpreadsheetProperties": {
                "properties": {"title": "MockTitle"},
                "fields": "title",
            }
        }
        CREATE_REQUEST = {
            "addSheet": {
                "properties": {
                    "title": "sheet1",
                    "gridProperties": {"rowCount": 1000, "columnCount": 26},
                }
            }
        }
        DELETE_REQUEST = {"deleteSheet": {"sheetId": SHEET3_ID}}

        ss_mock = SpreadsheetMock(CONFIG, "test_id")
        ss_mock.sheets = {
            "sheet2": SheetMock("sheet2", SPREADSHEET_ID, 123),
            "sheet3": SheetMock("sheet3", SPREADSHEET_ID, SHEET3_ID),
        }

        batch_mock = self._prepare_batch_mock()
        ss_mock._ss_resource = mock.Mock(batchUpdate=batch_mock)
        ss_mock._config_updated = True

        actual_mock = ss_mock._actualize_sheets = mock.Mock()
        with mock.patch(
            "sheet_builder.SheetBuilder._login_on_github", return_value=github.Github
        ):
            ss_mock.update_structure()
            actual_mock.assert_called_once()

        batch_mock.assert_called_once_with(
            body={"requests": [RENAME_REQUEST, CREATE_REQUEST, DELETE_REQUEST]},
            spreadsheetId="ss_id",
        )

    def test_update_all_sheets(self):

0 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_init_existing_sheets(self):
        SHEET1 = "sheet1"
        SHEET2 = "sheet2"
        SHEET1_ID = 6345345
        SHEET2_ID = 1456241

        execute_mock = mock.Mock(
            return_value={
                "sheets": [
                    {"properties": {"title": SHEET1, "sheetId": SHEET1_ID}},
                    {"properties": {"title": SHEET2, "sheetId": SHEET2_ID}},
                ]
            }
        )
        get_mock = mock.Mock(return_value=mock.Mock(execute=execute_mock))

        ss_mock = SpreadsheetMock(CONFIG)
        ss_mock._ss_resource = mock.Mock(get=get_mock)

        with mock.patch("sheet_builder.SheetBuilder", return_value=SheetBuilderMock):
            sheets = ss_mock._init_existing_sheets()
        self.assertEqual(sheets[SHEET1].id, SHEET1_ID)
        self.assertEqual(sheets[SHEET1].name, SHEET1)

        self.assertEqual(sheets[SHEET2].id, SHEET2_ID)
        self.assertEqual(sheets[SHEET2].name, SHEET2)

    def test_actualize_sheets(self):

0 Source : test_spreadsheet.py
with Apache License 2.0
from IlyaFaer

    def test_actualize_sheets(self):
        SHEET1 = "sheet1"
        SHEET2 = "sheet2"
        SHEET2_ID = 1456241

        ss_mock = SpreadsheetMock(CONFIG)
        ss_mock.sheets = {
            SHEET1: SheetMock(SHEET1, SPREADSHEET_ID),
            SHEET2: SheetMock(SHEET2, SPREADSHEET_ID),
        }

        execute_mock = mock.Mock(
            return_value={
                "sheets": [{"properties": {"title": SHEET2, "sheetId": SHEET2_ID}}]
            }
        )
        get_mock = mock.Mock(return_value=mock.Mock(execute=execute_mock))
        ss_mock._ss_resource = mock.Mock(get=get_mock)

        ss_mock._actualize_sheets()
        self.assertIsNone(ss_mock.sheets.get(SHEET1))
        self.assertEqual(ss_mock.sheets[SHEET2].id, SHEET2_ID)

    def test_new_sheets_requests(self):