Пример #1
0
 def tearDown(self):
     """
     tearDown
     :return:
     """
     super(TestServer, self).tearDown()
     minimock.restore()
Пример #2
0
    def test_no_email_notify(self):
        topost = {
            'login_id': 'usersaved_noemailnotify',
            'email_address': '*****@*****.**',
            'user-submit-flag': 'submitted',
            'approved_permissions': None,
            'denied_permissions': None,
            'assigned_groups': None,
            'super_user': 1,
            'inactive_flag': False,
            'inactive_date': '10/11/2010',
            'name_first': 'test',
            'name_last': 'user'
        }

        # setup the mock objects so we can test the email getting sent out
        tt = minimock.TraceTracker()
        smtplib.SMTP = minimock.Mock('smtplib.SMTP', tracker=None)
        smtplib.SMTP.mock_returns = minimock.Mock('smtp_connection',
                                                  tracker=tt)

        req, r = self.c.post('users/add', data=topost, follow_redirects=True)
        assert r.status_code == 200, r.status
        assert b'User added' in r.data
        assert req.url.endswith('users/manage')

        assert len(tt.dump()) == 0
        minimock.restore()
Пример #3
0
    def test_get_client_invalid(self):
        """Test getting a connection

        """
        minimock.restore()
        mongothin.connection.register_connection("invalid", "mongothin", host="localhost", invalidaram="invalid")
        self.assertRaises(mongothin.connection.ConnectionError, mongothin.connection.get_connection, "invalid")
Пример #4
0
    def test_parseUrl(self):
        "CSSParser.parseUrl()"
        if mock:
            # parseUrl(self, href, encoding=None, media=None, title=None):
            parser = cssutils.CSSParser()
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, u''))
            sheet = parser.parseUrl('http://example.com',
                                    media='tv,print', 
                                    title='test')
            restore()
            #self.assertEqual(sheet, 1)
            self.assertEqual(sheet.href, 'http://example.com')
            self.assertEqual(sheet.encoding, 'utf-8')
            self.assertEqual(sheet.media.mediaText, 'tv, print')
            self.assertEqual(sheet.title, 'test')
            
            # URL and content tests
            tests = {
                # (url, content): isSheet, encoding, cssText
                ('', None): (False, None, None), 
                ('1', None): (False, None, None), 
                ('mailto:[email protected]', None): (False, None, None), 
                ('http://example.com/x.css', None): (False, None, None), 
                ('http://example.com/x.css', ''): (True, u'utf-8', u''), 
               # ('http://example.com/x.css', 'a'): (True, u'utf-8', u''), 
#                ('http://example.com/x.css', 'a {color: red}'): (True, u'utf-8', 
#                                                                 u'a {\n    color: red\n    }'), 
#                ('http://example.com/x.css', 'a {color: red}'): (True, u'utf-8', 
#                                                                 u'a {\n    color: red\n    }'), 
#                ('http://example.com/x.css', '@charset "ascii";a {color: red}'): (True, u'ascii', 
#                                                                 u'@charset "ascii";\na {\n    color: red\n    }'), 
            }
            override = 'iso-8859-1'
            overrideprefix = u'@charset "iso-8859-1";'
            httpencoding = None
            
            for (url, content), (isSheet, expencoding, cssText) in tests.items():
                mock("cssutils.util._defaultFetcher", 
                     mock_obj=self._make_fetcher(httpencoding, content))
                #parser.setFetcher(self._make_fetcher(httpencoding, content))
                sheet1 = parser.parseUrl(url)
                sheet2 = parser.parseUrl(url, encoding=override)
                restore()
                if isSheet:
                    self.assertEqual(sheet1.encoding, expencoding)
                    self.assertEqual(sheet1.cssText, cssText)
                    self.assertEqual(sheet2.encoding, override)
                    if sheet1.cssText and sheet1.cssText.startswith('@charset'):
                        self.assertEqual(sheet2.cssText, cssText.replace('ascii', override))
                    elif sheet1.cssText:
                        self.assertEqual(sheet2.cssText, overrideprefix + '\n' + cssText)
                    else:
                        self.assertEqual(sheet2.cssText, overrideprefix + cssText)
                else:
                    self.assertEqual(sheet1, None)
                    self.assertEqual(sheet2, None)

            self.assertRaises(ValueError, parser.parseUrl, '../not-valid-in-urllib')
            self.assertRaises(urllib2.HTTPError, parser.parseUrl, 'http://example.com/not-present.css')
Пример #5
0
    def test_parseUrl(self):
        "CSSParser.parseUrl()"
        if mock:
            # parseUrl(self, href, encoding=None, media=None, title=None):
            parser = cssutils.CSSParser()
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, u''))
            sheet = parser.parseUrl('http://example.com',
                                    media='tv,print', 
                                    title='test')
            restore()
            #self.assertEqual(sheet, 1)
            self.assertEqual(sheet.href, 'http://example.com')
            self.assertEqual(sheet.encoding, 'utf-8')
            self.assertEqual(sheet.media.mediaText, 'tv, print')
            self.assertEqual(sheet.title, 'test')
            
            # URL and content tests
            tests = {
                # (url, content): isSheet, encoding, cssText
                ('', None): (False, None, None), 
                ('1', None): (False, None, None), 
                ('mailto:[email protected]', None): (False, None, None), 
                ('http://example.com/x.css', None): (False, None, None), 
                ('http://example.com/x.css', ''): (True, u'utf-8', u''), 
               # ('http://example.com/x.css', 'a'): (True, u'utf-8', u''), 
#                ('http://example.com/x.css', 'a {color: red}'): (True, u'utf-8', 
#                                                                 u'a {\n    color: red\n    }'), 
#                ('http://example.com/x.css', 'a {color: red}'): (True, u'utf-8', 
#                                                                 u'a {\n    color: red\n    }'), 
#                ('http://example.com/x.css', '@charset "ascii";a {color: red}'): (True, u'ascii', 
#                                                                 u'@charset "ascii";\na {\n    color: red\n    }'), 
            }
            override = 'iso-8859-1'
            overrideprefix = u'@charset "iso-8859-1";'
            httpencoding = None
            
            for (url, content), (isSheet, expencoding, cssText) in tests.items():
                mock("cssutils.util._defaultFetcher", 
                     mock_obj=self._make_fetcher(httpencoding, content))
                #parser.setFetcher(self._make_fetcher(httpencoding, content))
                sheet1 = parser.parseUrl(url)
                sheet2 = parser.parseUrl(url, encoding=override)
                restore()
                if isSheet:
                    self.assertEqual(sheet1.encoding, expencoding)
                    self.assertEqual(sheet1.cssText, cssText)
                    self.assertEqual(sheet2.encoding, override)
                    if sheet1.cssText and sheet1.cssText.startswith('@charset'):
                        self.assertEqual(sheet2.cssText, cssText.replace('ascii', override))
                    elif sheet1.cssText:
                        self.assertEqual(sheet2.cssText, overrideprefix + '\n' + cssText)
                    else:
                        self.assertEqual(sheet2.cssText, overrideprefix + cssText)
                else:
                    self.assertEqual(sheet1, None)
                    self.assertEqual(sheet2, None)

            self.assertRaises(ValueError, parser.parseUrl, '../not-valid-in-urllib')
            self.assertRaises(urllib2.HTTPError, parser.parseUrl, 'http://example.com/not-present.css')
Пример #6
0
    def testTripwireTripped(self):
        '''Tripwire test, tripped'''
        tt = minimock.TraceTracker()
        minimock.mock('tripwire.sendEmail', tracker=tt)
        
        datevalue = datetime.date(2011, 8, 16)
        
        dtmock = minimock.Mock('datetime.date')
        minimock.mock('datetime.date', mock_obj=dtmock)
        dtmock.mock_returns = dtmock
        dtmock.today.mock_returns = datevalue
        dtmock.today.mock_tracker = tt
        #can't just do minimock.mock('datetime.date.today', returns=datevalue, tracker=tt)
        #because datetime.date is in an extension (ie, not native python)
        
        #Add another value to make the tripwire trip
        self.session.add(model.MeasuredValue(6))
        self.session.commit()
        
        tripwire.main()
        expected = r'''Called datetime.date.today()
Called tripwire.sendEmail(3.0, datetime.date(2011, 8, 16))'''
        self.assertTrue(tt.check(expected), tt.diff(expected))
        
        minimock.restore()
        
Пример #7
0
def mock_get(url, **kwargs):
    """
    Mock requests.get with the contents of a local file. If the file doesn't
    exist, make the request and save to the file.
    """
    cache_dir = build_path(join_path('cache'))
    key = hashlib.md5(url.encode('utf-8')).hexdigest()
    file_path = os.path.join(cache_dir, '{}.xml'.format(key))
    try:
        return MockResponse(file_path)
    except IOError:
        restore()  # restore normal function
        r = requests.get(url)
        if not r.ok:
            mock('requests.get', returns_func=mock_get,
                 tracker=None)  # re-mock it.
            return r
        with open(file_path, 'w') as tmp:
            try:
                tmp.write(r.text)
            except UnicodeEncodeError:
                tmp.write(r.content)  # python 2
        mock('requests.get', returns_func=mock_get,
             tracker=None)  # re-mock it.
        return MockResponse(file_path)
Пример #8
0
 def test_ini_non_exist_or_bad(self):
     """
     bad file
     """
     os.environ = minimock.Mock("os.environ")
     os.environ.get.mock_returns = "/tmp/tdtdtd"
     self.assertEqual(h.base.load_ini().sections(), [])
     minimock.restore()
Пример #9
0
 def test_env_not_set(self):
     """
     no ini set in environ
     """
     os.environ = minimock.Mock("os.environ")
     os.environ.get.mock_returns = None
     self.assertRaises(ValueError, h.base.load_ini)
     minimock.restore()
Пример #10
0
 def tearDown(self):
   restore()
   try:
     # logging
     for record in self.log.records:
       if self.is_endpoints and len(record.args) >= 3:
         exception = record.args[2]
         if isinstance(exception, endpoints.ServiceException):
           try:
             str(exception)
           except TypeError:
             record.args[2].args = [str(arg) for arg in exception.args]
           except UnicodeEncodeError:
             record.args[2].args = [unicode(arg) for arg in exception.args]
       pathname = get_tail(record.pathname).group("tail")
       curdir_abspath = os.path.abspath(os.curdir)
       if is_mac:
         if curdir_abspath.startswith(mac_volumes_prefix):
           curdir_abspath = curdir_abspath[mac_volumes_prefix_length:]
         if pathname.startswith(mac_volumes_prefix):
           pathname = pathname[mac_volumes_prefix_length:]
       else:
         curdir_abspath = curdir_abspath.lstrip('/')
         pathname = pathname.lstrip('/')
       log = (record.levelname, pathname.replace(curdir_abspath, "").lstrip("/"), record.funcName, record.getMessage())
       if getattr(self, "expected_logs", None):
         if log in self.expected_logs:
           continue
         matched = None
         for expected_log in self.expected_logs:
           if "..." in expected_log[3]:
             if log[:2] == expected_log[:2]:
               if doctest._ellipsis_match(expected_log[3], log[3]):
                 matched = True
                 continue
         if matched:
           continue
       elif self.is_endpoints:
         expected_log = (
           'WARNING',
           'google/appengine/ext/ndb/tasklets.py',
           '_help_tasklet_along',
           '... generator ...(....py:...) raised ...Exception(...)')
         if log[:2] == expected_log[:2]:
           if doctest._ellipsis_match(expected_log[3], log[3]):
             matched = True
             continue
       print(record.levelname, pathname, record.lineno, record.funcName, record.getMessage())
       assert not log
   finally:
     self.log.clear()
     self.log.uninstall()
     # testbed
     self.testbed.deactivate()
     # os.environ
     for key, value in self.origin_environ.iteritems():
       if value is not None:
         os.environ[key] = value
 def test_EncodeDataCheckProblemsFile_Exists(self):
     showname = "test show"
     inputname = "test input"
     outputname = "test_output_.mkv"
     data = EncodeData(showname, inputname, outputname)
     mock("os.path.exists", returns_iter=[False, True])
     result = data.checkproblems()
     self.assertIn("FILE_EXISTS", result)
     minimock.restore()
Пример #12
0
 def test_make_server_spawn(self):
     """
     Check the spawn option for the backend that support it
     :return:
     """
     for backend in ['gevent', 'fastgevent', 'geventwebsocket', 'socketio']:
         self.tt = minimock.TraceTracker()
         self._check_make_server_spawn(backend)
         minimock.restore()
Пример #13
0
def test_gpsdshm_Shm_error():
    gpsdshm.shm.shm_get = minimock.Mock('gpsdshm.shm.shm_get')
    gpsdshm.shm.shm_get.mock_returns = None
    try:
        gpsdshm.Shm()
    except OSError:
        minimock.restore()
        return
    raise Exception('gpsdshm.shm.shm_get did nto raise OSError')
Пример #14
0
 def tearDown(self):
     """
     tearDown
     :return:
     """
     super(TestServer, self).tearDown()
     minimock.restore()
     if self.old is not None:
         socket.socket.bind = self.old
Пример #15
0
def test_gpsdshm_Shm_error():
    gpsdshm.shm.shm_get = minimock.Mock("gpsdshm.shm.shm_get")
    gpsdshm.shm.shm_get.mock_returns = None
    try:
        gpsdshm.Shm()
    except OSError:
        minimock.restore()
        return
    raise Exception("gpsdshm.shm.shm_get did nto raise OSError")
Пример #16
0
    def tearDown(self):
        """Teardown

        """
        super(Testconnection, self).tearDown()
        mongothin.connection._connection_settings.clear()
        mongothin.connection._connections.clear()
        mongothin.connection._dbs.clear()
        minimock.restore()
Пример #17
0
 def test_make_server_spawn(self):
     """
     Check the spawn option for the backend that support it
     :return:
     """
     for backend in ['gevent', 'fastgevent', 'geventwebsocket', 'socketio']:
         self.tt = minimock.TraceTracker()
         self._check_make_server_spawn(backend)
         minimock.restore()
Пример #18
0
Файл: test.py Проект: me-and/mdf
    def tearDown(self):
        '''Clean up from each test case.

        Switch back to the original working directory, and clean up the test
        directory and any mocked objects.
        '''
        os.chdir(self.orig_cwd)
        self.test_dir.cleanup()
        minimock.restore()
Пример #19
0
 def testRight(self):
     '''
     Will test if a right click event will open the correct Right-Click Menu.
     '''
     evt = wx.TreeEvent(wx.wxEVT_COMMAND_TREE_ITEM_RIGHT_CLICK)
     minimock.mock("evt.GetItem",returns=self.frame.tree.GetRootItem())
     minimock.mock("evt.GetPoint",returns=wx.Point(83, 121))
     self.frame.OnRight(evt)
     minimock.restore()
     minimock.restore()
    def test_checkduplicatethomas(self):
        settings = Mock('libsettings.Settings')
        filemanager = FileManager(settings)

        os.walk = thomaswalk

        result = filemanager.checkduplicates("/path/to/S12E05 - Henry Gets It Wrong - SD TV.mkv")

        self.assertTrue(result)
        minimock.restore()
    def test_checkfileexistscaseinsensitive(self):
        settings = Mock('libsettings.Settings')
        filemanager = FileManager(settings)

        mock("os.listdir", returns=["filename.test"])

        result = filemanager.checkfileexists("/path/to/fiLename.test", False)

        self.assertTrue(result)
        minimock.restore()
    def test_checkduplicatesameextension(self):
        settings = Mock('libsettings.Settings')
        filemanager = FileManager(settings)

        os.walk = dummywalk

        result = filemanager.checkduplicates("/path/to/S03E14 - Test - SD TV.avi")

        self.assertFalse(result)
        minimock.restore()
Пример #23
0
 def testTripwireNormal(self):
     '''Tripwire test, not tripped'''
     tt = minimock.TraceTracker()
     minimock.mock('tripwire.sendEmail', tracker=tt)
     
     tripwire.main()
     
     self.assertTrue(tt.check(''), tt.diff(''))
     
     minimock.restore()
Пример #24
0
    def test_make_server(self):
        """
        Test all backends with default params
        :return:
        """

        # nose does not have great support for parameterized tests
        for backend in backends():
            self.tt = minimock.TraceTracker()
            self._check_make_server(backend)
            minimock.restore()
Пример #25
0
 def test_check_prereqs(self):
     "Test active_site_finder.check_prereqs method"
     # THIS TEST HAS TO BE UPDATED WHEN NEW PROFILES ARE ADDED TO THE MODULE!
     
     self.tt = TraceTracker()
     
     mock('utils.locate_executable', returns="/my/path/to/executable", tracker=self.tt)
     mock('utils.locate_file', returns="/my/path/to/file", tracker=self.tt)
     
     result = self.my_ASF.check_prereqs()
     
     self.assertListEqual(result, [], "return empty list if executables/files are found")
     
     expected = """    Called utils.locate_executable('blastp')
 Called utils.locate_executable('hmmpfam2')
 Called utils.locate_executable('hmmscan')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-KR.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-KS_N.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-KS_C.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-AT.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-ACP.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-DH.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-KR.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/Thioesterase.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-ER.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/PKSI-AT.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/aa-activating.aroundLys.hmm2')
 Called utils.locate_file(
     '.../antismash/generic_modules/active_site_finder/hmm/p450.hmm3')"""
     assert_same_trace(self.tt, expected)
     
     restore()
     self.tt = TraceTracker()
     
     mock('utils.locate_executable', returns=None, tracker=self.tt)
     mock('utils.locate_file', returns="/my/path/to/file", tracker=self.tt)
     
     result = self.my_ASF.check_prereqs()
     expected = ["Failed to locate file: 'blastp'", "Failed to locate file: 'hmmpfam2'", "Failed to locate file: 'hmmscan'"]
     self.assertListEqual(result, expected, "test result if file not found")
     restore()
Пример #26
0
    def test_resolveImports(self):
        "cssutils.resolveImports(sheet)"
        if mock:
            self._tempSer()
            cssutils.ser.prefs.useMinified()

            a = u'@charset "iso-8859-1";@import"b.css";ä{color:green}'.encode('iso-8859-1')
            b = u'@charset "ascii";\E4 {color:red}'.encode('ascii')
            
            # normal
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()            
            self.assertEqual(a, s.cssText)
            self.assertEqual(b, s.cssRules[1].styleSheet.cssText)
            c = cssutils.resolveImports(s)
            self.assertEqual('\xc3\xa4{color:red}\xc3\xa4{color:green}', 
                             c.cssText)

            c.encoding = 'ascii'
            self.assertEqual(r'@charset "ascii";\E4 {color:red}\E4 {color:green}', 
                             c.cssText)

            # b cannot be found
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, None))
            s = cssutils.parseString(a)
            restore()            
            self.assertEqual(a, s.cssText)
            self.assertEqual(None, s.cssRules[1].styleSheet)
            c = cssutils.resolveImports(s)
            self.assertEqual('@import"b.css";\xc3\xa4{color:green}', 
                             c.cssText)

            # @import with media
            a = u'@import"b.css";@import"b.css" print, tv ;@import"b.css" all;'
            b = u'a {color: red}'            
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()            
            c = cssutils.resolveImports(s)
            self.assertEqual('a{color:red}@media print,tv{a{color:red}}a{color:red}', 
                             c.cssText)
            
            # cannot resolve with media => keep original
            a = u'@import"b.css"print;'
            b = u'@namespace "http://example.com";'            
            mock("cssutils.util._defaultFetcher", 
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()            
            c = cssutils.resolveImports(s)
            self.assertEqual(a, c.cssText)


            
        else:
            self.assertEqual(False, u'Minimock needed for this test')
Пример #27
0
    def test_resolveImports(self):
        "cssutils.resolveImports(sheet)"
        if mock:
            self._tempSer()
            cssutils.ser.prefs.useMinified()

            a = u'@charset "iso-8859-1";@import"b.css";ä{color:green}'.encode(
                'iso-8859-1')
            b = u'@charset "ascii";\E4 {color:red}'.encode('ascii')

            # normal
            mock("cssutils.util._defaultFetcher",
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()
            self.assertEqual(a, s.cssText)
            self.assertEqual(b, s.cssRules[1].styleSheet.cssText)
            c = cssutils.resolveImports(s)
            self.assertEqual('\xc3\xa4{color:red}\xc3\xa4{color:green}',
                             c.cssText)

            c.encoding = 'ascii'
            self.assertEqual(
                r'@charset "ascii";\E4 {color:red}\E4 {color:green}',
                c.cssText)

            # b cannot be found
            mock("cssutils.util._defaultFetcher",
                 mock_obj=self._make_fetcher(None, None))
            s = cssutils.parseString(a)
            restore()
            self.assertEqual(a, s.cssText)
            self.assertEqual(None, s.cssRules[1].styleSheet)
            c = cssutils.resolveImports(s)
            self.assertEqual('@import"b.css";\xc3\xa4{color:green}', c.cssText)

            # @import with media
            a = u'@import"b.css";@import"b.css" print, tv ;@import"b.css" all;'
            b = u'a {color: red}'
            mock("cssutils.util._defaultFetcher",
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()
            c = cssutils.resolveImports(s)
            self.assertEqual(
                'a{color:red}@media print,tv{a{color:red}}a{color:red}',
                c.cssText)

            # cannot resolve with media => keep original
            a = u'@import"b.css"print;'
            b = u'@namespace "http://example.com";'
            mock("cssutils.util._defaultFetcher",
                 mock_obj=self._make_fetcher(None, b))
            s = cssutils.parseString(a)
            restore()
            c = cssutils.resolveImports(s)
            self.assertEqual(a, c.cssText)

        else:
            self.assertEqual(False, u'Minimock needed for this test')
Пример #28
0
def test_nose_component_app_package_by_environ():
    import minimock
    import tempfile
    from minimal2.config.settings import Default
    new_script_test_path = tempfile.mkdtemp()
    Default.get_storage_dir = minimock.Mock('get_storage_dir')
    Default.get_storage_dir.mock_returns = os.path.join(new_script_test_path, 'minimal2')

    base_environ['BLAZEWEB_APP_PACKAGE'] = 'minimal2'
    newenv = BWTestFileEnvironment(new_script_test_path, environ=base_environ)
    res = newenv.run('nosetests', 'minimal2', expect_error=True, cwd=apps_path)
    assert 'Ran 1 test in' in res.stderr, res.stderr
    assert 'OK' in res.stderr, res.stderr
    minimock.restore()
Пример #29
0
 def test_fail_at_acquireWriteLock(self):
     self.setUpContext()
     mock('ReadWriteLock.acquireWriteLock', raises=Exception, tracker=None)
     self.assertRaises(Exception, self.context.dispatch, 'pseudo-init', self.obj)
     self.assertEqual(0, _FantasmFanIn.all().count())
     self.assertEqual(None, memcache.get('foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
     restore()
     
     self.setUpContext(retryCount=1)
     self.context.dispatch('pseudo-init', self.obj)
     self.assertEqual(1, _FantasmFanIn.all().count())
     self.assertEqual('foo--InitialState--ok--FanInState--step-2-2957927341', 
                      _FantasmFanIn.all().get().workIndex)
     self.assertEqual('65536', memcache.get('foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
Пример #30
0
 def test_sends_message_to_all_connected_clients(self):
     minimock.restore()
     self.tracker = minimock.TraceTracker()
     models.User.create_user('4321', email='*****@*****.**')
     user = models.User.lookup(email='*****@*****.**')
     client_id = '{0}.{1}.1234.5678'.format(self.server.key.id(), self.user.key.id())
     client_id2 = '{0}.{1}.1234.5678'.format(self.server.key.id(), user.key.id())
     minimock.mock('channel.ServerChannels.get_client_ids', returns=[client_id, client_id2], tracker=None)
     minimock.mock('gae_channel.send_message', tracker=self.tracker)
     self.interesting_log_line.send_message()
     js = '{"username": "******", "achievement_message": null, "chat": "is there anybody in there?", "time": "12:01am", "date": "Mar 24, 2013", "event": "chat", "death_message": null}'
     trace = "Called gae_channel.send_message(\n    '{0}',\n    '{1}')".format(client_id, js)
     trace += "\nCalled gae_channel.send_message(\n    '{0}',\n    '{1}')".format(client_id, js)
     minimock.assert_same_trace(self.tracker, trace)
Пример #31
0
    def test_fail_at_put(self):
        self.setUpContext()
        mock('db.put', raises=Exception, tracker=None)
        self.assertRaises(Exception, self.context.dispatch, 'pseudo-init', copy.copy(self.obj))
        self.assertEqual(0, _FantasmFanIn.all(namespace='').count())
        # notice the +1 extra on the lock
        self.assertEqual(65537, memcache.get('foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
        restore()

        self.setUpContext(retryCount=1)
        self.context.dispatch('pseudo-init', self.obj)
        self.assertEqual(1, _FantasmFanIn.all(namespace='').count())
        self.assertEqual('foo--InitialState--ok--FanInState--step-2-2957927341',
                         _FantasmFanIn.all(namespace='').get().workIndex)
        self.assertEqual(65537, memcache.get('foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
Пример #32
0
def test_nose_component_app_package_by_environ():
    import minimock
    import tempfile
    from minimal2.config.settings import Default
    new_script_test_path = tempfile.mkdtemp()
    Default.get_storage_dir = minimock.Mock('get_storage_dir')
    Default.get_storage_dir.mock_returns = os.path.join(
        new_script_test_path, 'minimal2')

    base_environ['BLAZEWEB_APP_PACKAGE'] = 'minimal2'
    newenv = BWTestFileEnvironment(new_script_test_path, environ=base_environ)
    res = newenv.run('nosetests', 'minimal2', expect_error=True, cwd=apps_path)
    assert 'Ran 1 test in' in res.stderr, res.stderr
    assert 'OK' in res.stderr, res.stderr
    minimock.restore()
Пример #33
0
    def test__execute_tool(self):
        "Test active_site_finder._execute_tool method"
        self.tt = TraceTracker()
        self.tt2 = TraceTracker()
        mock('utils.execute', returns=["shell output", 0, 0], tracker=self.tt)

        mock('SearchIO.parse', returns=["SearchIO object"], tracker=self.tt2)

        result = self.my_ASF._execute_tool(self.ETObj.find('./analysis'),
                                           fileName="testTempfile")

        self.assertListEqual(result, ["SearchIO object"])

        expected = """        Called utils.execute(
            ['hmmscan', '--domE', '0.1', '--cpu', '1', '.../antismash/generic_modules/active_site_finder/hmm/p450.hmm3', 'testTempfile'])
      """
        assert_same_trace(self.tt, expected)

        expected = "   Called SearchIO.parse(<cStringIO.StringI object at ...>, 'hmmer3-text')"
        assert_same_trace(self.tt2, expected)

        restore()

        self.tt = TraceTracker()
        self.tt2 = TraceTracker()
        mock('utils.execute', returns=["shell output", 0, 0], tracker=self.tt)

        mock('SearchIO.parse', returns=["SearchIO object"], tracker=self.tt2)

        result = self.my_ASF._execute_tool(
            self.ETObj.find('./analysis'),
            stdin_data="fasta sequence from stdin")

        self.assertListEqual(result, ["SearchIO object"])

        expected = """        Called utils.execute(
            ['hmmscan', '--domE', '0.1', '--cpu', '1', '.../antismash/generic_modules/active_site_finder/hmm/p450.hmm3'],
            input='fasta sequence from stdin')
      """
        assert_same_trace(self.tt, expected)

        expected = "   Called SearchIO.parse(<cStringIO.StringI object at ...>, 'hmmer3-text')"
        assert_same_trace(self.tt2, expected)

        restore()
        self.tt = TraceTracker()
        del self.tt2
Пример #34
0
    def testSendEmail(self):
        tt = minimock.TraceTracker()
        smtpconn = minimock.Mock('smtplib.SMTP', tracker=tt)
        minimock.mock('smtplib.SMTP', mock_obj=smtpconn)
        smtpconn.mock_returns = smtpconn
        smtpconn.sendmail.mock_returns = {}
        
        tripwire.sendEmail(2.5, datetime.date(2011, 8, 16))
        expected = r"""Called smtplib.SMTP('localhost')
Called smtplib.SMTP.sendmail(
    '*****@*****.**',
    ['*****@*****.**', '*****@*****.**'],
    'Content-Type: text/plain; charset="us-ascii"\nMIME-Version: 1.0\nContent-Transfer-Encoding: 7bit\nSubject: [DBTW] Tripwire Report 2011-08-16\nTo: [email protected], [email protected]\nFrom: [email protected]\n\nDatabase trip wire has been tripped.\n\nRMS value was: 2.5\n')
Called smtplib.SMTP.quit()"""
        self.assertTrue(tt.check(expected), tt.diff(expected))
        
        minimock.restore()
Пример #35
0
    def setUp(self):
        "set up database test framework and test for db connection"
        self.options = Namespace()
        self.options.BioSQLconfig = Namespace()
        self.options.BioSQLconfig.dbdriver = 'psycopg2'
        self.options.BioSQLconfig.dbuser = '******'
        self.options.BioSQLconfig.dbpass = '******'
        self.options.BioSQLconfig.dbhost = 'localhost'
        self.options.BioSQLconfig.dbport = '5432'
        self.options.BioSQLconfig.dbdb = 'antiSMASHnosetest'
        self.options.BioSQLnamespace = 'test'

        self.trace_tracker = TraceTracker()

        # test correct method call for db.asDB)()
        mock('BioSeqDatabase.open_database',
             tracker=self.trace_tracker,
             returns=['mock result'])

        expected = """Called BioSeqDatabase.open_database(
        db='antiSMASHnosetest',
        driver='psycopg2',
        host='localhost',
        passwd='biosql',
        port='5432',
        user='******')
        """

        mydb = db.aSDB(self.options)
        assert_same_trace(self.trace_tracker, expected)

        # restore mocked objects that other tests can run with the original code
        restore()
        self.trace_tracker = TraceTracker()

        #test initializing database object db.__init__"

        try:
            mydb = db.aSDB(self.options)
        except Exception:
            self.skipTest(
                'database connection could not be established, skipping tests')

        self.assertIsInstance(mydb, db.aSDB)

        self.mydb = mydb
Пример #36
0
    def test__init(self):
        "Test active_site_finder.__init__ method"

        self.tt = TraceTracker()

        class dummyET:
            def __init__(self):
                pass

            def getroot(self):
                return dummyET()

            def findall(self, a):
                return "ET-Subtree"

        mock('ET.parse', returns=dummyET(), tracker=self.tt)
        #mock('ET.getroot', returns="XMLTree", tracker=self.tt)
        #mock('ET.findall', returns="XMLSubtree", tracker=self.tt)

        mock(
            'antismash.generic_modules.active_site_finder.active_site_finder.check_prereqs',
            returns=[],
            tracker=self.tt)

        expected = """    Called ET.parse(
        '.../antismash/generic_modules/active_site_finder/config/SignatureResources.xml')"""

        ASF = antismash.generic_modules.active_site_finder.active_site_finder(
            self.seq_record, self.options)

        assert_same_trace(self.tt, expected)

        # restore mocked objects that other tests can run with the original code
        restore()
        self.tt = TraceTracker()

        # now acutally generate ASF object
        myASF = antismash.generic_modules.active_site_finder.active_site_finder(
            self.seq_record, self.options)

        assert_is_instance(
            myASF,
            antismash.generic_modules.active_site_finder.active_site_finder)

        self.my_ASF = myASF
Пример #37
0
    def setUp(self):
        "set up database connection"
        # restore all mocked methods
        restore()

        #Open testfile and parse it with Biopython
        testfile = path.join(path.dirname(__file__), "AAA26493.embl")
        testfile_fh = open(testfile, "rU")
        self.seq_record = SeqIO.read(testfile_fh, 'embl')

        self.options = Namespace()
        self.options.BioSQLconfig = Namespace()
        self.options.BioSQLconfig.dbdriver = 'psycopg2'
        self.options.BioSQLconfig.dbuser = '******'
        self.options.BioSQLconfig.dbpass = '******'
        self.options.BioSQLconfig.dbhost = 'localhost'
        self.options.BioSQLconfig.dbport = '5432'
        self.options.BioSQLconfig.dbdb = 'antiSMASHnosetest'
        self.options.BioSQLnamespace = 'test'
        self.options.dbnamespace = self.options.BioSQLnamespace
        self.trace_tracker = TraceTracker()

        try:
            mydb = db.aSDB(self.options)
        except Exception:
            self.skipTest(
                'database connection could not be established, skipping tests')

        if not mydb:
            self.skipTest('database connection impossible; skipping tests')

        try:
            del mydb.server[self.options.BioSQLnamespace]
            mydb.commit()
        except Exception:
            pass

        mydb.generate_namespace(self.options.BioSQLnamespace,
                                'test namespace for nosetests')

        mydb.connect(self.options.BioSQLnamespace)

        rec_no = mydb.load_records([self.seq_record])
        mydb.commit()
        self.mydb = mydb
Пример #38
0
def urlopen_stub(url, data=None, timeout=30):
    """Mock urllib2.urlopen and return a local file handle or create file if not existent and then return it."""

    if isinstance(url, urllib2.Request):
        key = md5(url.get_full_url()).hexdigest()
    else:
        key = md5(url).hexdigest()
    data_file = os.path.join(os.path.dirname(__file__), 'cache', "%s.xml" % key)
    try:
        f = open(data_file)
    except IOError:
        restore()
        data = urllib2.urlopen(url, data).read()
        # re-mock it.
        mock('urllib2.urlopen', returns_func=urlopen_stub, tracker=None)
        f = open(data_file, "w").write(data)
        f = open(data_file, "r")
    return f
Пример #39
0
 def test_sends_log_line_data_as_json(self):
     minimock.restore()
     self.user.timezone_name = 'US/Pacific'
     self.user.put()
     self.tracker = minimock.TraceTracker()
     client_id = '{0}.{1}.1234.5678'.format(self.server.key.id(), self.user.key.id())
     minimock.mock('channel.ServerChannels.get_client_ids', returns=[client_id], tracker=None)
     minimock.mock('gae_channel.send_message', tracker=self.tracker)
     js = '{"username": "******", "achievement_message": null, "chat": "is there anybody in there?", "time": "05:01pm", "date": "Mar 23, 2013", "event": "chat", "death_message": null}'
     self.interesting_log_line.send_message()
     minimock.assert_same_trace(self.tracker, "Called gae_channel.send_message('{0}','{1}')".format(client_id, js))
     log_line = json.loads(js)
     self.assertEqual('chat', log_line['event'])
     self.assertEqual('Mar 23, 2013', log_line['date'])
     self.assertEqual('05:01pm', log_line['time'])
     self.assertEqual('quazifene', log_line['username'])
     self.assertEqual('is there anybody in there?', log_line['chat'])
     self.assertEqual(None, log_line['death_message'])
Пример #40
0
def mock_get(url, **kwargs):
    """
    Mock requests.get with the contents of a local file. If the file doesn't
    exist, make the request and save to the file.
    """
    key = hashlib.md5(url.encode('utf-8')).hexdigest()
    file_path = os.path.join(test_dir, 'cache', '{0}.xml'.format(key))
    try:
        return MockResponse(file_path)
    except IOError:
        restore()  # restore normal function
        r = requests.get(url)
        with open(file_path, 'w') as tmp:
            try:
                tmp.write(r.text)
            except UnicodeEncodeError:
                tmp.write(r.content)  # python 2
        mock('requests.get', returns_func=mock_get, tracker=None)  # re-mock it.
        return MockResponse(file_path)
Пример #41
0
 def test_sends_message_to_all_connected_clients(self):
     minimock.restore()
     self.tracker = minimock.TraceTracker()
     models.User.create_user('4321', email='*****@*****.**')
     user = models.User.lookup(email='*****@*****.**')
     client_id = '{0}.{1}.1234.5678'.format(self.server.key.id(),
                                            self.user.key.id())
     client_id2 = '{0}.{1}.1234.5678'.format(self.server.key.id(),
                                             user.key.id())
     minimock.mock('channel.ServerChannels.get_client_ids',
                   returns=[client_id, client_id2],
                   tracker=None)
     minimock.mock('gae_channel.send_message', tracker=self.tracker)
     self.interesting_log_line.send_message()
     js = '{"username": "******", "achievement_message": null, "chat": "is there anybody in there?", "time": "12:01am", "date": "Mar 24, 2013", "event": "chat", "death_message": null}'
     trace = "Called gae_channel.send_message(\n    '{0}',\n    '{1}')".format(
         client_id, js)
     trace += "\nCalled gae_channel.send_message(\n    '{0}',\n    '{1}')".format(
         client_id, js)
     minimock.assert_same_trace(self.tracker, trace)
Пример #42
0
def mock_get(url, **kwargs):
    """
    Mock requets.get and return a local file handle or create file if
    not existent and then return it.
    """
    key = md5(url).hexdigest()
    file_path = join(test_dir, 'cache', '{0}.xml'.format(key))
    try:
        f = MockFile(file_path, 'r')
    except IOError:
        restore() # restore normal function
        tmp = MockFile(file_path, 'w')
        tmp.write(requests.get(url).content)
        tmp.close()
        mock('requests.get', returns_func=mock_get, tracker=None) # re-mock it.
        f = MockFile(file_path, 'r')

    f.populate_content()
    f.status_code = requests.codes.ok
    if not f.content:
        f.status_code = requests.codes.not_found
    return f
Пример #43
0
    def test_fail_at_acquireWriteLock(self):
        self.setUpContext()
        mock('ReadWriteLock.acquireWriteLock', raises=Exception, tracker=None)
        self.assertRaises(Exception, self.context.dispatch, 'pseudo-init',
                          self.obj)
        self.assertEqual(0, _FantasmFanIn.all().count())
        self.assertEqual(
            None,
            memcache.get(
                'foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
        restore()

        self.setUpContext(retryCount=1)
        self.context.dispatch('pseudo-init', self.obj)
        self.assertEqual(1, _FantasmFanIn.all().count())
        self.assertEqual(
            'foo--InitialState--ok--FanInState--step-2-2957927341',
            _FantasmFanIn.all().get().workIndex)
        self.assertEqual(
            '65536',
            memcache.get(
                'foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
Пример #44
0
    def test_fail_at_put(self):
        self.setUpContext()
        mock('db.put', raises=Exception, tracker=None)
        self.assertRaises(Exception, self.context.dispatch, 'pseudo-init',
                          copy.copy(self.obj))
        self.assertEqual(0, _FantasmFanIn.all().count())
        # notice the +1 extra on the lock
        self.assertEqual(
            '65537',
            memcache.get(
                'foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
        restore()

        self.setUpContext(retryCount=1)
        self.context.dispatch('pseudo-init', self.obj)
        self.assertEqual(1, _FantasmFanIn.all().count())
        self.assertEqual(
            'foo--InitialState--ok--FanInState--step-2-2957927341',
            _FantasmFanIn.all().get().workIndex)
        self.assertEqual(
            '65537',
            memcache.get(
                'foo--InitialState--ok--FanInState--step-2-lock-3255389373'))
Пример #45
0
    def test_get_record(self):
        "Test get_record() function"

        Fake_seq_rec = Namespace()
        Fake_seq_rec.name = 'fakedName'

        mock('db.aSDB.get_record_by_name',
             tracker=self.trace_tracker,
             returns=Fake_seq_rec)

        db_seq_rec = db.get_record('AAA26493', self.options)
        expected = "Called db.aSDB.get_record_by_name('AAA26493')"
        assert_same_trace(self.trace_tracker, expected)
        restore()
        self.trace_tracker = TraceTracker()

        db_seq_rec = db.get_record('AAA26493', self.options)
        myrecord_class = db_seq_rec.__class__.__name__

        self.assertEqual(myrecord_class, "DBSeqRecord")
        self.assertEqual(str(db_seq_rec.seq), str(self.seq_record.seq))
        self.assertEqual(db_seq_rec.id, self.seq_record.id)
        self.assertEqual(db_seq_rec.name, self.seq_record.name)
Пример #46
0
 def test_sends_log_line_data_as_json(self):
     minimock.restore()
     self.user.timezone_name = 'US/Pacific'
     self.user.put()
     self.tracker = minimock.TraceTracker()
     client_id = '{0}.{1}.1234.5678'.format(self.server.key.id(),
                                            self.user.key.id())
     minimock.mock('channel.ServerChannels.get_client_ids',
                   returns=[client_id],
                   tracker=None)
     minimock.mock('gae_channel.send_message', tracker=self.tracker)
     js = '{"username": "******", "achievement_message": null, "chat": "is there anybody in there?", "time": "05:01pm", "date": "Mar 23, 2013", "event": "chat", "death_message": null}'
     self.interesting_log_line.send_message()
     minimock.assert_same_trace(
         self.tracker,
         "Called gae_channel.send_message('{0}','{1}')".format(
             client_id, js))
     log_line = json.loads(js)
     self.assertEqual('chat', log_line['event'])
     self.assertEqual('Mar 23, 2013', log_line['date'])
     self.assertEqual('05:01pm', log_line['time'])
     self.assertEqual('quazifene', log_line['username'])
     self.assertEqual('is there anybody in there?', log_line['chat'])
     self.assertEqual(None, log_line['death_message'])
Пример #47
0
 def tearDown(self):
     restore()
     # pylint: disable-msg=W0212
     ReadWriteLock.BUSY_WAIT_ITER_SECS = ReadWriteLock._BUSY_WAIT_ITER_SECS
     ReadWriteLock.BUSY_WAIT_ITERS = ReadWriteLock._BUSY_WAIT_ITERS
     super(ReadWriteLockTest, self).tearDown()
Пример #48
0
 def tearDown(self):
     restore()
     super(RunOnceSemaphoreTest, self).tearDown()
Пример #49
0
 def tearDown(self):
     restore()
Пример #50
0
 def tearDown(self):
     restore()
     super(FanInQueueDispatchTest, self).tearDown()
Пример #51
0
 def tearDown(self):
     restore()
     super(FanInMergeJoinDispatchTest, self).tearDown()
Пример #52
0
 def tearDown(self):
     super(FanInTxnException, self).tearDown()
     restore()
Пример #53
0
 def tearDown(self):
     super(TaskDoubleExecutionTest, self).tearDown()
     restore()
Пример #54
0
 def teardown(self):
     restore()
Пример #55
0
    def test_defaultFetcher(self):
        """util._defaultFetcher"""
        if mock:

            class Response(object):
                """urllib2.Reponse mock"""
                def __init__(self, url,
                             contenttype, content,
                             exception=None, args=None):
                    self.url = url

                    mt, params = cgi.parse_header(contenttype)
                    self.mimetype = mt
                    self.charset = params.get('charset', None)

                    self.text = content

                    self.exception = exception
                    self.args = args

                def geturl(self):
                    return self.url

                def info(self):
                    mimetype, charset = self.mimetype, self.charset
                    class Info(object):
                        def gettype(self):
                            return mimetype
                        def getparam(self, name):
                            return charset

                    return Info()

                def read(self):
                    # returns fake text or raises fake exception
                    if not self.exception:
                        return self.text
                    else:
                        raise self.exception(*self.args)

            def urlopen(url,
                        contenttype=None, content=None,
                        exception=None, args=None):
                # return an mock which returns parameterized Response
                def x(*ignored):
                    if exception:
                        raise exception(*args)
                    else:
                        return Response(url,
                                        contenttype, content,
                                        exception=exception, args=args)
                return x

            # positive tests
            tests = {
                # content-type, contentstr: encoding, contentstr
                ('text/css', u'€'.encode('utf-8')):
                        (None, u'€'.encode('utf-8')),
                ('text/css;charset=utf-8', u'€'.encode('utf-8')):
                        ('utf-8', u'€'.encode('utf-8')),
                ('text/css;charset=ascii', 'a'):
                        ('ascii', 'a')
            }
            url = 'http://example.com/test.css'
            for (contenttype, content), exp in tests.items():

                mock("urllib2.urlopen", mock_obj=urlopen(url, contenttype, content))

                #print url, exp == _readUrl(url, encoding), exp, _readUrl(url, encoding)
                self.assertEqual(exp, _defaultFetcher(url))

            # wrong mimetype
            mock("urllib2.urlopen", mock_obj=urlopen(url, 'text/html', 'a'))
            self.assertRaises(ValueError, _defaultFetcher, url)

            # calling url results in fake exception
            tests = {
                '1': (ValueError, ['invalid value for url']),
                'e2': (urllib2.HTTPError, ['u', 500, 'server error', {}, None]),
                #_readUrl('http://cthedot.de/__UNKNOWN__.css')
                'e3': (urllib2.HTTPError, ['u', 404, 'not found', {}, None]),
                #_readUrl('mailto:a.css')
                'mailto:e4': (urllib2.URLError, ['urlerror']),
                # cannot resolve x, IOError
                'http://x': (urllib2.URLError, ['ioerror']),
            }
            for url, (exception, args) in tests.items():
                mock("urllib2.urlopen",
                        mock_obj=urlopen(url, exception=exception, args=args))
                self.assertRaises(exception, _defaultFetcher, url)

            restore()
Пример #56
0
 def tearDown(self):
     minimock.restore()