class StorageutilsTest(unittest.TestCase):

    def setUpClass():
        # launch webserver
        app = tornado.web.Application([(r"/samples/.*", CreateTestServer(b"hello world!"))])
        app.listen(8017)

    def tearDownClass():
        # shutdown webserver
        time.sleep(0.5)
        tornado.ioloop.IOLoop.instance().stop()

    def setUp(self):
        self.storage = Storage(
            address="http://127.0.0.1:8017",
            user_id="user-1"
        )
        self.sample = StorageSample(
            filecontents=b"hello world!",
            source="Unknown",
            name="testfile.txt",
            date=rfc3339.now().isoformat(),
            tags=["malware","really nasty file ;)"],
            comment="just some test"
        )
        self.sha256 = self.sample.sha256()

    def test_0_submit(self):
        self.storage.submitSample(self.sample)

    def test_1_get(self):
        r = self.storage.getSample(self.sha256)
        self.assertTrue(r == self.sample.filecontents)
class StorageutilsTest(unittest.TestCase):
    def setUpClass():
        # launch webserver
        app = tornado.web.Application([(r"/samples/.*",
                                        CreateTestServer(b"hello world!"))])
        app.listen(8017)

    def tearDownClass():
        # shutdown webserver
        time.sleep(0.5)
        tornado.ioloop.IOLoop.instance().stop()

    def setUp(self):
        self.storage = Storage(address="http://127.0.0.1:8017",
                               user_id="user-1")
        self.sample = StorageSample(filecontents=b"hello world!",
                                    source="Unknown",
                                    name="testfile.txt",
                                    date=rfc3339.now().isoformat(),
                                    tags=["malware", "really nasty file ;)"],
                                    comment="just some test")
        self.sha256 = self.sample.sha256()

    def test_0_submit(self):
        self.storage.submitSample(self.sample)

    def test_1_get(self):
        r = self.storage.getSample(self.sha256)
        self.assertTrue(r == self.sample.filecontents)
 def setUp(self):
     self.storage = Storage(address="http://127.0.0.1:8017",
                            user_id="user-1")
     self.sample = StorageSample(filecontents=b"hello world!",
                                 source="Unknown",
                                 name="testfile.txt",
                                 date=rfc3339.now().isoformat(),
                                 tags=["malware", "really nasty file ;)"],
                                 comment="just some test")
     self.sha256 = self.sample.sha256()
 def setUp(self):
     self.storage = Storage(
         address="http://127.0.0.1:8017",
         user_id="user-1"
     )
     self.sample = StorageSample(
         filecontents=b"hello world!",
         source="Unknown",
         name="testfile.txt",
         date=rfc3339.now().isoformat(),
         tags=["malware","really nasty file ;)"],
         comment="just some test"
     )
     self.sha256 = self.sample.sha256()