Add check() function

2 files changed, 9 insertions, 0 deletions

diff --git a/.SRCINFO b/.SRCINFO
index aa0a1f669459..d7f81b525b92 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -5,6 +5,7 @@ pkgbase = whipper-git
 	url = https://github.com/JoeLametta/whipper
 	arch = any
 	license = GPL3
+	checkdepends = python2-twisted
 	makedepends = git
 	makedepends = python2-gobject2
 	depends = accuraterip-checksum
diff --git a/PKGBUILD b/PKGBUILD
index 5d1ccc6276cc..dc9ff981cb84 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -29,6 +29,9 @@ makedepends=(
     'git'
     'python2-gobject2'
     )
+checkdepends=(
+    'python2-twisted'
+    )
 provides=("${pkgname%-git}")
 conflicts=("${pkgname%-git}" 'morituri')
 source=("git+${url}.git")
@@ -40,6 +43,11 @@ pkgver() {
     git describe --long --tags | sed 's/^v//;s/\([^-]*-g\)/r\1/;s/-/./g'
 }
 
+check() {
+    cd "$srcdir/${pkgname%-git}"
+    python2 -m unittest discover
+}
+
 package() {
     cd "$srcdir/${pkgname%-git}"
     python2 setup.py install --root="${pkgdir}"/ --optimize=1