コード例 #1
0
ファイル: test.py プロジェクト: Kokonaut/py-mstr
    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)
コード例 #2
0
ファイル: test.py プロジェクト: Kokonaut/py-mstr
 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'
     }
コード例 #3
0
ファイル: test.py プロジェクト: Kokonaut/py-mstr
 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")
コード例 #4
0
ファイル: test.py プロジェクト: kiran-data-engineer/py-mstr
    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)
コード例 #5
0
ファイル: test.py プロジェクト: kiran-data-engineer/py-mstr
 def setUp(self):
     with mock.patch.object(MstrClient, '_login'):
         self.client = MstrClient('url?', 'username', 'pw', 'source',
                                  'name')
     self.client._session = 'session'