1 files changed, 94 insertions, 0 deletions
diff --git a/tests/model-package.scm b/tests/model-package.scm
new file mode 100644
index 0000000..8029c3b
--- /dev/null
+++ b/tests/model-package.scm
@@ -0,0 +1,94 @@
+(define-module (test-model-package)
+  #:use-module (ice-9 match)
+  #:use-module (srfi srfi-1)
+  #:use-module (srfi srfi-64)
+  #:use-module (guix utils)
+  #:use-module (guix tests)
+  #:use-module (tests mock-inferior)
+  #:use-module (guix-data-service model license-set)
+  #:use-module (guix-data-service model package)
+  #:use-module (guix-data-service model package-metadata)
+  #:use-module (guix-data-service database))
+
+(test-begin "test-model-package")
+
+(define mock-inferior-package-foo
+  (mock-inferior-package
+   (name "foo")
+   (version "2")
+   (synopsis "Foo")
+   (description "Foo description")
+   (home-page "https://example.com")
+   (location (location "file.scm" 5 0))))
+
+(define mock-inferior-package-foo-2
+  (mock-inferior-package
+   (name "foo")
+   (version "2")
+   (synopsis "Foo")
+   (description "Foo description")
+   (home-page #f)
+   (location #f)))
+
+(define (test-license-set-ids conn)
+  (mock
+   ((guix-data-service model license)
+    inferior-packages->license-data
+    (lambda (inf packages)
+      '((("License 1"
+          "https://gnu.org/licenses/test-1.html"
+          "https://example.com/why-license-1")))))
+
+   (inferior-packages->license-set-ids conn #f #f)))
+
+(define mock-inferior-packages
+  (list mock-inferior-package-foo
+        mock-inferior-package-foo-2))
+
+(with-mock-inferior-packages
+ (lambda ()
+   (use-modules (guix-data-service model package)
+                (guix-data-service model git-repository)
+                (guix-data-service model guix-revision)
+                (guix-data-service model package-metadata))
+
+   (with-postgresql-connection
+    "test-model-package"
+    (lambda (conn)
+      (with-postgresql-transaction
+       conn
+       (lambda (conn)
+         (test-assert "inferior-packages->package-ids works once"
+           (let ((package-metadata-ids (inferior-packages->package-metadata-ids
+                                        conn
+                                        mock-inferior-packages
+                                        (test-license-set-ids conn))))
+             (match (inferior-packages->package-ids
+                     conn
+                     (zip (map mock-inferior-package-name mock-inferior-packages)
+                          (map mock-inferior-package-version mock-inferior-packages)
+                          package-metadata-ids))
+               ((x) (number? x))))))
+       #:always-rollback? #t)
+
+      (with-postgresql-transaction
+       conn
+       (lambda (conn)
+         (let ((package-metadata-ids (inferior-packages->package-metadata-ids
+                                      conn
+                                      mock-inferior-packages
+                                      (test-license-set-ids conn))))
+           (test-equal
+               (inferior-packages->package-ids
+                conn
+                (zip (map mock-inferior-package-name mock-inferior-packages)
+                     (map mock-inferior-package-version mock-inferior-packages)
+                     package-metadata-ids))
+             (inferior-packages->package-ids
+              conn
+              (zip (map mock-inferior-package-name mock-inferior-packages)
+                   (map mock-inferior-package-version mock-inferior-packages)
+                   package-metadata-ids)))))
+       #:always-rollback? #t)))))
+
+(test-end)