1 // Copyright 2020 The Tint Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "src/writer/wgsl/test_helper.h"
16
17 namespace tint {
18 namespace writer {
19 namespace wgsl {
20 namespace {
21
22 using WgslGeneratorImplTest = TestHelper;
23
TEST_F(WgslGeneratorImplTest,IndexAccessor)24 TEST_F(WgslGeneratorImplTest, IndexAccessor) {
25 Global("ary", ty.array<i32, 10>(), ast::StorageClass::kPrivate);
26 auto* expr = IndexAccessor("ary", 5);
27 WrapInFunction(expr);
28
29 GeneratorImpl& gen = Build();
30
31 std::stringstream out;
32 ASSERT_TRUE(gen.EmitExpression(out, expr)) << gen.error();
33 EXPECT_EQ(out.str(), "ary[5]");
34 }
35
TEST_F(WgslGeneratorImplTest,IndexAccessor_OfDref)36 TEST_F(WgslGeneratorImplTest, IndexAccessor_OfDref) {
37 Global("ary", ty.array<i32, 10>(), ast::StorageClass::kPrivate);
38
39 auto* p = Const("p", nullptr, AddressOf("ary"));
40 auto* expr = IndexAccessor(Deref("p"), 5);
41 WrapInFunction(p, expr);
42
43 GeneratorImpl& gen = Build();
44
45 std::stringstream out;
46 ASSERT_TRUE(gen.EmitExpression(out, expr)) << gen.error();
47 EXPECT_EQ(out.str(), "(*(p))[5]");
48 }
49
50 } // namespace
51 } // namespace wgsl
52 } // namespace writer
53 } // namespace tint
54