From 6c98c2f7c58323b6b94a9fcb373354321c16805d Mon Sep 17 00:00:00 2001
From: ben <benjaminlgreenman@gmail.com>
Date: Wed, 9 Mar 2016 23:07:01 -0500
Subject: [PATCH] [db] reasonable tests

---
 private/db.rkt        |   1 +
 private/db/schema.rkt |   8 +++
 test/db-pass.rkt      | 121 +++++++++++++++++++++++++++++++++++-------
 3 files changed, 111 insertions(+), 19 deletions(-)

diff --git a/private/db.rkt b/private/db.rkt
index f5ebd4e..358261b 100644
--- a/private/db.rkt
+++ b/private/db.rkt
@@ -4,6 +4,7 @@
 
 (provide
   (all-from-out db)
+  Connection
 
   ;(rename-out [quasiquote DB]) ;; TODO try using struct types
 
diff --git a/private/db/schema.rkt b/private/db/schema.rkt
index f6ac43f..ab5132c 100644
--- a/private/db/schema.rkt
+++ b/private/db/schema.rkt
@@ -120,6 +120,14 @@
         (syntax->datum stx)))
     (car typ+num)))
 
+;; -----------------------------------------------------------------------------
+;; TODO , but it's more work than I can do now (2016-03-09)
+;(define (schema->sql schema)
+;  (map tbl-schema->sql schema))
+;
+;(define (tbl-schema->sql tbl-schema)
+;  (format "CREATE TABLE ~a" 'foo))
+
 ;; =============================================================================
 
 (module+ test
diff --git a/test/db-pass.rkt b/test/db-pass.rkt
index 9d54087..00eb5e4 100644
--- a/test/db-pass.rkt
+++ b/test/db-pass.rkt
@@ -1,33 +1,116 @@
 #lang typed/racket/base
 
-;(module+ test
+;; To run these test:
+;; - Install postgres, start server
+;; - Create superuser 'postgres'
+;; - Create database 'travis_ci_test'
+;; Then you can `raco test ...` as usual
+
+(module+ test
   (require
+    typed/rackunit
     trivial/vector
     trivial/private/db)
+
+  (require/typed db
+    (start-transaction (-> Connection Void))
+    (rollback-transaction (-> Connection Void))
+    (query-exec (-> Connection String Any * Void)))
+
   (define-type Id Natural)
-  (define-schema: M '((word
-                       [(id   . Id)
-                        (word . String)])
-                      (word_syllables
-                       [(word      . Id)
-                        (syllables . Id)])))
-  (define-connection: c (postgresql-connect: M #:user "ben" #:database "ipoe"))
 
-  (define r0 (query-row: c "SELECT * FROM word LIMIT 1"))
-  (ann r0 (Vector Id String))
+  ;; -- create fake database
+  (define fish-sql
+    (string-append
+      "CREATE TABLE fish ( "
+      "id serial PRIMARY KEY, "
+      "name text UNIQUE NOT NULL, "
+      "weight int NOT NULL);"))
 
-  (define r1 (query-row: c "SELECT * FROM word WHERE word.word = \"blossom\""))
-  (ann r1 (Vector Id String))
+  (define cube-sql
+    (string-append
+      "CREATE TABLE cube ( "
+      "id serial PRIMARY KEY, "
+      "length smallint NOT NULL, "
+      "width integer NOT NULL, "
+      "height bigint NOT NULL);"))
 
-  (define r2 (query-row: c "SELECT * FROM word WHERE word.word = $1" "blossom"))
-  (ann r2 (Vector Id String))
+  (define-schema: schema
+    '((fish ((id . Id) (name . String) (weight . Integer)))
+      (cube ((id . Id) (length . Integer) (width . Integer) (height . Integer)))))
 
-  (define-vector: r3 (query-row: c "SELECT id FROM word WHERE word.id = 1"))
-  (ann r3 (Vector Id))
+  (define-connection: conn (postgresql-connect: schema #:user "postgres" #:database "travis_ci_test"))
 
-  ;(vector-ref: r3 2)
-  ;(vector-ref: (vector-map: add1 r3) 3) ;; Yes
+  (struct fish ([name : String] [weight : Integer]))
+  (define-type Fish fish)
+  (struct cube ([length : Integer] [width : Integer] [height : Integer]))
+  (define-type Cube cube)
+
+  (: insert-fish (-> Fish Void))
+  (define (insert-fish f)
+    (query-exec conn "INSERT INTO fish (name, weight) VALUES ($1, $2);"
+      (fish-name f) (fish-weight f)))
+
+  (: insert-cube (-> Cube Void))
+  (define (insert-cube c)
+    (query-exec conn "INSERT INTO cube (length, width, height) VALUES ($1, $2, $3);"
+      (cube-length c) (cube-width c) (cube-height c)))
+
+  (: with-transaction (-> (-> Any) Void))
+  (define (with-transaction thunk)
+    (begin
+      (define maybe-exn
+        (with-handlers ([exn? (lambda ([e : exn]) e)])
+          (start-transaction conn)
+          (thunk)
+          #f))
+      (rollback-transaction conn)
+      (if maybe-exn (raise maybe-exn) (void))))
+
+  (with-transaction (lambda ()
+  ;; ---------------------------------------------------------------------------
+  ;; Insert some things
+    (query-exec conn fish-sql)
+    (query-exec conn cube-sql)
+
+    (define f1 (fish "Swordfish" 432))
+    (define f2 (fish "Tuna" 9999))
+    (define f3 (fish "Dorado" 12))
+    (insert-fish f1)
+    (insert-fish f2)
+    (insert-fish f3)
+
+    (define c1 (cube 1 1 1))
+    (define c2 (cube 88 88 132))
+    (define c3 (cube 1 20 300))
+    (insert-cube c1)
+    (insert-cube c2)
+    (insert-cube c3)
+
+    ;; ---------------------------------------------------------------------------
+
+    (check-equal?
+      (ann (query-row: conn "SELECT * FROM fish LIMIT 1") (Vector Id String Integer))
+      (vector 1 (fish-name f1) (fish-weight f1)))
+
+    (check-equal?
+      (ann (query-row: conn "SELECT * FROM cube LIMIT 1") (Vector Id Integer Integer Integer))
+      (vector 1 (cube-length c1) (cube-width c1) (cube-height c1)))
+
+    (check-equal?
+      (ann (query-row: conn "SELECT * FROM fish WHERE fish.name = 'Tuna'") (Vector Id String Integer))
+      (vector 2 (fish-name f2) (fish-weight f2)))
+
+    (check-equal?
+      (ann (query-row: conn "SELECT * FROM cube WHERE cube.width = $1" 20) (Vector Id Integer Integer Integer))
+      (vector 3 (cube-length c3) (cube-width c3) (cube-height c3)))
+
+    (let-vector: ([v (query-row: conn "SELECT id FROM fish WHERE fish.id = 2")])
+      (check-equal? (vector-ref: v 0) 2)
+      (check-equal?
+        (ann (vector-length: v) One)
+        1))
 
   ;(query-row: c "SELECT * FROM word WHERE word.word = $1 word.id = $2" "blossom")
   ;(query-row: c "SELECT * FROM word WHERE word.word = $1 word.id = $4" "blossom" 2)
-
+)))