Kiểm thử đơn vị của Cloud Functions

Trang này mô tả các phương pháp hay nhất và công cụ để viết mã kiểm thử đơn vị cho các hàm, chẳng hạn như các bài kiểm thử sẽ là một phần của hệ thống Tích hợp liên tục (CI). Để giúp việc kiểm thử trở nên dễ dàng hơn, Firebase cung cấp Firebase Test SDK cho Cloud Functions. SDK này được phân phối trên npm dưới dạng firebase-functions-test và là SDK kiểm thử đồng hành với firebase-functions. Firebase Test SDK cho Cloud Functions:

  • Chịu trách nhiệm thiết lập và huỷ bỏ thích hợp cho các bài kiểm thử, chẳng hạn như thiết lập và huỷ thiết lập các biến môi trường mà firebase-functions cần.
  • Tạo dữ liệu mẫu và bối cảnh sự kiện để bạn chỉ phải chỉ định các trường có liên quan đến bài kiểm thử.

Thiết lập kiểm thử

Cài đặt cả firebase-functions-testMocha ( một khung kiểm thử) bằng cách chạy các lệnh sau trong thư mục hàm:

npm install --save-dev firebase-functions-test
npm install --save-dev mocha

Tiếp theo, hãy tạo một thư mục test bên trong thư mục hàm, tạo một tệp mới bên trong thư mục đó cho mã kiểm thử và đặt tên cho tệp đó, chẳng hạn như index.test.js.

Cuối cùng, hãy sửa đổi functions/package.json để thêm nội dung sau:

"scripts": {
  "test": "mocha --reporter spec"
}

Sau khi viết mã kiểm thử, bạn có thể chạy mã kiểm thử đó bằng cách chạy npm test bên trong thư mục hàm.

Khởi chạy Firebase Test SDK cho Cloud Functions

Có 2 cách để sử dụng firebase-functions-test:

  1. Chế độ trực tuyến (nên dùng): Viết mã kiểm thử tương tác với một dự án Firebase dành riêng cho việc kiểm thử để các thao tác ghi vào cơ sở dữ liệu, tạo người dùng, v.v. thực sự diễn ra và mã kiểm thử có thể kiểm tra kết quả. Điều này cũng có nghĩa là các SDK khác của Google được dùng trong hàm cũng sẽ hoạt động.
  2. Chế độ ngoại tuyến: Viết mã kiểm thử đơn vị ngoại tuyến và biệt lập mà không có tác dụng phụ. Điều này có nghĩa là mọi lệnh gọi phương thức tương tác với một sản phẩm Firebase (ví dụ: ghi vào cơ sở dữ liệu hoặc tạo người dùng) đều cần được tạo bản sao. Bạn thường không nên sử dụng chế độ ngoại tuyến nếu có các hàm Cloud Firestore hoặc Realtime Database, vì chế độ này làm tăng đáng kể độ phức tạp của mã kiểm thử.

Khởi chạy SDK ở chế độ trực tuyến (nên dùng)

Nếu muốn viết mã kiểm thử tương tác với một dự án kiểm thử, bạn cần cung cấp các giá trị cấu hình dự án cần thiết để khởi chạy ứng dụng thông qua firebase-admin và đường dẫn đến tệp khoá tài khoản dịch vụ.

Cách lấy các giá trị cấu hình của dự án Firebase:

  1. Mở phần cài đặt dự án trong bảng điều khiển Firebase.
  2. Trong phần Your apps (Ứng dụng của bạn), hãy chọn ứng dụng mong muốn.
  3. Trong ngăn bên phải, hãy chọn tuỳ chọn tải tệp cấu hình xuống cho ứng dụng Apple và Android.

    Đối với ứng dụng web, hãy chọn Config (Cấu hình) để hiển thị các giá trị cấu hình.

Cách tạo tệp khoá:

  1. Mở ngăn Tài khoản dịch vụ của bảng điều khiển Google Cloud.
  2. Chọn tài khoản dịch vụ mặc định App Engine rồi sử dụng trình đơn tùy chọn ở bên phải để chọn Create key (Tạo khoá).
  3. Khi được nhắc, hãy chọn JSON cho loại khoá rồi nhấp vào Create (Tạo).

Sau khi lưu tệp khoá, hãy khởi chạy SDK:

// At the top of test/index.test.js
// Make sure to use values from your actual Firebase configuration
const test = require('firebase-functions-test')({
  databaseURL: 'https://PROJECT_ID.firebaseio.com',
  storageBucket: 'PROJECT_ID.firebasestorage.app',
  projectId: 'PROJECT_ID',
}, 'path/to/serviceAccountKey.json');

Khởi chạy SDK ở chế độ ngoại tuyến

Nếu muốn viết mã kiểm thử hoàn toàn ngoại tuyến, bạn có thể khởi chạy SDK mà không cần tham số:

// At the top of test/index.test.js
const test = require('firebase-functions-test')();

Mô phỏng các giá trị cấu hình

Nếu sử dụng functions.config() trong mã hàm, bạn có thể mô phỏng các giá trị cấu hình. Ví dụ: nếu functions/index.js chứa mã sau:

const functions = require('firebase-functions/v1');
const key = functions.config().stripe.key;

Sau đó, bạn có thể mô phỏng giá trị bên trong tệp kiểm thử như sau:

// Mock functions config values
test.mockConfig({ stripe: { key: '23wr42ewr34' }});

Nhập hàm

Để nhập hàm, hãy sử dụng require để nhập tệp hàm chính dưới dạng mô-đun. Hãy nhớ chỉ thực hiện việc này sau khi khởi chạy firebase-functions-test, và mô phỏng các giá trị cấu hình.

// after firebase-functions-test has been initialized
const myFunctions = require('../index.js'); // relative path to functions code

Nếu bạn đã khởi chạy firebase-functions-testchế độ ngoại tuyến và có admin.initializeApp() trong mã hàm, thì bạn cần tạo bản sao cho mã đó trước khi nhập hàm:

// If index.js calls admin.initializeApp at the top of the file,
// we need to stub it out before requiring index.js. This is because the
// functions will be executed as a part of the require process.
// Here we stub admin.initializeApp to be a dummy function that doesn't do anything.
adminInitStub = sinon.stub(admin, 'initializeApp');
// Now we can require index.js and save the exports inside a namespace called myFunctions.
myFunctions = require('../index');

Kiểm thử các hàm nền (không phải HTTP)

Quy trình kiểm thử các hàm không phải HTTP bao gồm các bước sau:

  1. Gói hàm mà bạn muốn kiểm thử bằng phương thức test.wrap
  2. Tạo dữ liệu kiểm thử
  3. Gọi hàm được gói bằng dữ liệu kiểm thử mà bạn đã tạo và mọi trường bối cảnh sự kiện mà bạn muốn chỉ định.
  4. Đưa ra các xác nhận về hành vi.

Trước tiên, hãy gói hàm mà bạn muốn kiểm thử. Giả sử bạn có một hàm trong functions/index.js có tên là makeUppercase mà bạn muốn kiểm thử. Viết nội dung sau trong functions/test/index.test.js

// "Wrap" the makeUpperCase function from index.js
const myFunctions = require('../index.js');
const wrapped = test.wrap(myFunctions.makeUppercase);

wrapped là một hàm gọi makeUppercase khi được gọi. wrapped có 2 tham số:

  1. data (bắt buộc): dữ liệu cần gửi đến makeUppercase. Tham số này tương ứng trực tiếp với tham số đầu tiên được gửi đến trình xử lý hàm mà bạn đã viết. firebase-functions-test cung cấp các phương thức để tạo dữ liệu tuỳ chỉnh hoặc dữ liệu mẫu.
  2. eventContextOptions (không bắt buộc): các trường của bối cảnh sự kiện mà bạn muốn chỉ định. Bối cảnh sự kiện là tham số thứ hai được gửi đến trình xử lý hàm mà bạn đã viết. Nếu bạn không đưa tham số eventContextOptions vào khi gọi wrapped, thì bối cảnh sự kiện vẫn được tạo bằng các trường hợp lý. Bạn có thể ghi đè một số trường được tạo bằng cách chỉ định các trường đó tại đây. Xin lưu ý rằng bạn chỉ phải đưa vào các trường mà bạn muốn ghi đè. Mọi trường mà bạn không ghi đè đều được tạo.
const data =  // See next section for constructing test data

// Invoke the wrapped function without specifying the event context.
wrapped(data);

// Invoke the function, and specify params
wrapped(data, {
  params: {
    pushId: '234234'
  }
});

// Invoke the function, and specify auth and auth Type (for real time database functions only)
wrapped(data, {
  auth: {
    uid: 'jckS2Q0'
  },
  authType: 'USER'
});

// Invoke the function, and specify all the fields that can be specified
wrapped(data, {
  eventId: 'abc',
  timestamp: '2018-03-23T17:27:17.099Z',
  params: {
    pushId: '234234'
  },
  auth: {
    uid: 'jckS2Q0' // only for real time database functions
  },
  authType: 'USER' // only for real time database functions
});

Tạo dữ liệu kiểm thử

Tham số đầu tiên của hàm được gói là dữ liệu kiểm thử để gọi hàm cơ bản. Có một số cách để tạo dữ liệu kiểm thử.

Sử dụng dữ liệu tuỳ chỉnh

firebase-functions-test có một số hàm để tạo dữ liệu cần thiết nhằm kiểm thử các hàm. Ví dụ: sử dụng test.firestore.makeDocumentSnapshot để tạo một DocumentSnapshot Firestore. Đối số đầu tiên là dữ liệu và đối số thứ hai là đường dẫn tham chiếu đầy đủ và có đối số thứ ba không bắt buộc cho các thuộc tính khác của ảnh chụp nhanh mà bạn có thể chỉ định.

// Make snapshot
const snap = test.firestore.makeDocumentSnapshot({foo: 'bar'}, 'document/path');
// Call wrapped function with the snapshot
const wrapped = test.wrap(myFunctions.myFirestoreDeleteFunction);
wrapped(snap);

Nếu đang kiểm thử hàm onUpdate hoặc onWrite, bạn cần tạo 2 ảnh chụp nhanh: một cho trạng thái trước và một cho trạng thái sau. Sau đó, bạn có thể sử dụng phương thức makeChange để tạo đối tượng Change bằng các ảnh chụp nhanh này.

// Make snapshot for state of database beforehand
const beforeSnap = test.firestore.makeDocumentSnapshot({foo: 'bar'}, 'document/path');
// Make snapshot for state of database after the change
const afterSnap = test.firestore.makeDocumentSnapshot({foo: 'faz'}, 'document/path');
const change = test.makeChange(beforeSnap, afterSnap);
// Call wrapped function with the Change object
const wrapped = test.wrap(myFunctions.myFirestoreUpdateFunction);
wrapped(change);

Xem tài liệu tham khảo API để biết các hàm tương tự cho tất cả các loại dữ liệu khác.

Sử dụng dữ liệu mẫu

Nếu không cần tuỳ chỉnh dữ liệu được dùng trong các bài kiểm thử, thì firebase-functions-test sẽ cung cấp các phương thức để tạo dữ liệu mẫu cho từng loại hàm.

// For Firestore onCreate or onDelete functions
const snap = test.firestore.exampleDocumentSnapshot();
// For Firestore onUpdate or onWrite functions
const change = test.firestore.exampleDocumentSnapshotChange();

Xem tài liệu tham khảo API để biết các phương thức lấy dữ liệu mẫu cho mọi loại hàm.

Sử dụng dữ liệu được tạo bản sao (cho chế độ ngoại tuyến)

Nếu bạn đã khởi chạy SDK ở chế độ ngoại tuyến và đang kiểm thử một Cloud Firestore hoặc Realtime Database hàm, thì bạn nên sử dụng một đối tượng thuần tuý có các bản sao thay vì tạo DocumentSnapshot hoặc DataSnapshot thực tế.

Giả sử bạn đang viết mã kiểm thử đơn vị cho hàm sau:

// Listens for new messages added to /messages/:pushId/original and creates an
// uppercase version of the message to /messages/:pushId/uppercase
exports.makeUppercase = functions.database.ref('/messages/{pushId}/original')
    .onCreate((snapshot, context) => {
      // Grab the current value of what was written to the Realtime Database.
      const original = snapshot.val();
      functions.logger.log('Uppercasing', context.params.pushId, original);
      const uppercase = original.toUpperCase();
      // You must return a Promise when performing asynchronous tasks inside a Functions such as
      // writing to the Firebase Realtime Database.
      // Setting an "uppercase" sibling in the Realtime Database returns a Promise.
      return snapshot.ref.parent.child('uppercase').set(uppercase);
    });

Bên trong hàm, snap được dùng 2 lần:

  • snap.val()
  • snap.ref.parent.child('uppercase').set(uppercase)

Trong mã kiểm thử, hãy tạo một đối tượng thuần tuý để cả 2 đường dẫn mã này đều hoạt động, và sử dụng Sinon để tạo mã giả lập cho các phương thức.

// The following lines creates a fake snapshot, 'snap', which returns 'input' when snap.val() is called,
// and returns true when snap.ref.parent.child('uppercase').set('INPUT') is called.
const snap = {
  val: () => 'input',
  ref: {
    parent: {
      child: childStub,
    }
  }
};
childStub.withArgs(childParam).returns({ set: setStub });
setStub.withArgs(setParam).returns(true);

Đưa ra các xác nhận

Sau khi khởi chạy SDK, gói các hàm và tạo dữ liệu, bạn có thể gọi các hàm được gói bằng dữ liệu đã tạo và đưa ra các xác nhận về hành vi. Bạn có thể sử dụng một thư viện như Chai để đưa ra các xác nhận này.

Đưa ra các xác nhận ở chế độ trực tuyến

Nếu đã khởi chạy Firebase Test SDK cho Cloud Functions ở chế độ trực tuyến, bạn có thể xác nhận rằng các hành động mong muốn (chẳng hạn như thao tác ghi vào cơ sở dữ liệu) đã diễn ra bằng cách sử dụng SDK firebase-admin.

Ví dụ bên dưới xác nhận rằng "INPUT" đã được ghi vào cơ sở dữ liệu của dự án kiểm thử.

// Create a DataSnapshot with the value 'input' and the reference path 'messages/11111/original'.
const snap = test.database.makeDataSnapshot('input', 'messages/11111/original');

// Wrap the makeUppercase function
const wrapped = test.wrap(myFunctions.makeUppercase);
// Call the wrapped function with the snapshot you constructed.
return wrapped(snap).then(() => {
  // Read the value of the data at messages/11111/uppercase. Because `admin.initializeApp()` is
  // called in functions/index.js, there's already a Firebase app initialized. Otherwise, add
  // `admin.initializeApp()` before this line.
  return admin.database().ref('messages/11111/uppercase').once('value').then((createdSnap) => {
    // Assert that the value is the uppercased version of our input.
    assert.equal(createdSnap.val(), 'INPUT');
  });
});

Đưa ra các xác nhận ở chế độ ngoại tuyến

Bạn có thể đưa ra các xác nhận về giá trị trả về dự kiến của hàm:

const childParam = 'uppercase';
const setParam = 'INPUT';
// Stubs are objects that fake and/or record function calls.
// These are excellent for verifying that functions have been called and to validate the
// parameters passed to those functions.
const childStub = sinon.stub();
const setStub = sinon.stub();
// The following lines creates a fake snapshot, 'snap', which returns 'input' when snap.val() is called,
// and returns true when snap.ref.parent.child('uppercase').set('INPUT') is called.
const snap = {
  val: () => 'input',
  ref: {
    parent: {
      child: childStub,
    }
  }
};
childStub.withArgs(childParam).returns({ set: setStub });
setStub.withArgs(setParam).returns(true);
// Wrap the makeUppercase function.
const wrapped = test.wrap(myFunctions.makeUppercase);
// Since we've stubbed snap.ref.parent.child(childParam).set(setParam) to return true if it was
// called with the parameters we expect, we assert that it indeed returned true.
return wrapped(snap).then(makeUppercaseResult => {
  return assert.equal(makeUppercaseResult, true);
});

Bạn cũng có thể sử dụng các gián điệp Sinon để xác nhận rằng một số phương thức đã được gọi và có các tham số mà bạn mong đợi.

Kiểm thử các hàm HTTP

Để kiểm thử các hàm HTTP onCall, hãy sử dụng cùng một phương pháp như kiểm thử các hàm nền.

Nếu đang kiểm thử các hàm HTTP onRequest, bạn nên sử dụng firebase-functions-test nếu:

  • Bạn sử dụng functions.config()
  • Hàm của bạn tương tác với một dự án Firebase hoặc các API khác của Google và bạn muốn sử dụng một dự án Firebase thực tế và thông tin đăng nhập của dự án đó cho các bài kiểm thử.

Hàm HTTP onRequest có 2 tham số: một đối tượng yêu cầu và một đối tượng phản hồi. Sau đây là cách bạn có thể kiểm thử addMessage() hàm mẫu:

  • Ghi đè hàm chuyển hướng trong đối tượng phản hồi, vì sendMessage() gọi hàm này.
  • Trong hàm chuyển hướng, hãy sử dụng chai.assert để giúp đưa ra các xác nhận về những tham số mà hàm chuyển hướng sẽ được gọi:
// A fake request object, with req.query.text set to 'input'
const req = { query: {text: 'input'} };
// A fake response object, with a stubbed redirect function which asserts that it is called
// with parameters 303, 'new_ref'.
const res = {
  redirect: (code, url) => {
    assert.equal(code, 303);
    assert.equal(url, 'new_ref');
    done();
  }
};

// Invoke addMessage with our fake request and response objects. This will cause the
// assertions in the response object to be evaluated.
myFunctions.addMessage(req, res);

Dọn dẹp kiểm thử

Ở cuối mã kiểm thử, hãy gọi hàm dọn dẹp. Hàm này sẽ huỷ thiết lập các biến môi trường mà SDK đã thiết lập khi được khởi chạy và xoá các ứng dụng Firebase có thể đã được tạo nếu bạn sử dụng SDK để tạo cơ sở dữ liệu theo thời gian thực DataSnapshot hoặc Firestore DocumentSnapshot.

test.cleanup();

Xem lại các ví dụ hoàn chỉnh và tìm hiểu thêm

Bạn có thể xem lại các ví dụ hoàn chỉnh trên kho lưu trữ Firebase GitHub.

Để tìm hiểu thêm, hãy tham khảo Tài liệu tham khảo API cho firebase-functions-test.