page_storage_test.dart 2.83 KB
Newer Older
Ian Hickson's avatar
Ian Hickson committed
1
// Copyright 2014 The Flutter Authors. All rights reserved.
2 3 4
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

5 6
// @dart = 2.8

7 8 9 10 11
import 'package:flutter_test/flutter_test.dart';
import 'package:flutter/material.dart';

void main() {
  testWidgets('PageStorage read and write', (WidgetTester tester) async {
12
    const Key builderKey = PageStorageKey<String>('builderKey');
13 14 15 16
    StateSetter setState;
    int storedValue = 0;

    await tester.pumpWidget(
17 18
      MaterialApp(
        home: StatefulBuilder(
19 20 21 22
          key: builderKey,
          builder: (BuildContext context, StateSetter setter) {
            PageStorage.of(context).writeState(context, storedValue);
            setState = setter;
23
            return Center(
24
              child: Text('storedValue: $storedValue'),
25
            );
26 27
          },
        ),
28
      ),
29 30
    );

31
    final Element builderElement = tester.element(find.byKey(builderKey));
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
    expect(PageStorage.of(builderElement), isNotNull);
    expect(PageStorage.of(builderElement).readState(builderElement), equals(storedValue));

    setState(() {
      storedValue = 1;
    });
    await tester.pump();
    expect(PageStorage.of(builderElement).readState(builderElement), equals(storedValue));
  });

  testWidgets('PageStorage read and write by identifier', (WidgetTester tester) async {
    StateSetter setState;
    int storedValue = 0;

    Widget buildWidthKey(Key key) {
47 48
      return MaterialApp(
        home: StatefulBuilder(
49 50 51 52
          key: key,
          builder: (BuildContext context, StateSetter setter) {
            PageStorage.of(context).writeState(context, storedValue, identifier: 123);
            setState = setter;
53
            return Center(
54
              child: Text('storedValue: $storedValue'),
55
            );
56 57
          },
        ),
58 59 60
      );
    }

61
    Key key = const Key('Key one');
62 63 64 65 66 67 68 69
    await tester.pumpWidget(buildWidthKey(key));
    Element builderElement = tester.element(find.byKey(key));
    expect(PageStorage.of(builderElement), isNotNull);
    expect(PageStorage.of(builderElement).readState(builderElement), isNull);
    expect(PageStorage.of(builderElement).readState(builderElement, identifier: 123), equals(storedValue));

    // New StatefulBuilder widget - different key - but the same PageStorage identifier.

70
    key = const Key('Key two');
71 72 73 74 75 76 77 78 79 80 81 82 83 84
    await tester.pumpWidget(buildWidthKey(key));
    builderElement = tester.element(find.byKey(key));
    expect(PageStorage.of(builderElement), isNotNull);
    expect(PageStorage.of(builderElement).readState(builderElement), isNull);
    expect(PageStorage.of(builderElement).readState(builderElement, identifier: 123), equals(storedValue));

    setState(() {
      storedValue = 1;
    });
    await tester.pump();
    expect(PageStorage.of(builderElement).readState(builderElement, identifier: 123), equals(storedValue));
  });

}