From 4df2b77fdf42046fe19bd371be7c7255986b5980 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Tue, 6 Mar 2012 13:36:39 +0100
Subject: renamed hooks to addons, new filemanager and database, many new api
 methods you will loose ALL your LINKS, webinterface will NOT work

---
 tests/test_database.py | 171 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 171 insertions(+)
 create mode 100644 tests/test_database.py

(limited to 'tests/test_database.py')

diff --git a/tests/test_database.py b/tests/test_database.py
new file mode 100644
index 000000000..9fe7796a4
--- /dev/null
+++ b/tests/test_database.py
@@ -0,0 +1,171 @@
+# -*- coding: utf-8 -*-
+
+from collections import defaultdict
+
+from helper.Stubs import Core
+from helper.BenchmarkTest import BenchmarkTest
+
+from module.database import DatabaseBackend
+
+# disable asyncronous queries
+DatabaseBackend.async = DatabaseBackend.queue
+
+from random import choice
+
+class TestDatabase(BenchmarkTest):
+    bench = ["insert", "insert_links", "insert_many", "get_packages",
+             "get_files", "get_files_queued", "get_package_childs", "get_package_files",
+             "get_package_data", "get_file_data", "find_files", "collector", "purge"]
+    pids = None
+    fids = None
+
+    @classmethod
+    def setUpClass(cls):
+        cls.pids = [-1]
+        cls.fids = []
+
+        cls.db = DatabaseBackend(Core())
+        cls.db.manager = cls.db.core
+        cls.db.manager.statusMsg = defaultdict(lambda: "statusmsg")
+
+        cls.db.setup()
+        cls.db.purgeAll()
+
+    @classmethod
+    def tearDownClass(cls):
+        cls.db.purgeAll()
+        cls.db.shutdown()
+
+    # benchmarker ignore setup
+    def setUp(self):
+        self.db.purgeAll()
+        self.pids = [-1]
+        self.fids = []
+
+        self.test_insert(20)
+        self.test_insert_many()
+        self.fids = self.db.getAllFiles().keys()
+
+
+    def test_insert(self, n=200):
+        for i in range(n):
+            pid = self.db.addPackage("name", "folder", choice(self.pids), "password", "site", "comment", 0)
+            self.pids.append(pid)
+
+    def test_insert_links(self):
+        for i in range(10000):
+            fid = self.db.addLink("url %s" % i, "name", "plugin", choice(self.pids))
+            self.fids.append(fid)
+
+    def test_insert_many(self):
+        for pid in self.pids:
+            self.db.addLinks([("url %s" % i, "plugin") for i in range(50)], pid)
+
+    def test_get_packages(self):
+        packs = self.db.getAllPackages()
+        n = len(packs)
+        assert n == len(self.pids) -1
+
+        print "Fetched %d packages" % n
+        self.assert_pack(choice(packs.values()))
+
+    def test_get_files(self):
+        files = self.db.getAllFiles()
+        n = len(files)
+        assert n >= len(self.pids)
+
+        print "Fetched %d files" % n
+        self.assert_file(choice(files.values()))
+
+    def test_get_files_queued(self):
+        files = self.db.getAllFiles(unfinished=True)
+        print "Fetched %d files queued" % len(files)
+
+    def test_delete(self):
+        pid = choice(self.pids)
+        self.db.deletePackage(pid)
+        self.pids.remove(pid)
+
+    def test_get_package_childs(self):
+        pid = choice(self.pids)
+        packs = self.db.getAllPackages(root=pid)
+
+        print "Package %d has %d packages" % (pid, len(packs))
+        self.assert_pack(choice(packs.values()))
+
+    def test_get_package_files(self):
+        pid = choice(self.pids)
+        files = self.db.getAllFiles(package=pid)
+
+        print "Package %d has %d files" % (pid, len(files))
+        self.assert_file(choice(files.values()))
+
+    def test_get_package_data(self, stats=False):
+        pid = choice(self.pids)
+        p = self.db.getPackageInfo(pid, stats)
+        self.assert_pack(p)
+        # test again with stat
+        if not stats:
+            self.test_get_package_data(True)
+
+    def test_get_file_data(self):
+        fid = choice(self.fids)
+        f = self.db.getFileInfo(fid)
+        self.assert_file(f)
+
+    def test_find_files(self):
+        files = self.db.getAllFiles(search="1")
+        print "Found %s files" % len(files)
+        f = choice(files.values())
+
+        assert "1" in f.name
+
+    def test_collector(self):
+        self.db.deleteCollector()
+        assert not self.db.getCollector()
+
+        self.db.addCollector("plugin", "package", [("name", 0, 0, "url %d" % i) for i in range(10)])
+        coll = self.db.getCollector()
+        assert len(coll) == 10
+        assert coll[0].plugin == "plugin"
+        assert coll[0].packagename == "package"
+        assert coll[0].name == "name"
+        assert "url" in coll[0].url
+
+        self.db.deleteCollector(url="url 1")
+        assert len(self.db.getCollector()) == 9
+        self.db.deleteCollector(package="package")
+        assert not self.db.getCollector()
+
+    def test_purge(self):
+        self.db.purgeLinks()
+
+    def assert_file(self, f):
+        try:
+            assert f is not None
+            self.assert_int(f, ("fid", "status", "size", "media", "fileorder", "added", "package"))
+            assert f.status in range(5)
+            assert f.media in range(1024)
+            assert f.package in self.pids
+            assert f.added > 10 ** 6 # date is usually big integer
+        except:
+            print f
+            raise
+
+    def assert_pack(self, p):
+        try:
+            assert p is not None
+            self.assert_int(p, ("pid", "root", "added", "status", "packageorder"))
+            assert p.pid in self.pids
+            assert p.status in range(5)
+            assert p.root in self.pids
+            assert p.added > 10 ** 6
+        except:
+            print p
+            raise
+
+    def assert_int(self, obj, list):
+        for attr in list: assert type(getattr(obj, attr)) == int
+
+if __name__ == "__main__":
+    TestDatabase.benchmark()
\ No newline at end of file
-- 
cgit v1.2.3


From 829244a6140763712d50ed046c33f415f2b04301 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Tue, 15 May 2012 19:22:34 +0200
Subject: some multiuser db changes

---
 tests/test_database.py | 59 ++++++++++++++++++++++++++++++++------------------
 1 file changed, 38 insertions(+), 21 deletions(-)

(limited to 'tests/test_database.py')

diff --git a/tests/test_database.py b/tests/test_database.py
index 9fe7796a4..e3e6b9963 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -18,6 +18,8 @@ class TestDatabase(BenchmarkTest):
              "get_package_data", "get_file_data", "find_files", "collector", "purge"]
     pids = None
     fids = None
+    owner = 123
+    pstatus = 0
 
     @classmethod
     def setUpClass(cls):
@@ -49,22 +51,23 @@ class TestDatabase(BenchmarkTest):
 
     def test_insert(self, n=200):
         for i in range(n):
-            pid = self.db.addPackage("name", "folder", choice(self.pids), "password", "site", "comment", 0)
+            pid = self.db.addPackage("name", "folder", choice(self.pids), "password", "site", "comment", self.pstatus,
+                self.owner)
             self.pids.append(pid)
 
     def test_insert_links(self):
         for i in range(10000):
-            fid = self.db.addLink("url %s" % i, "name", "plugin", choice(self.pids))
+            fid = self.db.addLink("url %s" % i, "name", "plugin", choice(self.pids), self.owner)
             self.fids.append(fid)
 
     def test_insert_many(self):
         for pid in self.pids:
-            self.db.addLinks([("url %s" % i, "plugin") for i in range(50)], pid)
+            self.db.addLinks([("url %s" % i, "plugin") for i in range(50)], pid, self.owner)
 
     def test_get_packages(self):
         packs = self.db.getAllPackages()
         n = len(packs)
-        assert n == len(self.pids) -1
+        assert n == len(self.pids) - 1
 
         print "Fetched %d packages" % n
         self.assert_pack(choice(packs.values()))
@@ -121,30 +124,43 @@ class TestDatabase(BenchmarkTest):
         assert "1" in f.name
 
     def test_collector(self):
-        self.db.deleteCollector()
-        assert not self.db.getCollector()
-
-        self.db.addCollector("plugin", "package", [("name", 0, 0, "url %d" % i) for i in range(10)])
-        coll = self.db.getCollector()
-        assert len(coll) == 10
-        assert coll[0].plugin == "plugin"
-        assert coll[0].packagename == "package"
-        assert coll[0].name == "name"
-        assert "url" in coll[0].url
-
-        self.db.deleteCollector(url="url 1")
-        assert len(self.db.getCollector()) == 9
-        self.db.deleteCollector(package="package")
-        assert not self.db.getCollector()
+        self.db.saveCollector(0, "data")
+        assert self.db.retrieveCollector(0) == "data"
+        self.db.deleteCollector(0)
 
     def test_purge(self):
         self.db.purgeLinks()
 
+
+    def test_user_context(self):
+        self.db.purgeAll()
+
+        p1 = self.db.addPackage("name", "folder", 0, "password", "site", "comment", self.pstatus, 0)
+        self.db.addLink("url", "name", "plugin", p1, 0)
+        p2 = self.db.addPackage("name", "folder", 0, "password", "site", "comment", self.pstatus, 1)
+        self.db.addLink("url", "name", "plugin", p2, 1)
+
+        assert len(self.db.getAllPackages(owner=0)) == 1 == len(self.db.getAllFiles(owner=0))
+        assert len(self.db.getAllPackages(root=0, owner=0)) == 1 == len(self.db.getAllFiles(package=p1, owner=0))
+        assert len(self.db.getAllPackages(owner=1)) == 1 == len(self.db.getAllFiles(owner=1))
+        assert len(self.db.getAllPackages(root=0, owner=1)) == 1 == len(self.db.getAllFiles(package=p2, owner=1))
+        assert len(self.db.getAllPackages()) == 2 == len(self.db.getAllFiles())
+
+        self.db.deletePackage(p1, 1)
+        assert len(self.db.getAllPackages(owner=0)) == 1 == len(self.db.getAllFiles(owner=0))
+        self.db.deletePackage(p1, 0)
+        assert len(self.db.getAllPackages(owner=1)) == 1 == len(self.db.getAllFiles(owner=1))
+        self.db.deletePackage(p2)
+
+        assert len(self.db.getAllPackages()) == 0
+
+
     def assert_file(self, f):
         try:
             assert f is not None
-            self.assert_int(f, ("fid", "status", "size", "media", "fileorder", "added", "package"))
+            self.assert_int(f, ("fid", "status", "size", "media", "fileorder", "added", "package", "owner"))
             assert f.status in range(5)
+            assert f.owner == self.owner
             assert f.media in range(1024)
             assert f.package in self.pids
             assert f.added > 10 ** 6 # date is usually big integer
@@ -155,8 +171,9 @@ class TestDatabase(BenchmarkTest):
     def assert_pack(self, p):
         try:
             assert p is not None
-            self.assert_int(p, ("pid", "root", "added", "status", "packageorder"))
+            self.assert_int(p, ("pid", "root", "added", "status", "packageorder", "owner"))
             assert p.pid in self.pids
+            assert p.owner == self.owner
             assert p.status in range(5)
             assert p.root in self.pids
             assert p.added > 10 ** 6
-- 
cgit v1.2.3


From cbd4f4b5375f89362733e10a9b98e5a74c2a5734 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Fri, 31 Aug 2012 23:25:26 +0200
Subject: first js models/views

---
 tests/test_database.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'tests/test_database.py')

diff --git a/tests/test_database.py b/tests/test_database.py
index e3e6b9963..b7408e213 100644
--- a/tests/test_database.py
+++ b/tests/test_database.py
@@ -154,6 +154,12 @@ class TestDatabase(BenchmarkTest):
 
         assert len(self.db.getAllPackages()) == 0
 
+    def test_count(self):
+        self.db.purgeAll()
+
+        assert self.db.filecount() == 0
+        assert self.db.queuecount() == 0
+        assert self.db.proccesscount() == 0
 
     def assert_file(self, f):
         try:
-- 
cgit v1.2.3