<?php

class ProductAttributeQueriesTest extends \Codeception\TestCase\WPTestCase {
    private $shop_manager;
    private $customer;
    private $helper;
    private $variation_helper;
    private $product_id;
    private $variation_ids;

    public function setUp() {
        parent::setUp();

        $this->shop_manager     = $this->factory->user->create( array( 'role' => 'shop_manager' ) );
        $this->customer         = $this->factory->user->create( array( 'role' => 'customer' ) );
        $this->helper           = $this->getModule('\Helper\Wpunit')->product();
        $this->variation_helper = $this->getModule('\Helper\Wpunit')->product_variation();        
        $this->product_id       = $this->helper->create_variable();
        $this->variation_ids    = $this->variation_helper->create( $this->product_id )['variations'];
        
    }

    public function tearDown() {
        parent::tearDown();
    }

    // tests
    public function testProductAttributeQuery() {
        $query = '
            query attributeQuery( $id: ID! ) {
                product( id: $id ) {
                    id
                    attributes {
                        nodes {
                            id
                            attributeId
                            name
                            options
                            position
                            visible
                            variation
                        }
                    }
                }
            }
        ';

        $variables = array( 'id' => $this->helper->to_relay_id( $this->product_id ) );
        $actual    = graphql(
            array(
                'query' => $query,
                'operation_name' => 'attributeQuery',
                'variables' => $variables,
            )
        );
		$expected = array(
            'data' => array(
                'product' => array(
                    'id'         => $this->helper->to_relay_id( $this->product_id ),
                    'attributes' => $this->helper->print_attributes( $this->product_id ),
                ),
            ),
        );

        // use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEqualSets( $expected, $actual );
    }

    public function testProductAttributeToProductConnectionQuery() {
        $query = '
            query attributeConnectionQuery( $color: [String!] ) {
                paColors( where: { name: $color } ) {
                    nodes {
                        products {
                            nodes {
                                id
                            }
                        }
                    }
                }
            }
        ';

        $variables = array( 'color' => 'red' );
        $actual    = graphql(
            array(
                'query'          => $query,
                'operation_name' =>'attributeConnectionQuery',
                'variables'      => $variables,
            )
        );
		$expected = array(
            'data' => array(
                'paColors' => array (
                    'nodes' => array(
                        array(
                            'products' => array(
                                'nodes' => array(
                                    array(
                                        'id' => $this->helper->to_relay_id( $this->product_id )
                                    )
                                )
                            )
                        )
                    )
                ),
            ),
        );

        // use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEqualSets( $expected, $actual );
    }

    public function testProductAttributeToVariationConnectionQuery() {
        $query = '
            query attributeConnectionQuery( $size: [String!] ) {
                paSizes( where: { name: $size } ) {
                    nodes {
                        variations {
                            nodes {
                                id
                            }
                        }
                    }
                }
            }
        ';

        $variables = array( 'size' => 'small' );
        $actual    = graphql(
            array(
                'query'          => $query,
                'operation_name' => 'attributeConnectionQuery',
                'variables'      => $variables,
            )
        );
		$expected = array(
            'data' => array(
                'paSizes' => array (
                    'nodes' => array(
                        array(
                            'variations' => array(
                                'nodes' => $this->variation_helper->print_nodes(
                                    $this->variation_ids,
                                    array(
                                        'filter' => function( $id ) {
                                            $variation = new \WC_Product_Variation( $id );
                                            $small_attribute = array_filter(
                                                $variation->get_attributes(),
                                                function( $attribute ) {
                                                    return 'small' === $attribute;
                                                }
                                            );
                                            return ! empty( $small_attribute );
                                        },
                                    )
                                ),
                            ),
                        ),
                    ),
                ),
            ),
        );

        // use --debug flag to view.
		codecept_debug( $actual );

		$this->assertEqualSets( $expected, $actual );
    }
}