feat: ansible run unit and migration tests passing!

Author: samrose

ansible/tasks/stage2/playbook.yml

@@ -15,7 +15,16 @@
 
     - set_fact:
         parallel_jobs: 16
-    
+
+    - name: stat unit test file copy
+      copy:
+        src: /tmp/unit-tests/unit-test-01.sql
+        dest: /home/postgres/unit-test-01.sql
+        # state: present
+        owner: postgres
+        group: postgres
+        mode: '0644'
+
     - name: locale-gen
       command: sudo locale-gen en_US.UTF-8
 

ansible/tasks/stage2/stage2-setup-postgres.yml

@@ -3,11 +3,12 @@
   shell: |
     sudo -u postgres bash -c ". /nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh && nix profile install github:supabase/postgres#psql_15/bin"
 
-- name: Install glibcLocales from nix binary cache
+- name: Install pg_prove from nix binary cache
   become: yes
   shell: |
-    sudo -u postgres bash -c ". /nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh && nix profile install nixpkgs#glibcLocales"
+    sudo -u postgres bash -c ". /nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh && nix profile install github:supabase/postgres/sam/2-stage-ami-nix#pg_prove"
 
+#TODO switch pg_prove sourcing to develop branch once PR is merged
 
 # - name: Generate en_US.UTF-8 locale
 #   command: sudo localedef -i en_US -c -f UTF-8 -A /usr/share/locale/locale.alias en_US.UTF-8

ansible/tasks/stage2/test-image.yml

@@ -1,8 +1,3 @@
-- name: install pg_prove
-  apt:
-    pkg:
-      - libtap-parser-sourcehandler-pgtap-perl
-
 - name: Temporarily disable PG Sodium references in config
   become: yes
   become_user: postgres
@@ -22,17 +17,58 @@
     LC_CTYPE: en_US.UTF-8
     LOCALE_ARCHIVE: /usr/lib/locale/locale-archive
 
+
+- name: check contents of unit test file
+  become: yes
+  become_user: postgres
+  shell: cat /home/postgres/unit-test-01.sql && ls -l /home/postgres/unit-test-01.sql
+
+- name: verify postgres server is running
+  become: yes
+  become_user: postgres
+  shell: psql -U postgres -h localhost -c "SELECT version();"
+  register: pg_isready
+  failed_when: pg_isready.rc != 0
+  args:
+    executable: /bin/bash
+  environment:
+    LANG: en_US.UTF-8
+    LANGUAGE: en_US.UTF-8
+    LC_ALL: en_US.UTF-8
+    LC_CTYPE: en_US.UTF-8
+    LOCALE_ARCHIVE: /usr/lib/locale/locale-archive
+    PATH: /home/postgres/.nix-profile/bin:$PATH
+
 - name: Run Unit tests (with filename unit-test-*) on Postgres Database
-  shell: /usr/bin/pg_prove -U postgres -h localhost -d postgres -v /tmp/unit-tests/unit-test-01.sql
+  become: yes
+  become_user: postgres
+  shell: source /home/postgres/.bashrc && /home/postgres/.nix-profile/bin/pg_prove -U postgres -h localhost -d postgres -f /home/postgres/unit-test-01.sql
   register: retval
   failed_when: retval.rc != 0
+  args:
+    executable: /bin/bash
+  environment:
+    LANG: en_US.UTF-8
+    LANGUAGE: en_US.UTF-8
+    LC_ALL: en_US.UTF-8
+    LC_CTYPE: en_US.UTF-8
+    LOCALE_ARCHIVE: /usr/lib/locale/locale-archive
+    PATH: /home/postgres/.nix-profile/bin:$PATH
 
 - name: Run migrations tests
-  shell: /usr/bin/pg_prove -U supabase_admin -h localhost -d postgres -v tests/test.sql
+  shell: /home/postgres/.nix-profile/bin/pg_prove -U supabase_admin -h localhost -d postgres -v tests/test.sql
   register: retval
   failed_when: retval.rc != 0
   args:
+    executable: /bin/bash
     chdir: /tmp/migrations
+  environment:
+    LANG: en_US.UTF-8
+    LANGUAGE: en_US.UTF-8
+    LC_ALL: en_US.UTF-8
+    LC_CTYPE: en_US.UTF-8
+    LOCALE_ARCHIVE: /usr/lib/locale/locale-archive
+    PATH: /home/postgres/.nix-profile/bin:$PATH
 
 - name: Re-enable PG Sodium references in config
   become: yes
@@ -55,12 +91,11 @@
     LC_CTYPE: en_US.UTF-8
     LOCALE_ARCHIVE: /usr/lib/locale/locale-archive
 
-- name: remove pg_prove
-  apt:
-    pkg:
-      - libtap-parser-sourcehandler-pgtap-perl
-    state: absent
-    autoremove: yes
+# - name: remove pg_prove
+- name: Remove  pg_prove 
+  become: yes
+  shell: |
+    sudo -u postgres bash -c ". /nix/var/nix/profiles/default/etc/profile.d/nix-daemon.sh && nix profile remove pg_prove"
 
 - name: Stop Postgres Database
   become: yes

stage2-nix-psql.pkr.hcl

@@ -70,13 +70,14 @@ build {
     source = "migrations"
     destination = "/tmp"
   }
-  
+
+  provisioner "file" {
+    source       = "ebssurrogate/files/unit-tests"
+    destination  = "/tmp/unit-tests"
+  }
+
   provisioner "shell" {
      script = "scripts/nix-provision.sh"
   }
 
-  provisioner "file" {
-    source = "ebssurrogate/files/unit-tests"
-    destination = "/tmp"
-  }
 }