TestGeneration.cs

using StrawberryShake.Tools.Configuration;
using static StrawberryShake.CodeGeneration.CSharp.GeneratorTestHelper;

namespace StrawberryShake.CodeGeneration.CSharp.Integration;

public class TestGeneration
{
    [Fact]
    public void StarWarsGetHero() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                }
            }");

    [Fact]
    public void StarWarsGetHeroWithFragmentIncludeAndSkipDirective() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHeroWithFragmentIncludeAndSkipDirective($includePageInfo: Boolean = false, $skipPageInfo: Boolean = true) {
                hero(episode: NEW_HOPE) {
                    ...HeroFragment
                }
            }

            fragment HeroFragment on Character {
                id
                friends {
                    ...FriendsFragment
                }
            }

            fragment FriendsFragment on FriendsConnection {
                includedPageInfo: pageInfo @include(if: $includePageInfo) {
                    ...PageInfoFragment
                }
                skippedPageInfo: pageInfo @skip(if: $skipPageInfo) {
                    ...PageInfoFragment
                }
            }

            fragment PageInfoFragment on PageInfo {
                hasNextPage
            }");

    [Fact]
    public void StarWarsGetFriendsNoStore() =>
        AssertStarWarsResult(
            CreateIntegrationTest(noStore: true),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    friends {
                        nodes {
                            name
                        }
                    }
                }
            }");

    [Fact]
    public void StarWarsGetFriends() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    friends {
                        nodes {
                            name
                        }
                    }
                }
            }");

    [Fact]
    public void MultiProfile() =>
        AssertStarWarsResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("InMemory", TransportType.InMemory),
                TransportProfile.Default
            ]),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    friends {
                        nodes {
                            name
                        }
                    }
                }
            }",
            @"subscription OnReviewSub {
                onReview(episode: NEW_HOPE) {
                  __typename
                  stars
                  commentary
                }
              }
            ",
            @"mutation createReviewMut($episode: Episode!, $review: ReviewInput!) {
                createReview(episode: $episode, review: $review) {
                    stars
                    commentary
                }
            }");

    [Fact]
    public void AnyScalarDefaultSerialization() =>
        AssertResult(
            CreateIntegrationTest(),
            skipWarnings: true,
            @"query GetJson {
                json
            }",
            @"type Query {
                json: Any!
            }");

    [Fact]
    public void StarWarsTypeNameOnInterfaces() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    __typename
                }
            }");

    [Fact]
    public void StarWarsTypeNameOnUnions() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            """
            query SearchHero {
                search(text: "l") {
                    __typename
                }
            }
            """);

    [Fact]
    public void StarWarsUnionList() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            """
            query SearchHero {
                search(text: "l") {
                    ... on Human {
                        name
                        friends {
                            nodes {
                                name
                            }
                        }
                    }
                    ... on Droid {
                        name
                    }
                }
            }
            """);

    [Fact]
    public void StarWarsGetHeroTraits() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    traits
                }
            }");

    [Fact]
    public void EntityIdOrData() =>
        AssertResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("Default", TransportType.InMemory)
            ]),
            skipWarnings: true,
            @"
                query GetFoo {
                    foo {
                        ... on Baz {
                            id
                        }
                        ... on Quox {
                            foo
                        }
                        ... on Baz2 {
                            id
                        }
                        ... on Quox2 {
                            foo
                        }
                    }
                }
                ",
            @"
                type Query {
                    foo: [Bar]
                }

                type Baz {
                    id: String
                }

                type Baz2 {
                    id: String
                }

                type Quox {
                    foo: String
                }

                type Quox2 {
                    foo: String
                }

                union Bar = Baz | Quox | Baz2 | Quox2
                ",
            "extend schema @key(fields: \"id\")");

    [Fact]
    public void StarWarsIntrospection() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            FileResource.Open("IntrospectionQuery.graphql"));

    [Fact]
    public void StarWarsGetFriendsDeferred() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            """
            query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    ... FriendsList @defer(label: "FriendsListLabel")
                }
            }

            fragment FriendsList on Character {
                friends {
                    nodes {
                        name
                    }
                }
            }
            """);

    [Fact]
    public void StarWarsGetFriendsDeferInList() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            """
            query GetHero {
                hero(episode: NEW_HOPE) {
                    friends {
                        nodes {
                            ... CharacterName @defer(label: "CharacterName")
                        }
                    }
                }
            }

            fragment CharacterName on Character {
                name
            }
            """);

    [Fact]
    public void StarWarsOnReviewSubCompletion() =>
        AssertStarWarsResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("InMemory", TransportType.InMemory),
                TransportProfile.Default
            ]),
            @"subscription OnReviewSub {
                onReview(episode: NEW_HOPE) {
                    __typename
                    stars
                    commentary
                }
            }");

    [Fact]
    public void StarWarsOnReviewSubNoStore() =>
        AssertStarWarsResult(
            CreateIntegrationTest(noStore: true),
            @"subscription OnReviewSub {
                onReview(episode: NEW_HOPE) {
                    __typename
                    stars
                    commentary
                }
            }");

    [Fact]
    public void StarWarsOnReviewSubGraphQLSSE() =>
        AssertStarWarsResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("default", TransportType.Http)
            ]),
            @"subscription OnReviewSub {
                onReview(episode: NEW_HOPE) {
                    __typename
                    stars
                    commentary
                }
            }");

    /*
    [Fact]
    public void StarWarsGetFriendsDeferredData() =>
        AssertStarWarsResult(
            CreateIntegrationTest(),
            @"query GetHero {
                hero(episode: NEW_HOPE) {
                    name
                    friends {
                        ... FriendsList @defer(label: ""FriendsListLabel"")
                    }
                }
            }

            fragment FriendsList on FriendsConnection {
                nodes {
                    name
                }
            }");
            */
    private const string UploadQueries = @"
        query TestUpload(
                $nonUpload: String
                $single: Upload
                $list: [Upload]
                $nested: [[Upload]]
                $object: TestInput
                $objectList: [TestInput]
                $objectNested: [[TestInput]]) {
            upload(
                nonUpload: $nonUpload
                single: $single
                list: $list
                nested: $nested
                object: $object
                objectList: $objectList
                objectNested: $objectNested)
        }";
    private const string UploadSchema = @"
        type Query {
            upload(
                nonUpload: String
                single: Upload
                list: [Upload]
                nested: [[Upload]]
                object: TestInput
                objectList: [TestInput]
                objectNested: [[TestInput]]): String
        }

        input TestInput {
            bar: BarInput
        }

        input BarInput {
            baz: BazInput
        }

        input BazInput {
            file: Upload
        }

        scalar Upload
        ";

    [Fact]
    public void UploadScalar() =>
        AssertResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("Default", TransportType.Http)
            ]),
            skipWarnings: true,
            UploadQueries,
            UploadSchema,
            "extend schema @key(fields: \"id\")");

    [Fact]
    public void UploadScalar_InMemory() =>
        AssertResult(
            CreateIntegrationTest(profiles:
            [
                new TransportProfile("Default", TransportType.InMemory)
            ]),
            skipWarnings: true,
            UploadQueries,
            UploadSchema,
            "extend schema @key(fields: \"id\")");

    [Fact]
    public void LocalTypes() =>
        AssertResult(
            CreateIntegrationTest(),
            skipWarnings: true,
            """
            query LocalTypes {
                localDate(input: "2021-10-10")
                localDateTime(input: "2021-10-10T10:10:10")
                localTime(input: "10:10:10")
            }
            """,
            """
            type Query {
                localDate(input: LocalDate!): LocalDate!
                localDateTime(input: LocalDateTime!): LocalDateTime!
                localTime(input: LocalTime!): LocalTime!
            }

            scalar LocalDate
            scalar LocalDateTime
            scalar LocalTime
            """);

    [Fact]
    public void UnsignedTypes() =>
        AssertResult(
            CreateIntegrationTest(),
            skipWarnings: true,
            """
            query UnsignedTypes {
                unsignedByte(input: 1)
                unsignedShort(input: 256)
                unsignedInt(input: 65536)
                unsignedLong(input: 4294967296)
            }
            """,
            """
            type Query {
                unsignedByte(input: UnsignedByte!): UnsignedByte!
                unsignedShort(input: UnsignedShort!): UnsignedShort!
                unsignedInt(input: UnsignedInt!): UnsignedInt!
                unsignedLong(input: UnsignedLong!): UnsignedLong!
            }

            scalar UnsignedByte
            scalar UnsignedShort
            scalar UnsignedInt
            scalar UnsignedLong
            """);
}