add readme and update test units

README.md

@@ -12,17 +12,36 @@ Operating on the NODE_ENV environmental variable
 
 ## Usage
 
+### CONST
+
 ```javascript
 const { ENV } = require("node-env-manager");
 
 ENV.PROD  // -> "production"
 ENV.DEV   // -> "development"
 ENV.LOCAL // -> "local"
 ENV.TEST  // -> "test"
-ENV.EXP   // ->"experience"
+ENV.EXP   // -> "experience"
 ENV.GRAY  // -> "gray"
 ENV.CI    // -> "ci"
+```
+
+### Mothods
+
+```javascript
+const env = require("node-env-manager");
+
+env.init(); // set `process.env.NODE_ENV` value if this is null
+env.set();  // set `process.env.NODE_ENV` value
+env.get();  // get `process.env.NODE_ENV` value
 
+env.isLocal();
+env.isTest();
+env.isDev();
+env.isProd();
+env.isExp();
+env.isGray();
+env.isCI();
 ```
 
 [NPM_URL]: https://img.shields.io/node/v/node-env-manager.svg?style=flat-square&maxAge=600

lib/index.ts

@@ -36,9 +36,10 @@ export const get = () => {
  * @param env {string}
  */
 export const set = (env: ENV | string) => {
-    if (typeof env === "string") {
-        env = env.trim().toLowerCase();
+    if (typeof env !== "string") {
+        return false;
     }
+    env = env.trim().toLowerCase();
     process.env.NODE_ENV = env;
     return true;
 };
@@ -137,7 +138,7 @@ export const isGray = () => {
  * 是否CI 环境
  * @returns boolean
  */
-export const isCi = () => {
+export const isCI = () => {
     let env = process.env.NODE_ENV;
     if (typeof env !== "string") {
         return false;

test/get.spec.ts

@@ -0,0 +1,15 @@
+import test from "ava";
+import * as env from "../lib";
+
+test.beforeEach(() => {
+    delete process.env.NODE_ENV;
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.PROD;
+    t.is(env.get(), env.ENV.PROD);
+});
+
+test((t) => {
+    t.is(env.get(), undefined);
+});

test/index.spec.ts → test/init.spec.ts

@@ -5,21 +5,15 @@ test.beforeEach(() => {
     delete process.env.NODE_ENV;
 });
 
-test((t) => {
-    env.init();
+test("Default", (t) => {
+    t.true(env.init());
     t.true(env.isDev());
 });
 
 test((t) => {
-    process.env.NODE_ENV = "test";
-    env.init();
-    t.false(env.isDev());
+    process.env.NODE_ENV = env.ENV.TEST;
+    t.false(env.init());
     t.true(env.isTest());
-});
-
-test((t) => {
-    env.init("test");
     t.false(env.isDev());
-    env.set("dev");
-    t.true(env.isDev());
+    t.true(env.isTest());
 });

test/is.spec.ts

@@ -0,0 +1,68 @@
+import test from "ava";
+import * as env from "../lib";
+
+test.beforeEach(() => {
+    delete process.env.NODE_ENV;
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.PROD;
+    t.true(env.isProd());
+});
+
+test((t) => {
+    t.false(env.isProd());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.DEV;
+    t.true(env.isDev());
+});
+test((t) => {
+    t.false(env.isDev());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.LOCAL;
+    t.true(env.isLocal());
+});
+
+test((t) => {
+    t.false(env.isLocal());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.TEST;
+    t.true(env.isTest());
+});
+
+test((t) => {
+    t.false(env.isTest());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.EXP;
+    t.true(env.isExp());
+});
+
+test((t) => {
+    t.false(env.isExp());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.GRAY;
+    t.true(env.isGray());
+});
+
+test((t) => {
+    t.false(env.isGray());
+});
+
+test((t) => {
+    process.env.NODE_ENV = env.ENV.CI;
+    t.true(env.isCI());
+});
+
+test((t) => {
+    t.false(env.isCI());
+});

test/set.spec.ts

@@ -0,0 +1,11 @@
+import test from "ava";
+import * as env from "../lib";
+
+test.beforeEach(() => {
+    delete process.env.NODE_ENV;
+});
+
+test((t) => {
+    t.true(env.set(env.ENV.DEV));
+    t.true(env.isDev());
+});