flow fixes

Author: ZeeCoder

package.json

@@ -19,7 +19,7 @@
     "prettier": "^1.10.2"
   },
   "scripts": {
-    "test": "jest --coverage",
+    "test": "lerna run flow && jest --coverage",
     "coveralls": "cat ./coverage/lcov.info | coveralls",
     "watch:test": "jest --watch --notify",
     "precommit": "lint-staged",

packages/container-query/package.json

@@ -10,8 +10,9 @@
   "main": "dist/bundle.cjs.js",
   "module": "dist/bundle.esm.js",
   "scripts": {
-    "test": "jest && yarn run build && yarn run size",
-    "prebuild": "flow check",
+    "test": "yarn run flow && jest && yarn run build && yarn run size",
+    "flow": "flow check",
+    "prebuild": "yarn run flow",
     "build:esm": "BABEL_ENV=production rollup -c rollup/rollup.esm.js",
     "build:cjs": "BABEL_ENV=production rollup -c rollup/rollup.cjs.js",
     "build:umd": "BABEL_ENV=production rollup -c rollup/rollup.umd.js",

packages/container-query/src/containerRegistry.js

@@ -5,19 +5,19 @@ import type { RegistryData } from "../flow/types";
 const containerRegistry: WeakMap = new WeakMap();
 
 export default {
-  set: function(element: Element, data: RegistryData) {
+  set: function(element: Element | Node, data: RegistryData) {
     containerRegistry.set(element, data);
   },
 
-  get: function(element: Element) {
+  get: function(element: Element | Node) {
     return containerRegistry.get(element);
   },
 
-  has: function(element: Element) {
+  has: function(element: Element | Node) {
     return containerRegistry.has(element);
   },
 
-  delete: function(element: Element) {
+  delete: function(element: Element | Node) {
     containerRegistry.delete(element);
   }
 };

packages/postcss-container-query/package.json

@@ -12,9 +12,10 @@
     "fs": false
   },
   "scripts": {
-    "test": "jest",
+    "test": "yarn run flow && jest",
     "build": "rollup -c",
-    "prebuild": "flow check",
+    "flow": "flow check",
+    "prebuild": "yarn run flow",
     "prepublish": "yarn build"
   },
   "devDependencies": {