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

import 'package:flutter_test/flutter_test.dart';
import 'package:flutter/material.dart';

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

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

29
    final Element builderElement = tester.element(find.byKey(builderKey));
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
    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) {
45 46
      return MaterialApp(
        home: StatefulBuilder(
47 48 49 50
          key: key,
          builder: (BuildContext context, StateSetter setter) {
            PageStorage.of(context).writeState(context, storedValue, identifier: 123);
            setState = setter;
51
            return Center(
52
              child: Text('storedValue: $storedValue'),
53
            );
54 55
          },
        ),
56 57 58
      );
    }

59
    Key key = const Key('Key one');
60 61 62 63 64 65 66 67
    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.

68
    key = const Key('Key two');
69 70 71 72 73 74 75 76 77 78 79 80 81 82
    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));
  });

}