def _common_config_setup(self):
     mock_logging = Mock()
     required_config = FSRadixTreeStorage.get_required_config()
     required_config.add_option('logger', default=mock_logging)
     config_manager = ConfigurationManager(
         [required_config],
         app_name='testapp',
         app_version='1.0',
         app_description='app description',
         values_source_list=[{
             'logger': mock_logging
         }],
         argv_source=[])
     return config_manager
 def _common_config_setup(self):
     mock_logging = Mock()
     required_config = FSRadixTreeStorage.get_required_config()
     required_config.add_option('logger', default=mock_logging)
     config_manager = ConfigurationManager(
       [required_config],
       app_name='testapp',
       app_version='1.0',
       app_description='app description',
       values_source_list=[{
         'logger': mock_logging
       }],
       argv_source=[]
     )
     return config_manager
 def setUp(self):
     super(TestFSRadixTreeStorage, self).setUp()
     with self._common_config_setup().context() as config:
         self.fsrts = FSRadixTreeStorage(config)
class TestFSRadixTreeStorage(TestCase):
    CRASH_ID_1 = "0bba929f-8721-460c-dead-a43c20071025"
    CRASH_ID_2 = "0bba929f-8721-460c-dead-a43c20071026"
    CRASH_ID_3 = "0bba929f-8721-460c-dead-a43c20071027"

    def setUp(self):
        super(TestFSRadixTreeStorage, self).setUp()
        with self._common_config_setup().context() as config:
            self.fsrts = FSRadixTreeStorage(config)

    def tearDown(self):
        super(TestFSRadixTreeStorage, self).tearDown()
        shutil.rmtree(self.fsrts.config.fs_root)

    def _common_config_setup(self):
        mock_logging = Mock()
        required_config = FSRadixTreeStorage.get_required_config()
        required_config.add_option('logger', default=mock_logging)
        config_manager = ConfigurationManager(
            [required_config],
            app_name='testapp',
            app_version='1.0',
            app_description='app description',
            values_source_list=[{
                'logger': mock_logging
            }],
            argv_source=[])
        return config_manager

    def _make_test_crash(self, crash_id=CRASH_ID_1):
        self.fsrts.save_raw_crash({"test": "TEST"},
                                  MemoryDumpsMapping({
                                      'foo':
                                      'bar',
                                      self.fsrts.config.dump_field:
                                      'baz'
                                  }), crash_id)

    def _make_processed_test_crash(self):
        self.fsrts.save_processed({
            "uuid": self.CRASH_ID_2,
            "test": "TEST",
            "email": "should not exist"
        })

    def test_save_raw_crash(self):
        self._make_test_crash()
        ok_(
            os.path.exists(
                self.fsrts._get_radixed_parent_directory(self.CRASH_ID_1)))

    def test_save_processed(self):
        self._make_processed_test_crash()
        ok_(
            os.path.exists(
                os.path.join(
                    self.fsrts._get_radixed_parent_directory(self.CRASH_ID_2),
                    self.CRASH_ID_2 + self.fsrts.config.jsonz_file_suffix)))

    def test_get_raw_crash(self):
        self._make_test_crash()
        eq_(self.fsrts.get_raw_crash(self.CRASH_ID_1)['test'], "TEST")
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_crash,
                      self.CRASH_ID_2)

    def test_get_unredacted_processed_crash(self):
        self._make_processed_test_crash()
        eq_(
            self.fsrts.get_unredacted_processed(self.CRASH_ID_2)['test'],
            "TEST")
        ok_('email' in self.fsrts.get_unredacted_processed(self.CRASH_ID_2))
        assert_raises(CrashIDNotFound, self.fsrts.get_unredacted_processed,
                      self.CRASH_ID_1)

    def test_get_processed_crash(self):
        self._make_processed_test_crash()
        eq_(self.fsrts.get_processed(self.CRASH_ID_2)['test'], "TEST")
        ok_('email' not in self.fsrts.get_processed(self.CRASH_ID_2))
        assert_raises(CrashIDNotFound, self.fsrts.get_unredacted_processed,
                      self.CRASH_ID_1)

    def test_get_raw_dump(self):
        self._make_test_crash()
        eq_(self.fsrts.get_raw_dump(self.CRASH_ID_1, 'foo'), "bar")
        eq_(
            self.fsrts.get_raw_dump(self.CRASH_ID_1,
                                    self.fsrts.config.dump_field), "baz")
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_dump,
                      self.CRASH_ID_2, "foo")
        assert_raises(IOError, self.fsrts.get_raw_dump, self.CRASH_ID_1,
                      "foor")

    def test_get_raw_dumps(self):
        self._make_test_crash()
        eq_(
            self.fsrts.get_raw_dumps(self.CRASH_ID_1),
            MemoryDumpsMapping({
                'foo': 'bar',
                self.fsrts.config.dump_field: 'baz'
            }),
        )
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_dumps,
                      self.CRASH_ID_2)

    def test_remove(self):
        self._make_test_crash()
        self._make_test_crash(self.CRASH_ID_3)
        self.fsrts.remove(self.CRASH_ID_1)
        self.fsrts.remove(self.CRASH_ID_3)
        ok_(not os.path.exists(
            self.fsrts._get_radixed_parent_directory(self.CRASH_ID_1)))
        assert_raises(CrashIDNotFound, self.fsrts.remove, self.CRASH_ID_2)
 def setUp(self):
     super(TestFSRadixTreeStorage, self).setUp()
     with self._common_config_setup().context() as config:
         self.fsrts = FSRadixTreeStorage(config)
class TestFSRadixTreeStorage(TestCase):
    CRASH_ID_1 = "0bba929f-8721-460c-dead-a43c20071025"
    CRASH_ID_2 = "0bba929f-8721-460c-dead-a43c20071026"
    CRASH_ID_3 = "0bba929f-8721-460c-dead-a43c20071027"

    def setUp(self):
        super(TestFSRadixTreeStorage, self).setUp()
        with self._common_config_setup().context() as config:
            self.fsrts = FSRadixTreeStorage(config)

    def tearDown(self):
        super(TestFSRadixTreeStorage, self).tearDown()
        shutil.rmtree(self.fsrts.config.fs_root)

    def _common_config_setup(self):
        mock_logging = Mock()
        required_config = FSRadixTreeStorage.get_required_config()
        required_config.add_option('logger', default=mock_logging)
        config_manager = ConfigurationManager(
          [required_config],
          app_name='testapp',
          app_version='1.0',
          app_description='app description',
          values_source_list=[{
            'logger': mock_logging
          }],
          argv_source=[]
        )
        return config_manager

    def _make_test_crash(self, crash_id=CRASH_ID_1):
        self.fsrts.save_raw_crash({
            "test": "TEST"
        }, MemoryDumpsMapping({
            'foo': 'bar',
            self.fsrts.config.dump_field: 'baz'
        }), crash_id)

    def _make_processed_test_crash(self):
        self.fsrts.save_processed({
            "uuid": self.CRASH_ID_2,
            "test": "TEST",
            "email": "should not exist"
        })

    def test_save_raw_crash(self):
        self._make_test_crash()
        ok_(os.path.exists(
            self.fsrts._get_radixed_parent_directory(self.CRASH_ID_1)))

    def test_save_processed(self):
        self._make_processed_test_crash()
        ok_(os.path.exists(
            os.path.join(
                self.fsrts._get_radixed_parent_directory(self.CRASH_ID_2),
                self.CRASH_ID_2 + self.fsrts.config.jsonz_file_suffix)))

    def test_get_raw_crash(self):
        self._make_test_crash()
        eq_(self.fsrts.get_raw_crash(self.CRASH_ID_1)['test'],
                         "TEST")
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_crash,
                          self.CRASH_ID_2)

    def test_get_unredacted_processed_crash(self):
        self._make_processed_test_crash()
        eq_(self.fsrts.get_unredacted_processed(self.CRASH_ID_2)['test'],
                         "TEST")
        ok_('email' in
                        self.fsrts.get_unredacted_processed(self.CRASH_ID_2))
        assert_raises(CrashIDNotFound, self.fsrts.get_unredacted_processed,
                          self.CRASH_ID_1)

    def test_get_processed_crash(self):
        self._make_processed_test_crash()
        eq_(self.fsrts.get_processed(self.CRASH_ID_2)['test'],
                         "TEST")
        ok_('email' not in
                        self.fsrts.get_processed(self.CRASH_ID_2))
        assert_raises(CrashIDNotFound, self.fsrts.get_unredacted_processed,
                          self.CRASH_ID_1)

    def test_get_raw_dump(self):
        self._make_test_crash()
        eq_(self.fsrts.get_raw_dump(self.CRASH_ID_1, 'foo'),
                        "bar")
        eq_(self.fsrts.get_raw_dump(self.CRASH_ID_1,
                                                 self.fsrts.config.dump_field),
                         "baz")
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_dump,
                          self.CRASH_ID_2, "foo")
        assert_raises(IOError, self.fsrts.get_raw_dump,
                          self.CRASH_ID_1, "foor")

    def test_get_raw_dumps(self):
        self._make_test_crash()
        eq_(
            self.fsrts.get_raw_dumps(self.CRASH_ID_1),
            MemoryDumpsMapping({
                'foo': 'bar',
                self.fsrts.config.dump_field: 'baz'
            }),
        )
        assert_raises(CrashIDNotFound, self.fsrts.get_raw_dumps,
                          self.CRASH_ID_2)

    def test_remove(self):
        self._make_test_crash()
        self._make_test_crash(self.CRASH_ID_3)
        self.fsrts.remove(self.CRASH_ID_1)
        self.fsrts.remove(self.CRASH_ID_3)
        ok_(not os.path.exists(
            self.fsrts._get_radixed_parent_directory(self.CRASH_ID_1)))
        assert_raises(CrashIDNotFound, self.fsrts.remove,
                          self.CRASH_ID_2)