const { Locust, TaskSet, HTTPClient } = require('node-locust');

class UserTasks extends TaskSet {
  constructor() {
    super();

    this.http = new HTTPClient({
      baseURL: <test-server>,
      timeout: 2000
    },  this.recordSuccess.bind(this),
        this.recordFailure.bind(this));

    this.tasks = [{
      task: () => this.getUserInfo(),
      weight: 2
    }, {
      task: () => this.updateUserInfo(),
      weight: 1
    }];
  }

  async getUserInfo() {
    await this.http.request({
      method: 'GET',
      url: '/users/1'
    });
  }

  async updateUserInfo() {
    await this.http.request({
      method: 'PUT',
      url: '/users/1',
      body: {
        name: 'user-name'
      }
    });
  }
}

class MyLocust extends Locust {
  constructor() {
    super();

    this.taskSet = new MyTaskSet();
  }
}

module.exports = { MyLocust };