Модульное тестирование облачных функций

На этой странице описаны рекомендации и инструменты для написания модульных тестов для ваших функций, например, тестов, которые будут частью системы непрерывной интеграции (CI). Для упрощения тестирования Firebase предоставляет Firebase Test SDK для Cloud Functions . Он распространяется через npm под именем firebase-functions-test и является дополнительным тестовым SDK для firebase-functions . Firebase Test SDK для Cloud Functions :

  • Заботится о соответствующей настройке и завершении ваших тестов, например, об установке и снятии настроек переменных среды, необходимых для firebase-functions .
  • Генерирует образцы данных и контекст событий, так что вам остается только указать поля, имеющие отношение к вашему тесту.

Тестовая установка

Установите firebase-functions-test и Mocha , фреймворк для тестирования, выполнив следующие команды в папке функций:

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

Затем создайте test папку внутри папки функций, создайте в ней новый файл для вашего тестового кода и назовите его, например, index.test.js .

Наконец, измените functions/package.json , добавив следующее:

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

После того как вы написали тесты, вы можете запустить их, выполнив npm test внутри вашего каталога функций.

Инициализация Firebase Test SDK для Cloud Functions

Существует два способа использования firebase-functions-test :

  1. Онлайн-режим (рекомендуется): пишите тесты, взаимодействующие с проектом Firebase, предназначенным для тестирования, чтобы запись в базу данных, создание данных пользователем и т. д. происходили на самом деле, а ваш тестовый код мог проверять результаты. Это также означает, что другие SDK Google, используемые в ваших функциях, также будут работать.
  2. Автономный режим: пишите изолированные и автономные модульные тесты без побочных эффектов. Это означает, что любые вызовы методов, взаимодействующие с продуктом Firebase (например, запись в базу данных или создание пользователя), должны быть снабжены заглушками. Использование автономного режима, как правило, не рекомендуется при наличии функций Cloud Firestore или Realtime Database , поскольку это значительно увеличивает сложность тестового кода.

Инициализировать SDK в онлайн-режиме (рекомендуется)

Если вы хотите написать тесты, взаимодействующие с тестовым проектом, вам необходимо предоставить значения конфигурации проекта, необходимые для инициализации приложения через firebase-admin , а также путь к файлу ключа учетной записи службы.

Чтобы получить значения конфигурации вашего проекта Firebase:

  1. Откройте настройки вашего проекта в консоли Firebase .
  2. В разделе Ваши приложения выберите нужное приложение.
  3. На правой панели выберите опцию загрузки файла конфигурации для приложений Apple и Android.

    Для веб-приложений выберите Конфигурация , чтобы отобразить значения конфигурации.

Чтобы создать файл ключа:

  1. Откройте панель «Учетные записи служб» консоли Google Cloud .
  2. Выберите учетную запись службы App Engine по умолчанию и используйте меню параметров справа, чтобы выбрать Создать ключ .
  3. При появлении запроса выберите JSON в качестве типа ключа и нажмите Создать .

После сохранения файла ключа инициализируйте 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');

Инициализировать SDK в автономном режиме

Если вы хотите писать полностью автономные тесты, вы можете инициализировать SDK без каких-либо параметров:

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

Имитация значений конфигурации

Если вы используете functions.config() в коде функций, вы можете имитировать значения конфигурации. Например, если functions/index.js содержит следующий код:

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

Затем вы можете имитировать значение внутри вашего тестового файла следующим образом:

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

Импорт ваших функций

Чтобы импортировать функции, используйте require для импорта основного файла функций как модуля. Делайте это только после инициализации firebase-functions-test и создания фиктивных значений конфигурации.

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

Если вы инициализировали firebase-functions-test в автономном режиме и в коде ваших функций есть admin.initializeApp() , то вам необходимо добавить заглушку перед импортом ваших функций:

// 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');

Тестирование фоновых (не HTTP) функций

Процесс тестирования функций, не связанных с HTTP, включает следующие этапы:

  1. Оберните функцию, которую вы хотите протестировать, с помощью метода test.wrap
  2. Построить тестовые данные
  3. Вызовите обернутую функцию с созданными вами тестовыми данными и любыми полями контекста события, которые вы хотите указать.
  4. Высказывайте свои суждения о поведении.

Сначала оберните функцию, которую вы хотите протестировать. Допустим, у вас есть функция в functions/index.js с именем makeUppercase , которую вы хотите протестировать. В файле functions/test/index.test.js напишите следующее:

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

wrapped — это функция, которая вызывает makeUppercase при вызове. wrapped принимает 2 параметра:

  1. data (обязательно): данные для отправки в makeUppercase . Это напрямую соответствует первому параметру, переданному в написанный вами обработчик функции. firebase-functions-test предоставляет методы для создания пользовательских данных или примеров данных.
  2. eventContextOptions (необязательно): поля контекста события, которые вы хотите указать. Контекст события — это второй параметр, передаваемый обработчику функции, который вы написали. Если вы не включите параметр eventContextOptions при вызове wrapped , контекст события всё равно будет сгенерирован с осмысленными полями. Вы можете переопределить некоторые из сгенерированных полей, указав их здесь. Обратите внимание, что вам нужно указать только те поля, которые вы хотите переопределить. Все поля, которые вы не переопределили, будут сгенерированы.
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
});

Создание тестовых данных

Первый параметр обёрнутой функции — это тестовые данные, с которыми вызывается базовая функция. Существует несколько способов создания тестовых данных.

Использование пользовательских данных

firebase-functions-test есть ряд функций для создания данных, необходимых для тестирования ваших функций. Например, используйте test.firestore.makeDocumentSnapshot для создания снимка документа Firestore DocumentSnapshot . Первый аргумент — это данные, второй — полный путь к снимку, а третий аргумент может быть необязательным для указания других свойств снимка.

// 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);

При тестировании функции onUpdate или onWrite необходимо создать два снимка состояния: один для состояния «до» и один для состояния «после». Затем можно использовать метод makeChange для создания объекта Change с этими снимками.

// 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);

Аналогичные функции для всех остальных типов данных см. в справочнике API .

Использование примеров данных

Если вам не нужно настраивать данные, используемые в тестах, то firebase-functions-test предлагает методы для генерации примеров данных для каждого типа функции.

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

Методы получения примеров данных для каждого типа функции см. в справочнике API.

Использование заглушек данных (для автономного режима)

Если вы инициализировали SDK в автономном режиме и тестируете функцию Cloud Firestore или Realtime Database , вам следует использовать простой объект с заглушками вместо создания фактического DocumentSnapshot или DataSnapshot .

Предположим, вы пишете модульный тест для следующей функции:

// 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);
    });

Внутри функции snap используется дважды:

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

В тестовом коде создайте простой объект, в котором будут работать оба этих пути кода, и используйте Sinon для заглушек методов.

// 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);

Делать утверждения

После инициализации SDK, упаковки функций и построения данных вы можете вызывать обёрнутые функции с построенными данными и делать утверждения о поведении. Для этого можно использовать библиотеку, например Chai .

Высказывание утверждений в онлайн-режиме

Если вы инициализировали Firebase Test SDK для Cloud Functions в онлайн-режиме , вы можете подтвердить, что желаемые действия (например, запись в базу данных) были выполнены, используя firebase-admin SDK.

В примере ниже утверждается, что «INPUT» был записан в базу данных тестового проекта.

// 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');
  });
});

Выполнение утверждений в автономном режиме

Вы можете сделать утверждения об ожидаемом возвращаемом значении функции:

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);
});

Вы также можете использовать шпионов Sinon , чтобы подтвердить, что были вызваны определенные методы с ожидаемыми вами параметрами.

Тестирование HTTP-функций

Для тестирования функций HTTP onCall используйте тот же подход, что и для тестирования фоновых функций .

Если вы тестируете функции HTTP onRequest, вам следует использовать firebase-functions-test если:

  • Вы используете functions.config()
  • Ваша функция взаимодействует с проектом Firebase или другими API Google, и вы хотели бы использовать реальный проект Firebase и его учетные данные для своих тестов.

Функция HTTP onRequest принимает два параметра: объект запроса и объект ответа. Вот как можно протестировать пример функции addMessage() :

  • Переопределите функцию перенаправления в объекте ответа, поскольку sendMessage() ее вызывает.
  • Внутри функции перенаправления используйте chai.assert , чтобы сделать утверждения о том, с какими параметрами должна вызываться функция перенаправления:
// 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);

Тестовая очистка

В самом конце тестового кода вызовите функцию очистки. Это сбросит переменные среды, установленные SDK при инициализации, и удалит приложения Firebase, которые могли быть созданы при использовании SDK для создания снимков базы данных в режиме реального времени DataSnapshot или Firestore DocumentSnapshot .

test.cleanup();

Ознакомьтесь с полными примерами и узнайте больше

Полные примеры можно просмотреть в репозитории Firebase GitHub.

Более подробную информацию см. в справочнике API для firebase-functions-test .