def test_init(self): """ Test the format of retrieving the session when logging in. Requires creating a separate client object in order to test _login """ args = { 'taskId': 'login', 'server': 'source', 'project': 'name', 'userid': 'username', 'password': '******' } result = "<response><root><sessionState>session</sessionState><name>" +\ "</name></root></response>" self.mox.StubOutWithMock(MstrClient, '_request') self.mox.StubOutWithMock(MstrClient, '__del__') instance = MstrClient.__new__(MstrClient) instance._request(args).AndReturn(result) instance.__del__().AndReturn(None) instance.__del__().AndReturn(None) self.mox.ReplayAll() client = MstrClient('url?', 'username', 'pw', 'source', 'name') self.assertEqual('session', client._session) self.assertEqual('url?', client._base_url)
def setUp(self): mox.MoxTestBase.setUp(self) s = stubout.StubOutForTesting() s.Set(MstrClient, '_login', lambda self, source, name, username, password: None) self.client = MstrClient('url?', 'username', 'pw', 'source', 'name') self.client._session = 'session' self.mox.StubOutWithMock(self.client, "_request") self.report = Report(self.client, 'report_id') self.report_response = "<response><report_data_list><report_data>" + \ "<prs></prs><objects><attribute rfd='0' id='header1_id' " +\ "name='header1_name' type='header1_type'><form rfd='1' id='frm1_id'" +\ " base_form_type='frm1_basetype' name='frm1_name' id_form='1' " +\ "type='frm1_type'/></attribute><attribute rfd='2' id='header2_id' " +\ "name='header2_name' type='header2_type'><form/></attribute>" +\ "</objects><template></template><raw_data><headers><oi rfd='0'/>" +\ "<oi rfd='2'/></headers><rows cn='100000'><r fr='1'>" +\ "<v id='BB:header1_id:1:1:0:2:col1_val1'>col1_val1</v>" +\ "<v id='BB:header2_id:1:1:0:3:col2_val1'>col2_val1</v></r>" +\ "<r><v id='BB:header1_id:1:1:0:2:col1_val2'>col1_val2</v>" +\ "<v id='BB:header2_id:1:1:0:3:col2_val2'>col2_val2</v></r></rows>" +\ "</raw_data></report_data></report_data_list></response>" self.report_args = { 'taskId': 'reportExecute', 'startRow': 0, 'startCol': 0, 'maxRows': 100000, 'maxCols': 10, 'styleName': 'ReportDataVisualizationXMLStyle', 'resultFlags': '393216', 'reportID': 'report_id', 'sessionState': 'session' }
def setUp(self): mox.MoxTestBase.setUp(self) s = stubout.StubOutForTesting() s.Set(MstrClient, '_login', lambda self, source, name, username, password: None) self.client = MstrClient('url?', 'username', 'pw', 'source', 'name') self.client._session = 'session' self.mox.StubOutWithMock(self.client, "_request")
def test_init(self, mock_mstr_client_request): """ Test the format of retrieving the session when logging in. Requires creating a separate client object in order to test _login """ args = { 'taskId': 'login', 'server': 'source', 'project': 'name', 'userid': 'username', 'password': '******' } result = "<response><root><sessionState>session</sessionState><name>" +\ "</name></root></response>" mock_mstr_client_request.return_value = result client = MstrClient('url?', 'username', 'pw', 'source', 'name') mock_mstr_client_request.assert_called_with(args) self.assertEqual('session', client._session) self.assertEqual('url?', client._base_url)
def setUp(self): with mock.patch.object(MstrClient, '_login'): self.client = MstrClient('url?', 'username', 'pw', 'source', 'name') self.client._session = 'session'