366 lines
		
	
	
	
		
			11 KiB
		
	
	
	
		
			Scheme
		
	
	
	
	
	
			
		
		
	
	
			366 lines
		
	
	
	
		
			11 KiB
		
	
	
	
		
			Scheme
		
	
	
	
	
	
| ;;; GNU Guix --- Functional package management for GNU
 | ||
| ;;; Copyright © 2017 Christopher Baines <mail@cbaines.net>
 | ||
| ;;; Copyright © 2020 Marius Bakke <marius@gnu.org>
 | ||
| ;;;
 | ||
| ;;; This file is part of GNU Guix.
 | ||
| ;;;
 | ||
| ;;; GNU Guix is free software; you can redistribute it and/or modify it
 | ||
| ;;; under the terms of the GNU General Public License as published by
 | ||
| ;;; the Free Software Foundation; either version 3 of the License, or (at
 | ||
| ;;; your option) any later version.
 | ||
| ;;;
 | ||
| ;;; GNU Guix is distributed in the hope that it will be useful, but
 | ||
| ;;; WITHOUT ANY WARRANTY; without even the implied warranty of
 | ||
| ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | ||
| ;;; GNU General Public License for more details.
 | ||
| ;;;
 | ||
| ;;; You should have received a copy of the GNU General Public License
 | ||
| ;;; along with GNU Guix.  If not, see <http://www.gnu.org/licenses/>.
 | ||
| 
 | ||
| (define-module (gnu tests databases)
 | ||
|   #:use-module (gnu tests)
 | ||
|   #:use-module (gnu system)
 | ||
|   #:use-module (gnu system file-systems)
 | ||
|   #:use-module (gnu system shadow)
 | ||
|   #:use-module (gnu system vm)
 | ||
|   #:use-module (gnu services)
 | ||
|   #:use-module (gnu services databases)
 | ||
|   #:use-module (gnu services networking)
 | ||
|   #:use-module (gnu packages databases)
 | ||
|   #:use-module (guix gexp)
 | ||
|   #:use-module (guix store)
 | ||
|   #:export (%test-memcached
 | ||
|             %test-mongodb
 | ||
|             %test-postgresql
 | ||
|             %test-mysql))
 | ||
| 
 | ||
| (define %memcached-os
 | ||
|   (simple-operating-system
 | ||
|    (service dhcp-client-service-type)
 | ||
|    (service memcached-service-type)))
 | ||
| 
 | ||
| (define* (run-memcached-test #:optional (port 11211))
 | ||
|   "Run tests in %MEMCACHED-OS, forwarding PORT."
 | ||
|   (define os
 | ||
|     (marionette-operating-system
 | ||
|      %memcached-os
 | ||
|      #:imported-modules '((gnu services herd)
 | ||
|                           (guix combinators))))
 | ||
| 
 | ||
|   (define vm
 | ||
|     (virtual-machine
 | ||
|      (operating-system os)
 | ||
|      (port-forwardings `((11211 . ,port)))))
 | ||
| 
 | ||
|   (define test
 | ||
|     (with-imported-modules '((gnu build marionette))
 | ||
|       #~(begin
 | ||
|           (use-modules (srfi srfi-11) (srfi srfi-64)
 | ||
|                        (gnu build marionette)
 | ||
|                        (ice-9 rdelim))
 | ||
| 
 | ||
|           (define marionette
 | ||
|             (make-marionette (list #$vm)))
 | ||
| 
 | ||
|           (mkdir #$output)
 | ||
|           (chdir #$output)
 | ||
| 
 | ||
|           (test-begin "memcached")
 | ||
| 
 | ||
|           ;; Wait for memcached to be up and running.
 | ||
|           (test-assert "service running"
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (use-modules (gnu services herd))
 | ||
|                 (match (start-service 'memcached)
 | ||
|                   (#f #f)
 | ||
|                   (('service response-parts ...)
 | ||
|                    (match (assq-ref response-parts 'running)
 | ||
|                      ((pid) (number? pid))))))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (let* ((ai (car (getaddrinfo "localhost"
 | ||
|                                        #$(number->string port))))
 | ||
|                  (s  (socket (addrinfo:fam ai)
 | ||
|                              (addrinfo:socktype ai)
 | ||
|                              (addrinfo:protocol ai)))
 | ||
|                  (key "testkey")
 | ||
|                  (value "guix"))
 | ||
|             (connect s (addrinfo:addr ai))
 | ||
| 
 | ||
|             (test-equal "set"
 | ||
|               "STORED\r"
 | ||
|               (begin
 | ||
|                 (simple-format s "set ~A 0 60 ~A\r\n~A\r\n"
 | ||
|                                key
 | ||
|                                (string-length value)
 | ||
|                                value)
 | ||
|                 (read-line s)))
 | ||
| 
 | ||
|             (test-equal "get"
 | ||
|               (simple-format #f "VALUE ~A 0 ~A\r~A\r"
 | ||
|                              key
 | ||
|                              (string-length value)
 | ||
|                              value)
 | ||
|               (begin
 | ||
|                 (simple-format s "get ~A\r\n" key)
 | ||
|                 (string-append
 | ||
|                  (read-line s)
 | ||
|                  (read-line s))))
 | ||
| 
 | ||
|             (close-port s))
 | ||
| 
 | ||
|           ;; There should be a log file in here.
 | ||
|           (test-assert "log file"
 | ||
|             (marionette-eval
 | ||
|              '(file-exists? "/var/log/memcached")
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-end)
 | ||
|           (exit (= (test-runner-fail-count (test-runner-current)) 0)))))
 | ||
| 
 | ||
|   (gexp->derivation "memcached-test" test))
 | ||
| 
 | ||
| (define %test-memcached
 | ||
|   (system-test
 | ||
|    (name "memcached")
 | ||
|    (description "Connect to a running MEMCACHED server.")
 | ||
|    (value (run-memcached-test))))
 | ||
| 
 | ||
| (define %mongodb-os
 | ||
|   (operating-system
 | ||
|     (inherit
 | ||
|      (simple-operating-system
 | ||
|       (service dhcp-client-service-type)
 | ||
|       (service mongodb-service-type)))
 | ||
|     (packages (cons* mongodb
 | ||
|                      %base-packages))))
 | ||
| 
 | ||
| (define* (run-mongodb-test #:optional (port 27017))
 | ||
|   "Run tests in %MONGODB-OS, forwarding PORT."
 | ||
|   (define os
 | ||
|     (marionette-operating-system
 | ||
|      %mongodb-os
 | ||
|      #:imported-modules '((gnu services herd)
 | ||
|                           (guix combinators))))
 | ||
| 
 | ||
|   (define vm
 | ||
|     (virtual-machine
 | ||
|      (operating-system os)
 | ||
|      (memory-size 1024)
 | ||
|      (disk-image-size (* 1024 (expt 2 20)))
 | ||
|      (port-forwardings `((27017 . ,port)))))
 | ||
| 
 | ||
|   (define test
 | ||
|     (with-imported-modules '((gnu build marionette))
 | ||
|       #~(begin
 | ||
|           (use-modules (srfi srfi-11) (srfi srfi-64)
 | ||
|                        (gnu build marionette)
 | ||
|                        (ice-9 popen)
 | ||
|                        (ice-9 rdelim))
 | ||
| 
 | ||
|           (define marionette
 | ||
|             (make-marionette (list #$vm)))
 | ||
| 
 | ||
|           (mkdir #$output)
 | ||
|           (chdir #$output)
 | ||
| 
 | ||
|           (test-begin "mongodb")
 | ||
| 
 | ||
|           (test-assert "service running"
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (use-modules (gnu services herd))
 | ||
|                 (match (start-service 'mongodb)
 | ||
|                   (#f #f)
 | ||
|                   (('service response-parts ...)
 | ||
|                    (match (assq-ref response-parts 'running)
 | ||
|                      ((pid) (number? pid))))))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-eq "test insert"
 | ||
|             0
 | ||
|             (system* (string-append #$mongodb "/bin/mongo")
 | ||
|                      "test"
 | ||
|                      "--eval"
 | ||
|                      "db.testCollection.insert({data: 'test-data'})"))
 | ||
| 
 | ||
|           (test-equal "test find"
 | ||
|             "test-data"
 | ||
|             (let* ((port (open-pipe*
 | ||
|                           OPEN_READ
 | ||
|                           (string-append #$mongodb "/bin/mongo")
 | ||
|                           "test"
 | ||
|                           "--quiet"
 | ||
|                           "--eval"
 | ||
|                           "db.testCollection.findOne().data"))
 | ||
|                    (output (read-line port))
 | ||
|                    (status (close-pipe port)))
 | ||
|               output))
 | ||
| 
 | ||
|           (test-end)
 | ||
|           (exit (= (test-runner-fail-count (test-runner-current)) 0)))))
 | ||
| 
 | ||
|   (gexp->derivation "mongodb-test" test))
 | ||
| 
 | ||
| (define %test-mongodb
 | ||
|   (system-test
 | ||
|    (name "mongodb")
 | ||
|    (description "Connect to a running MONGODB server.")
 | ||
|    (value (run-mongodb-test))))
 | ||
| 
 | ||
| 
 | ||
| ;;;
 | ||
| ;;; The PostgreSQL service.
 | ||
| ;;;
 | ||
| 
 | ||
| (define %postgresql-os
 | ||
|   (simple-operating-system
 | ||
|    (service postgresql-service-type
 | ||
|             (postgresql-configuration
 | ||
|              (postgresql postgresql-10)))))
 | ||
| 
 | ||
| (define (run-postgresql-test)
 | ||
|   "Run tests in %POSTGRESQL-OS."
 | ||
|   (define os
 | ||
|     (marionette-operating-system
 | ||
|      %postgresql-os
 | ||
|      #:imported-modules '((gnu services herd)
 | ||
|                           (guix combinators))))
 | ||
| 
 | ||
|   (define vm
 | ||
|     (virtual-machine
 | ||
|      (operating-system os)
 | ||
|      (memory-size 512)))
 | ||
| 
 | ||
|   (define test
 | ||
|     (with-imported-modules '((gnu build marionette))
 | ||
|       #~(begin
 | ||
|           (use-modules (srfi srfi-64)
 | ||
|                        (gnu build marionette))
 | ||
| 
 | ||
|           (define marionette
 | ||
|             (make-marionette (list #$vm)))
 | ||
| 
 | ||
|           (mkdir #$output)
 | ||
|           (chdir #$output)
 | ||
| 
 | ||
|           (test-begin "postgresql")
 | ||
| 
 | ||
|           (test-assert "service running"
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (use-modules (gnu services herd))
 | ||
|                 (start-service 'postgres))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-end)
 | ||
|           (exit (= (test-runner-fail-count (test-runner-current)) 0)))))
 | ||
| 
 | ||
|   (gexp->derivation "postgresql-test" test))
 | ||
| 
 | ||
| (define %test-postgresql
 | ||
|   (system-test
 | ||
|    (name "postgresql")
 | ||
|    (description "Start the PostgreSQL service.")
 | ||
|    (value (run-postgresql-test))))
 | ||
| 
 | ||
| 
 | ||
| ;;;
 | ||
| ;;; The MySQL service.
 | ||
| ;;;
 | ||
| 
 | ||
| (define %mysql-os
 | ||
|   (simple-operating-system
 | ||
|    (service mysql-service-type)))
 | ||
| 
 | ||
| (define* (run-mysql-test)
 | ||
|   "Run tests in %MYSQL-OS."
 | ||
|   (define os
 | ||
|     (marionette-operating-system
 | ||
|      %mysql-os
 | ||
|      #:imported-modules '((gnu services herd)
 | ||
|                           (guix combinators))))
 | ||
| 
 | ||
|   (define vm
 | ||
|     (virtual-machine
 | ||
|      (operating-system os)
 | ||
|      (memory-size 512)))
 | ||
| 
 | ||
|   (define test
 | ||
|     (with-imported-modules '((gnu build marionette))
 | ||
|       #~(begin
 | ||
|           (use-modules (srfi srfi-11) (srfi srfi-64)
 | ||
|                        (gnu build marionette))
 | ||
| 
 | ||
|           (define marionette
 | ||
|             (make-marionette (list #$vm)))
 | ||
| 
 | ||
|           (mkdir #$output)
 | ||
|           (chdir #$output)
 | ||
| 
 | ||
|           (test-begin "mysql")
 | ||
| 
 | ||
|           (test-assert "service running"
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (use-modules (gnu services herd))
 | ||
|                 (match (start-service 'mysql)
 | ||
|                   (#f #f)
 | ||
|                   (('service response-parts ...)
 | ||
|                    (match (assq-ref response-parts 'running)
 | ||
|                      ((pid) (number? pid))))))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-assert "mysql_upgrade completed"
 | ||
|             (wait-for-file "/var/lib/mysql/mysql_upgrade_info" marionette))
 | ||
| 
 | ||
|           (test-eq "create database"
 | ||
|             0
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (system* #$(file-append mariadb "/bin/mysql")
 | ||
|                          "-e" "CREATE DATABASE guix;"))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-eq "create table"
 | ||
|             0
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (system*
 | ||
|                  #$(file-append mariadb "/bin/mysql") "guix"
 | ||
|                  "-e" "CREATE TABLE facts (id INT, data VARCHAR(12));"))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-eq "insert data"
 | ||
|             0
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (system* #$(file-append mariadb "/bin/mysql") "guix"
 | ||
|                          "-e" "INSERT INTO facts VALUES (1, 'awesome')"))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-equal "retrieve data"
 | ||
|             "awesome\n"
 | ||
|             (marionette-eval
 | ||
|              '(begin
 | ||
|                 (use-modules (ice-9 popen))
 | ||
|                 (let* ((port (open-pipe*
 | ||
|                               OPEN_READ
 | ||
|                               #$(file-append mariadb "/bin/mysql") "guix"
 | ||
|                               "-NB" "-e" "SELECT data FROM facts WHERE id=1;"))
 | ||
|                        (output (get-string-all port)))
 | ||
|                   (close-pipe port)
 | ||
|                   output))
 | ||
|              marionette))
 | ||
| 
 | ||
|           (test-end)
 | ||
|           (exit (= (test-runner-fail-count (test-runner-current)) 0)))))
 | ||
| 
 | ||
|   (gexp->derivation "mysql-test" test))
 | ||
| 
 | ||
| (define %test-mysql
 | ||
|   (system-test
 | ||
|    (name "mysql")
 | ||
|    (description "Start the MySQL service.")
 | ||
|    (value (run-mysql-test))))
 |