fsharp/tests/FSharp.Compiler.ComponentTests/Conformance/Expressions/ComputationExpressions/CEExtensionMethodCapture.fs at 0a74bdbdf5db6a8c938d56c88325a2e7a61e26b5 · dotnet/fsharp · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
module Conformance.Expressions.CEExtensionMethodCapture

open Xunit
open FSharp.Test.Compiler

[<Fact>]
let ``CE doesn't capture an extension method beyond the access domain``() =
    FSharp """
        open System.Runtime.CompilerServices

        type AsyncSeq<'T>(i: 'T) =
            class
            let l = [i]
            member this.Data = l
            end

        type AsyncSeqBuilder() =
            member _.Yield(x: 'T) : AsyncSeq<'T> =
                AsyncSeq(x)

        [<Extension>]
        type PrivateExtensions =
            [<Extension>]
            static member inline private Run(this: AsyncSeqBuilder) =
                this

        let asyncSeq = AsyncSeqBuilder()

        let xs : AsyncSeq<int> =
            asyncSeq {
                yield 1
            }
    """
    |> asExe
    |> compile
    |> shouldSucceed

[<Fact>]
let ``CE doesn't capture an extension method with generic type``() =
    FSharp """
        open System.Runtime.CompilerServices

        type FooClass = class end

        type AsyncSeq<'T>(i: 'T) =
            class
            let l = [i]
            member this.Data = l
            end

        type AsyncSeqBuilder() =
            member _.Yield(x: 'T) : AsyncSeq<'T> =
                AsyncSeq(x)

        [<Extension>]
        type PublicExtensions =
            [<Extension>]
            static member inline Run(this: #FooClass) =
                this

        let asyncSeq = AsyncSeqBuilder()

        let xs : AsyncSeq<int> =
            asyncSeq {
                yield 1
            }
    """
    |> asExe
    |> compile
    |> shouldSucceed

// Deliberately trigger an error to ensure that a method is captured
[<Fact>]
let ``CE captures a public extension method and procudes an error due to invalid args``() =
    FSharp """
        open System.Runtime.CompilerServices

        type AsyncSeq<'T>(i: 'T) =
            class
            let l = [i]
            member this.Data = l
            end

        type AsyncSeqBuilder() =
            member _.Yield(x: 'T) : AsyncSeq<'T> =
                AsyncSeq(x)

        [<Extension>]
        type PublicExtensions =
            [<Extension>]
            static member inline Run(this: AsyncSeqBuilder, invalidArg: string) =
                this

        let asyncSeq = AsyncSeqBuilder()

        let xs : AsyncSeq<int> =
            asyncSeq {
                yield 1
            }
    """
    |> asExe
    |> compile
    |> shouldFail

// Deliberately trigger an error to ensure that a method is captured
[<Fact>]
let ``CE captures a public extension method with valid generic constrainted type and procudes an error due to invalid args``() =
    FSharp """
        open System.Runtime.CompilerServices

        type AsyncSeq<'T>(i: 'T) =
            class
            let l = [i]
            member this.Data = l
            end

        type AsyncSeqBuilder() =
            member _.Yield(x: 'T) : AsyncSeq<'T> =
                AsyncSeq(x)

        [<Extension>]
        type PublicExtensions =
            [<Extension>]
            static member inline Run(this: #AsyncSeqBuilder, invalidArg: string) =
                this

        let asyncSeq = AsyncSeqBuilder()

        let xs : AsyncSeq<int> =
            asyncSeq {
                yield 1
            }
    """
    |> asExe
    |> compile
    |> shouldFail