budibase/packages/builder/cypress/integration/datasources/postgreSql.spec.js

context("PostgreSQL Datasource Testing", () => {
    if (Cypress.env("TEST_ENV")) {
    
        before(() => {
        cy.login()
        cy.createTestApp()
        })
        const datasource = "PostgreSQL"
        const queryName = "Cypress Test Query"
        const queryRename = "CT Query Rename"
        
        it("Should add PostgreSQL data source without configuration", () => {
            // Select PostgreSQL data source
            cy.selectExternalDatasource(datasource)
            // Attempt to fetch tables without applying configuration
            cy.intercept('**/datasources').as('datasource')
            cy.get(".spectrum-Button")
            .contains("Save and fetch tables")
            .click({ force: true })
            // Intercept Request after button click & apply assertions
            cy.wait("@datasource")
            cy.get("@datasource").its('response.body')
            .should('have.property', 'message', 'connect ECONNREFUSED 127.0.0.1:5432')
            cy.get("@datasource").its('response.body')
            .should('have.property', 'status', 500)
        })
        
        it("should add PostgreSQL data source and fetch tables", () => {
            // Add & configure PostgreSQL data source
            cy.selectExternalDatasource(datasource)
            cy.intercept('**/datasources').as('datasource')
            cy.addDatasourceConfig(datasource)
            // Check response from datasource after adding configuration
            cy.wait("@datasource")
            cy.get("@datasource").its('response.statusCode')
            .should('eq', 200)
            // Confirm fetch tables was successful
            cy.get(".spectrum-Table-body").eq(0)
            .find('tr')
            .its('length')
            .should('be.gt', 0)
        })
        
        it("should define a One relationship type", () => {
            // Select relationship type & configure
            cy.get(".spectrum-Button").contains("Define relationship").click({ force: true })
            cy.get(".spectrum-Dialog-grid").within(() => {
                cy.get(".spectrum-Picker").eq(0).click()
                cy.get(".spectrum-Popover").contains("One").click()
                cy.get(".spectrum-Picker").eq(1).click()
                cy.get(".spectrum-Popover").contains("REGIONS").click()
                cy.get(".spectrum-Picker").eq(2).click()
                cy.get(".spectrum-Popover").contains("REGION_ID").click()
                cy.get(".spectrum-Picker").eq(3).click()
                cy.get(".spectrum-Popover").contains("COUNTRIES").click()
                cy.get(".spectrum-Picker").eq(4).click()
                cy.get(".spectrum-Popover").contains("REGION_ID").click()
                // Save relationship & reload page
                cy.get(".spectrum-Button").contains("Save").click({ force: true })
                cy.reload()
            })
            // Confirm table length & column name
            cy.get(".spectrum-Table-body").eq(1)
            .find('tr')
            .its('length')
            .should('eq', 1)
            cy.get(".spectrum-Table-cell").should('contain', "COUNTRIES to REGIONS")
        })
        
        it("should define a Many relationship type", () => {
            // Select relationship type & configure
            cy.get(".spectrum-Button").contains("Define relationship").click({ force: true })
            cy.get(".spectrum-Dialog-grid").within(() => {
                cy.get(".spectrum-Picker").eq(0).click()
                cy.get(".spectrum-Popover").contains("Many").click()
                cy.get(".spectrum-Picker").eq(1).click()
                cy.get(".spectrum-Popover").contains("LOCATIONS").click()
                cy.get(".spectrum-Picker").eq(2).click()
                cy.get(".spectrum-Popover").contains("REGIONS").click()
                cy.get(".spectrum-Picker").eq(3).click()
                cy.get(".spectrum-Popover").contains("COUNTRIES").click()
                cy.get(".spectrum-Picker").eq(4).click()
                cy.get(".spectrum-Popover").contains("COUNTRY_ID").click()
                cy.get(".spectrum-Picker").eq(5).click()
                cy.get(".spectrum-Popover").contains("REGION_ID").click()
                // Save relationship & reload page
                cy.get(".spectrum-Button").contains("Save").click({ force: true })
                cy.reload()
            })
            // Confirm table length & relationship name
            cy.get(".spectrum-Table-body").eq(1)
            .find('tr')
            .its('length')
            .should('eq', 2)
            cy.get(".spectrum-Table-cell")
            .should('contain', "LOCATIONS through COUNTRIES → REGIONS")
        })
        
        it("should delete a relationship", () => {
            cy.get(".hierarchy-items-container").contains(datasource).click()
            cy.reload()
            // Delete one relationship
            cy.get(".spectrum-Table-body").eq(1).within(() => {
                cy.get(".spectrum-Table-row").eq(0).click()
                cy.wait(500)
                })
            cy.get(".spectrum-Dialog-grid").within(() => {
                cy.get(".spectrum-Button").contains("Delete").click({ force: true })
                })
            cy.reload()
            // Confirm relationship was deleted
            cy.get(".spectrum-Table-body")
            .eq(1).find('tr').its('length').should('eq', 1)
        })
        
        it("should add a query", () => {
            // Add query
            cy.get(".spectrum-Button").contains("Add query").click({ force: true })
            cy.get(".spectrum-Form-item").eq(0).within(() => {
                cy.get("input").type(queryName)
            })
            // Insert Query within Fields section
            cy.get(".CodeMirror textarea").eq(0)
            .type("SELECT * FROM books", { force: true })
            // Intercept query execution
            cy.intercept('**/queries/preview').as('query')
            cy.get(".spectrum-Button").contains("Run Query").click({ force: true })
            cy.wait(500)
            cy.wait("@query")
            // Assert against Status Code & Body
            cy.get("@query").its('response.statusCode')
            .should('eq', 200)
            cy.get("@query").its('response.body')
            .should('not.be.empty')
            // Save query
            cy.get(".spectrum-Button").contains("Save Query").click({ force: true })
            cy.get(".hierarchy-items-container").should('contain', queryName)
        })
        
        it("should switch to schema with no tables", () => {
            // Switch Schema - To one without any tables
            cy.get(".hierarchy-items-container").contains(datasource).click()
            switchSchema("randomText")
            
            // No tables displayed
            cy.get(".spectrum-Body").eq(2).should('contain', 'No tables found')
            
            // Previously created query should be visible
            cy.get(".spectrum-Table-body").should('contain', queryName)
        })
        
        it("should switch schemas", () => {
            // Switch schema - To one with tables
            switchSchema("1")
            
            // Confirm tables exist - Check for specific one
            cy.get(".spectrum-Table-body").eq(0).should('contain', 'test')
            cy.get(".spectrum-Table-body").eq(0).find('tr').its('length').should('eq', 1)
            
            // Confirm specific table visible within left nav bar
            cy.get(".hierarchy-items-container").should('contain', 'test')
            
            // Switch back to public schema
            switchSchema("public")
            
            // Confirm tables exist - again
            cy.get(".spectrum-Table-body").eq(0).should('contain', 'REGIONS')
            cy.get(".spectrum-Table-body").eq(0)
            .find('tr').its('length').should('be.gt', 1)
            
            // Confirm specific table visible within left nav bar
            cy.get(".hierarchy-items-container").should('contain', 'REGIONS')
            
            // No relationships and one query
            cy.get(".spectrum-Body").eq(3).should('contain', 'No relationships configured.')
            cy.get(".spectrum-Table-body").eq(1).should('contain', queryName)
        })
        
        it("should duplicate a query", () => {
            // Get last nav item - The query
            cy.get(".nav-item").last().within(() => {
                    cy.get(".icon").eq(1).click({ force: true })
                })
            // Select and confirm duplication
            cy.get(".spectrum-Menu").contains("Duplicate").click()
            cy.get(".nav-item").should('contain', queryName + ' (1)')
        })
        
        it("should edit a query name", () => {
            // Access query
            cy.get(".hierarchy-items-container").contains(queryName + ' (1)').click()
            
            // Rename query
            cy.get(".spectrum-Form-item").eq(0).within(() => {
                cy.get("input").clear().type(queryRename)
            })
            
            // Run and Save query
            cy.get(".spectrum-Button").contains("Run Query").click({ force: true })
            cy.wait(500)
            cy.get(".spectrum-Button").contains("Save Query").click({ force: true })
            cy.get(".nav-item").should('contain', queryRename)
        })
        
        it("should delete a query", () => {
            // Get last nav item - The query
            for (let i = 0; i < 2; i++) {
                cy.get(".nav-item").last().within(() => {
                    cy.get(".icon").eq(1).click({ force: true })
                })
                // Select Delete
                cy.get(".spectrum-Menu").contains("Delete").click()
                cy.get(".spectrum-Button").contains("Delete Query").click({ force: true })
                cy.wait(1000)
            }
            // Confirm deletion
            cy.get(".nav-item").should('not.contain', queryName)
            cy.get(".nav-item").should('not.contain', queryRename)
        })
        
        const switchSchema = (schema) => {
            // Edit configuration - Change Schema
            cy.get(".spectrum-Textfield").eq(6).within(() => {
                cy.get('input').clear().type(schema)
            })
            // Save configuration & fetch
            cy.get(".spectrum-Button").contains("Save").click({ force: true })
            cy.get(".spectrum-Button").contains("Fetch tables").click({ force: true })
            // Click fetch tables again within modal
            cy.get(".spectrum-Dialog-grid").within(() => {
                cy.get(".spectrum-Button").contains("Fetch tables").click({ force: true })
            })
            cy.reload()
            cy.wait(5000)
        }
    }
})